1212from finch .pagination import SyncSinglePage , AsyncSinglePage
1313from finch .types .hris import (
1414 CompanyBenefit ,
15+ SupportedBenefit ,
1516 UpdateCompanyBenefitResponse ,
1617 CreateCompanyBenefitsResponse ,
17- BenefitListSupportedBenefitsResponse ,
1818)
1919
2020base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
@@ -178,7 +178,7 @@ def test_streaming_response_list(self, client: Finch) -> None:
178178 @parametrize
179179 def test_method_list_supported_benefits (self , client : Finch ) -> None :
180180 benefit = client .hris .benefits .list_supported_benefits ()
181- assert_matches_type (SyncSinglePage [BenefitListSupportedBenefitsResponse ], benefit , path = ["response" ])
181+ assert_matches_type (SyncSinglePage [SupportedBenefit ], benefit , path = ["response" ])
182182
183183 @parametrize
184184 def test_raw_response_list_supported_benefits (self , client : Finch ) -> None :
@@ -187,7 +187,7 @@ def test_raw_response_list_supported_benefits(self, client: Finch) -> None:
187187 assert response .is_closed is True
188188 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
189189 benefit = response .parse ()
190- assert_matches_type (SyncSinglePage [BenefitListSupportedBenefitsResponse ], benefit , path = ["response" ])
190+ assert_matches_type (SyncSinglePage [SupportedBenefit ], benefit , path = ["response" ])
191191
192192 @parametrize
193193 def test_streaming_response_list_supported_benefits (self , client : Finch ) -> None :
@@ -196,7 +196,7 @@ def test_streaming_response_list_supported_benefits(self, client: Finch) -> None
196196 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
197197
198198 benefit = response .parse ()
199- assert_matches_type (SyncSinglePage [BenefitListSupportedBenefitsResponse ], benefit , path = ["response" ])
199+ assert_matches_type (SyncSinglePage [SupportedBenefit ], benefit , path = ["response" ])
200200
201201 assert cast (Any , response .is_closed ) is True
202202
@@ -359,7 +359,7 @@ async def test_streaming_response_list(self, async_client: AsyncFinch) -> None:
359359 @parametrize
360360 async def test_method_list_supported_benefits (self , async_client : AsyncFinch ) -> None :
361361 benefit = await async_client .hris .benefits .list_supported_benefits ()
362- assert_matches_type (AsyncSinglePage [BenefitListSupportedBenefitsResponse ], benefit , path = ["response" ])
362+ assert_matches_type (AsyncSinglePage [SupportedBenefit ], benefit , path = ["response" ])
363363
364364 @parametrize
365365 async def test_raw_response_list_supported_benefits (self , async_client : AsyncFinch ) -> None :
@@ -368,7 +368,7 @@ async def test_raw_response_list_supported_benefits(self, async_client: AsyncFin
368368 assert response .is_closed is True
369369 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
370370 benefit = response .parse ()
371- assert_matches_type (AsyncSinglePage [BenefitListSupportedBenefitsResponse ], benefit , path = ["response" ])
371+ assert_matches_type (AsyncSinglePage [SupportedBenefit ], benefit , path = ["response" ])
372372
373373 @parametrize
374374 async def test_streaming_response_list_supported_benefits (self , async_client : AsyncFinch ) -> None :
@@ -377,6 +377,6 @@ async def test_streaming_response_list_supported_benefits(self, async_client: As
377377 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
378378
379379 benefit = await response .parse ()
380- assert_matches_type (AsyncSinglePage [BenefitListSupportedBenefitsResponse ], benefit , path = ["response" ])
380+ assert_matches_type (AsyncSinglePage [SupportedBenefit ], benefit , path = ["response" ])
381381
382382 assert cast (Any , response .is_closed ) is True
0 commit comments