1111from tests .utils import assert_matches_type
1212from lithic .types import (
1313 Tokenization ,
14- TokenizationRetrieveResponse ,
1514 TokenizationSimulateResponse ,
1615 TokenizationUpdateDigitalCardArtResponse ,
1716)
@@ -29,7 +28,7 @@ def test_method_retrieve(self, client: Lithic) -> None:
2928 tokenization = client .tokenizations .retrieve (
3029 "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
3130 )
32- assert_matches_type (TokenizationRetrieveResponse , tokenization , path = ["response" ])
31+ assert_matches_type (Tokenization , tokenization , path = ["response" ])
3332
3433 @parametrize
3534 def test_raw_response_retrieve (self , client : Lithic ) -> None :
@@ -40,7 +39,7 @@ def test_raw_response_retrieve(self, client: Lithic) -> None:
4039 assert response .is_closed is True
4140 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
4241 tokenization = response .parse ()
43- assert_matches_type (TokenizationRetrieveResponse , tokenization , path = ["response" ])
42+ assert_matches_type (Tokenization , tokenization , path = ["response" ])
4443
4544 @parametrize
4645 def test_streaming_response_retrieve (self , client : Lithic ) -> None :
@@ -51,7 +50,7 @@ def test_streaming_response_retrieve(self, client: Lithic) -> None:
5150 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
5251
5352 tokenization = response .parse ()
54- assert_matches_type (TokenizationRetrieveResponse , tokenization , path = ["response" ])
53+ assert_matches_type (Tokenization , tokenization , path = ["response" ])
5554
5655 assert cast (Any , response .is_closed ) is True
5756
@@ -410,7 +409,7 @@ async def test_method_retrieve(self, async_client: AsyncLithic) -> None:
410409 tokenization = await async_client .tokenizations .retrieve (
411410 "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
412411 )
413- assert_matches_type (TokenizationRetrieveResponse , tokenization , path = ["response" ])
412+ assert_matches_type (Tokenization , tokenization , path = ["response" ])
414413
415414 @parametrize
416415 async def test_raw_response_retrieve (self , async_client : AsyncLithic ) -> None :
@@ -421,7 +420,7 @@ async def test_raw_response_retrieve(self, async_client: AsyncLithic) -> None:
421420 assert response .is_closed is True
422421 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
423422 tokenization = response .parse ()
424- assert_matches_type (TokenizationRetrieveResponse , tokenization , path = ["response" ])
423+ assert_matches_type (Tokenization , tokenization , path = ["response" ])
425424
426425 @parametrize
427426 async def test_streaming_response_retrieve (self , async_client : AsyncLithic ) -> None :
@@ -432,7 +431,7 @@ async def test_streaming_response_retrieve(self, async_client: AsyncLithic) -> N
432431 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
433432
434433 tokenization = await response .parse ()
435- assert_matches_type (TokenizationRetrieveResponse , tokenization , path = ["response" ])
434+ assert_matches_type (Tokenization , tokenization , path = ["response" ])
436435
437436 assert cast (Any , response .is_closed ) is True
438437
0 commit comments