Skip to content

Commit 1e72f23

Browse files
fix(tests): fix tests for module-level client
1 parent 54872cb commit 1e72f23

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

tests/test_module_client.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -28,32 +28,32 @@ def reset_state_fixture() -> None:
2828

2929
def test_base_url_option() -> None:
3030
assert replicate.base_url is None
31-
assert replicate.client._client.base_url == URL("https://api.replicate.com/v1/")
31+
assert replicate.collections._client.base_url == URL("https://api.replicate.com/v1/")
3232

3333
replicate.base_url = "http://foo.com"
3434

3535
assert replicate.base_url == URL("http://foo.com")
36-
assert replicate.client._client.base_url == URL("http://foo.com")
36+
assert replicate.collections._client.base_url == URL("http://foo.com")
3737

3838

3939
def test_timeout_option() -> None:
4040
assert replicate.timeout == replicate.DEFAULT_TIMEOUT
41-
assert replicate.client._client.timeout == replicate.DEFAULT_TIMEOUT
41+
assert replicate.collections._client.timeout == replicate.DEFAULT_TIMEOUT
4242

4343
replicate.timeout = 3
4444

4545
assert replicate.timeout == 3
46-
assert replicate.client._client.timeout == 3
46+
assert replicate.collections._client.timeout == 3
4747

4848

4949
def test_max_retries_option() -> None:
5050
assert replicate.max_retries == replicate.DEFAULT_MAX_RETRIES
51-
assert replicate.client._client.max_retries == replicate.DEFAULT_MAX_RETRIES
51+
assert replicate.collections._client.max_retries == replicate.DEFAULT_MAX_RETRIES
5252

5353
replicate.max_retries = 1
5454

5555
assert replicate.max_retries == 1
56-
assert replicate.client._client.max_retries == 1
56+
assert replicate.collections._client.max_retries == 1
5757

5858

5959
def test_default_headers_option() -> None:
@@ -62,26 +62,26 @@ def test_default_headers_option() -> None:
6262
replicate.default_headers = {"Foo": "Bar"}
6363

6464
assert replicate.default_headers["Foo"] == "Bar"
65-
assert replicate.client._client.default_headers["Foo"] == "Bar"
65+
assert replicate.collections._client.default_headers["Foo"] == "Bar"
6666

6767

6868
def test_default_query_option() -> None:
6969
assert replicate.default_query is None
70-
assert replicate.client._client._custom_query == {}
70+
assert replicate.collections._client._custom_query == {}
7171

7272
replicate.default_query = {"Foo": {"nested": 1}}
7373

7474
assert replicate.default_query["Foo"] == {"nested": 1}
75-
assert replicate.client._client._custom_query["Foo"] == {"nested": 1}
75+
assert replicate.collections._client._custom_query["Foo"] == {"nested": 1}
7676

7777

7878
def test_http_client_option() -> None:
7979
assert replicate.http_client is None
8080

81-
original_http_client = replicate.client._client._client
81+
original_http_client = replicate.collections._client._client
8282
assert original_http_client is not None
8383

8484
new_client = httpx.Client()
8585
replicate.http_client = new_client
8686

87-
assert replicate.client._client._client is new_client
87+
assert replicate.collections._client._client is new_client

0 commit comments

Comments
 (0)