88import pytest
99
1010from codex import Codex , AsyncCodex
11- from codex .types import User
1211from tests .utils import assert_matches_type
1312from codex ._utils import parse_datetime
13+ from codex .types .users import UserSchemaPublic
1414
1515base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
1616
@@ -25,7 +25,7 @@ def test_method_activate_account(self, client: Codex) -> None:
2525 first_name = "first_name" ,
2626 last_name = "last_name" ,
2727 )
28- assert_matches_type (User , user , path = ["response" ])
28+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
2929
3030 @pytest .mark .skip ()
3131 @parametrize
@@ -39,7 +39,7 @@ def test_method_activate_account_with_all_params(self, client: Codex) -> None:
3939 phone_number = "phone_number" ,
4040 user_provided_company_name = "user_provided_company_name" ,
4141 )
42- assert_matches_type (User , user , path = ["response" ])
42+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
4343
4444 @pytest .mark .skip ()
4545 @parametrize
@@ -52,7 +52,7 @@ def test_raw_response_activate_account(self, client: Codex) -> None:
5252 assert response .is_closed is True
5353 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
5454 user = response .parse ()
55- assert_matches_type (User , user , path = ["response" ])
55+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
5656
5757 @pytest .mark .skip ()
5858 @parametrize
@@ -65,7 +65,7 @@ def test_streaming_response_activate_account(self, client: Codex) -> None:
6565 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
6666
6767 user = response .parse ()
68- assert_matches_type (User , user , path = ["response" ])
68+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
6969
7070 assert cast (Any , response .is_closed ) is True
7171
@@ -80,7 +80,7 @@ async def test_method_activate_account(self, async_client: AsyncCodex) -> None:
8080 first_name = "first_name" ,
8181 last_name = "last_name" ,
8282 )
83- assert_matches_type (User , user , path = ["response" ])
83+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
8484
8585 @pytest .mark .skip ()
8686 @parametrize
@@ -94,7 +94,7 @@ async def test_method_activate_account_with_all_params(self, async_client: Async
9494 phone_number = "phone_number" ,
9595 user_provided_company_name = "user_provided_company_name" ,
9696 )
97- assert_matches_type (User , user , path = ["response" ])
97+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
9898
9999 @pytest .mark .skip ()
100100 @parametrize
@@ -107,7 +107,7 @@ async def test_raw_response_activate_account(self, async_client: AsyncCodex) ->
107107 assert response .is_closed is True
108108 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
109109 user = await response .parse ()
110- assert_matches_type (User , user , path = ["response" ])
110+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
111111
112112 @pytest .mark .skip ()
113113 @parametrize
@@ -120,6 +120,6 @@ async def test_streaming_response_activate_account(self, async_client: AsyncCode
120120 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
121121
122122 user = await response .parse ()
123- assert_matches_type (User , user , path = ["response" ])
123+ assert_matches_type (UserSchemaPublic , user , path = ["response" ])
124124
125125 assert cast (Any , response .is_closed ) is True
0 commit comments