1313from finch .pagination import SyncSinglePage , AsyncSinglePage
1414from finch .types .hris .benefits import (
1515 IndividualBenefit ,
16- EnrolledIndividualBenefit ,
1716 IndividualEnrolledIDsResponse ,
1817 IndividualUnenrollManyResponse ,
18+ EnrolledIndividualBenefitResponse ,
1919)
2020
2121base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
@@ -29,7 +29,7 @@ def test_method_enroll_many(self, client: Finch) -> None:
2929 individual = client .hris .benefits .individuals .enroll_many (
3030 benefit_id = "benefit_id" ,
3131 )
32- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
32+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
3333
3434 @parametrize
3535 def test_method_enroll_many_with_all_params (self , client : Finch ) -> None :
@@ -55,7 +55,7 @@ def test_method_enroll_many_with_all_params(self, client: Finch) -> None:
5555 }
5656 ],
5757 )
58- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
58+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
5959
6060 @parametrize
6161 def test_raw_response_enroll_many (self , client : Finch ) -> None :
@@ -66,7 +66,7 @@ def test_raw_response_enroll_many(self, client: Finch) -> None:
6666 assert response .is_closed is True
6767 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
6868 individual = response .parse ()
69- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
69+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
7070
7171 @parametrize
7272 def test_streaming_response_enroll_many (self , client : Finch ) -> None :
@@ -77,7 +77,7 @@ def test_streaming_response_enroll_many(self, client: Finch) -> None:
7777 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
7878
7979 individual = response .parse ()
80- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
80+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
8181
8282 assert cast (Any , response .is_closed ) is True
8383
@@ -227,7 +227,7 @@ async def test_method_enroll_many(self, async_client: AsyncFinch) -> None:
227227 individual = await async_client .hris .benefits .individuals .enroll_many (
228228 benefit_id = "benefit_id" ,
229229 )
230- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
230+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
231231
232232 @parametrize
233233 async def test_method_enroll_many_with_all_params (self , async_client : AsyncFinch ) -> None :
@@ -253,7 +253,7 @@ async def test_method_enroll_many_with_all_params(self, async_client: AsyncFinch
253253 }
254254 ],
255255 )
256- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
256+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
257257
258258 @parametrize
259259 async def test_raw_response_enroll_many (self , async_client : AsyncFinch ) -> None :
@@ -264,7 +264,7 @@ async def test_raw_response_enroll_many(self, async_client: AsyncFinch) -> None:
264264 assert response .is_closed is True
265265 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
266266 individual = response .parse ()
267- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
267+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
268268
269269 @parametrize
270270 async def test_streaming_response_enroll_many (self , async_client : AsyncFinch ) -> None :
@@ -275,7 +275,7 @@ async def test_streaming_response_enroll_many(self, async_client: AsyncFinch) ->
275275 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
276276
277277 individual = await response .parse ()
278- assert_matches_type (EnrolledIndividualBenefit , individual , path = ["response" ])
278+ assert_matches_type (EnrolledIndividualBenefitResponse , individual , path = ["response" ])
279279
280280 assert cast (Any , response .is_closed ) is True
281281
0 commit comments