@@ -165,7 +165,7 @@ def test_fields_with_over_100_levels_error(self, library):
165
165
"library" ,
166
166
["pandas" , "polars" ],
167
167
)
168
- def test_fit_no_warning_if_all_wanted_values_present (self , library ):
168
+ def test_fit_no_warning_if_all_wanted_values_present (self , library , recwarn ):
169
169
"""Test that OneHotEncodingTransformer.fit does NOT raise a warning when all levels in wanted_levels are present in the data."""
170
170
df = d .create_df_1 (library = library )
171
171
@@ -174,9 +174,10 @@ def test_fit_no_warning_if_all_wanted_values_present(self, library):
174
174
wanted_values = {"b" : ["a" , "b" , "c" , "d" , "e" , "f" ]},
175
175
)
176
176
177
- with pytest .warns (None ) as warnings :
178
- transformer .fit (df )
179
- assert len (warnings ) == 0
177
+ transformer .fit (df )
178
+ assert (
179
+ len (recwarn ) == 0
180
+ ), "OneHotEncodingTransformer.fit is raising unexpected warnings"
180
181
181
182
182
183
class TestTransform (
@@ -501,7 +502,7 @@ def test_transform_output_with_wanted_values_arg(self, library):
501
502
"library" ,
502
503
["pandas" , "polars" ],
503
504
)
504
- def test_transform_no_warning_if_all_wanted_values_present (self , library ):
505
+ def test_transform_no_warning_if_all_wanted_values_present (self , library , recwarn ):
505
506
"""Test that OneHotEncodingTransformer.transform does NOT raise a warning when all levels in wanted_levels are present in the data."""
506
507
df_train = d .create_df_8 (library = library )
507
508
df_test = d .create_df_7 (library = library )
@@ -511,9 +512,8 @@ def test_transform_no_warning_if_all_wanted_values_present(self, library):
511
512
wanted_values = {"b" : ["z" , "y" , "x" ]},
512
513
)
513
514
transformer .fit (df_train )
515
+ transformer .transform (df_test )
514
516
515
- with pytest .warns (None ) as warnings :
516
- transformer .transform (df_test )
517
517
assert (
518
- len (warnings ) == 0
518
+ len (recwarn ) == 0
519
519
), "OneHotEncodingTransformer.transform is raising unexpected warnings"
0 commit comments