99
1010from orb import Orb , AsyncOrb
1111from orb ._utils import parse_datetime
12- from orb ._client import Orb , AsyncOrb
1312from tests .utils import assert_matches_type
1413from orb .types .beta import PriceEvaluateResponse
1514
1615base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
17- api_key = "My API Key"
1816
1917
2018class TestPrice :
21- strict_client = Orb (base_url = base_url , api_key = api_key , _strict_response_validation = True )
22- loose_client = Orb (base_url = base_url , api_key = api_key , _strict_response_validation = False )
23- parametrize = pytest .mark .parametrize ("client" , [strict_client , loose_client ], ids = ["strict" , "loose" ])
19+ parametrize = pytest .mark .parametrize ("client" , [False , True ], indirect = True , ids = ["loose" , "strict" ])
2420
2521 @parametrize
2622 def test_method_evaluate (self , client : Orb ) -> None :
@@ -83,22 +79,20 @@ def test_path_params_evaluate(self, client: Orb) -> None:
8379
8480
8581class TestAsyncPrice :
86- strict_client = AsyncOrb (base_url = base_url , api_key = api_key , _strict_response_validation = True )
87- loose_client = AsyncOrb (base_url = base_url , api_key = api_key , _strict_response_validation = False )
88- parametrize = pytest .mark .parametrize ("client" , [strict_client , loose_client ], ids = ["strict" , "loose" ])
82+ parametrize = pytest .mark .parametrize ("async_client" , [False , True ], indirect = True , ids = ["loose" , "strict" ])
8983
9084 @parametrize
91- async def test_method_evaluate (self , client : AsyncOrb ) -> None :
92- price = await client .beta .price .evaluate (
85+ async def test_method_evaluate (self , async_client : AsyncOrb ) -> None :
86+ price = await async_client .beta .price .evaluate (
9387 "string" ,
9488 timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
9589 timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
9690 )
9791 assert_matches_type (PriceEvaluateResponse , price , path = ["response" ])
9892
9993 @parametrize
100- async def test_method_evaluate_with_all_params (self , client : AsyncOrb ) -> None :
101- price = await client .beta .price .evaluate (
94+ async def test_method_evaluate_with_all_params (self , async_client : AsyncOrb ) -> None :
95+ price = await async_client .beta .price .evaluate (
10296 "string" ,
10397 timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
10498 timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
@@ -110,8 +104,8 @@ async def test_method_evaluate_with_all_params(self, client: AsyncOrb) -> None:
110104 assert_matches_type (PriceEvaluateResponse , price , path = ["response" ])
111105
112106 @parametrize
113- async def test_raw_response_evaluate (self , client : AsyncOrb ) -> None :
114- response = await client .beta .price .with_raw_response .evaluate (
107+ async def test_raw_response_evaluate (self , async_client : AsyncOrb ) -> None :
108+ response = await async_client .beta .price .with_raw_response .evaluate (
115109 "string" ,
116110 timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
117111 timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
@@ -123,8 +117,8 @@ async def test_raw_response_evaluate(self, client: AsyncOrb) -> None:
123117 assert_matches_type (PriceEvaluateResponse , price , path = ["response" ])
124118
125119 @parametrize
126- async def test_streaming_response_evaluate (self , client : AsyncOrb ) -> None :
127- async with client .beta .price .with_streaming_response .evaluate (
120+ async def test_streaming_response_evaluate (self , async_client : AsyncOrb ) -> None :
121+ async with async_client .beta .price .with_streaming_response .evaluate (
128122 "string" ,
129123 timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
130124 timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
@@ -138,9 +132,9 @@ async def test_streaming_response_evaluate(self, client: AsyncOrb) -> None:
138132 assert cast (Any , response .is_closed ) is True
139133
140134 @parametrize
141- async def test_path_params_evaluate (self , client : AsyncOrb ) -> None :
135+ async def test_path_params_evaluate (self , async_client : AsyncOrb ) -> None :
142136 with pytest .raises (ValueError , match = r"Expected a non-empty value for `price_id` but received ''" ):
143- await client .beta .price .with_raw_response .evaluate (
137+ await async_client .beta .price .with_raw_response .evaluate (
144138 "" ,
145139 timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
146140 timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
0 commit comments