1010from lithic import Lithic , AsyncLithic
1111from tests .utils import assert_matches_type
1212from lithic .pagination import SyncCursorPage , AsyncCursorPage
13- from lithic .types .financial_accounts .statements import LineItemListResponse
13+ from lithic .types .financial_accounts .statements . statement_line_items import Data
1414
1515base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
1616
@@ -24,7 +24,7 @@ def test_method_list(self, client: Lithic) -> None:
2424 statement_token = "statement_token" ,
2525 financial_account_token = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
2626 )
27- assert_matches_type (SyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
27+ assert_matches_type (SyncCursorPage [Data ], line_item , path = ["response" ])
2828
2929 @parametrize
3030 def test_method_list_with_all_params (self , client : Lithic ) -> None :
@@ -35,7 +35,7 @@ def test_method_list_with_all_params(self, client: Lithic) -> None:
3535 page_size = 1 ,
3636 starting_after = "starting_after" ,
3737 )
38- assert_matches_type (SyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
38+ assert_matches_type (SyncCursorPage [Data ], line_item , path = ["response" ])
3939
4040 @parametrize
4141 def test_raw_response_list (self , client : Lithic ) -> None :
@@ -47,7 +47,7 @@ def test_raw_response_list(self, client: Lithic) -> None:
4747 assert response .is_closed is True
4848 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
4949 line_item = response .parse ()
50- assert_matches_type (SyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
50+ assert_matches_type (SyncCursorPage [Data ], line_item , path = ["response" ])
5151
5252 @parametrize
5353 def test_streaming_response_list (self , client : Lithic ) -> None :
@@ -59,7 +59,7 @@ def test_streaming_response_list(self, client: Lithic) -> None:
5959 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
6060
6161 line_item = response .parse ()
62- assert_matches_type (SyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
62+ assert_matches_type (SyncCursorPage [Data ], line_item , path = ["response" ])
6363
6464 assert cast (Any , response .is_closed ) is True
6565
@@ -89,7 +89,7 @@ async def test_method_list(self, async_client: AsyncLithic) -> None:
8989 statement_token = "statement_token" ,
9090 financial_account_token = "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e" ,
9191 )
92- assert_matches_type (AsyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
92+ assert_matches_type (AsyncCursorPage [Data ], line_item , path = ["response" ])
9393
9494 @parametrize
9595 async def test_method_list_with_all_params (self , async_client : AsyncLithic ) -> None :
@@ -100,7 +100,7 @@ async def test_method_list_with_all_params(self, async_client: AsyncLithic) -> N
100100 page_size = 1 ,
101101 starting_after = "starting_after" ,
102102 )
103- assert_matches_type (AsyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
103+ assert_matches_type (AsyncCursorPage [Data ], line_item , path = ["response" ])
104104
105105 @parametrize
106106 async def test_raw_response_list (self , async_client : AsyncLithic ) -> None :
@@ -112,7 +112,7 @@ async def test_raw_response_list(self, async_client: AsyncLithic) -> None:
112112 assert response .is_closed is True
113113 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
114114 line_item = response .parse ()
115- assert_matches_type (AsyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
115+ assert_matches_type (AsyncCursorPage [Data ], line_item , path = ["response" ])
116116
117117 @parametrize
118118 async def test_streaming_response_list (self , async_client : AsyncLithic ) -> None :
@@ -124,7 +124,7 @@ async def test_streaming_response_list(self, async_client: AsyncLithic) -> None:
124124 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
125125
126126 line_item = await response .parse ()
127- assert_matches_type (AsyncCursorPage [LineItemListResponse ], line_item , path = ["response" ])
127+ assert_matches_type (AsyncCursorPage [Data ], line_item , path = ["response" ])
128128
129129 assert cast (Any , response .is_closed ) is True
130130
0 commit comments