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