9
9
10
10
from orb import Orb , AsyncOrb
11
11
from orb ._utils import parse_datetime
12
- from orb ._client import Orb , AsyncOrb
13
12
from tests .utils import assert_matches_type
14
13
from orb .types .beta import PriceEvaluateResponse
15
14
16
15
base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
17
- api_key = "My API Key"
18
16
19
17
20
18
class 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" ])
24
20
25
21
@parametrize
26
22
def test_method_evaluate (self , client : Orb ) -> None :
@@ -83,22 +79,20 @@ def test_path_params_evaluate(self, client: Orb) -> None:
83
79
84
80
85
81
class 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" ])
89
83
90
84
@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 (
93
87
"string" ,
94
88
timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
95
89
timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
96
90
)
97
91
assert_matches_type (PriceEvaluateResponse , price , path = ["response" ])
98
92
99
93
@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 (
102
96
"string" ,
103
97
timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
104
98
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:
110
104
assert_matches_type (PriceEvaluateResponse , price , path = ["response" ])
111
105
112
106
@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 (
115
109
"string" ,
116
110
timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
117
111
timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
@@ -123,8 +117,8 @@ async def test_raw_response_evaluate(self, client: AsyncOrb) -> None:
123
117
assert_matches_type (PriceEvaluateResponse , price , path = ["response" ])
124
118
125
119
@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 (
128
122
"string" ,
129
123
timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
130
124
timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
@@ -138,9 +132,9 @@ async def test_streaming_response_evaluate(self, client: AsyncOrb) -> None:
138
132
assert cast (Any , response .is_closed ) is True
139
133
140
134
@parametrize
141
- async def test_path_params_evaluate (self , client : AsyncOrb ) -> None :
135
+ async def test_path_params_evaluate (self , async_client : AsyncOrb ) -> None :
142
136
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 (
144
138
"" ,
145
139
timeframe_end = parse_datetime ("2019-12-27T18:11:19.117Z" ),
146
140
timeframe_start = parse_datetime ("2019-12-27T18:11:19.117Z" ),
0 commit comments