Skip to content

Commit 33534de

Browse files
committed
No warning test case added
1 parent eed7eb1 commit 33534de

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

tests/nominal/test_OneHotEncodingTransformer.py

+11-6
Original file line numberDiff line numberDiff line change
@@ -170,11 +170,13 @@ def test_fit_no_warning_if_all_wanted_values_present(self, library):
170170
df = d.create_df_1(library=library)
171171

172172
transformer = OneHotEncodingTransformer(
173-
columns=["b"], wanted_values={"b": ["a", "b", "c", "d", "e", "f"]}
173+
columns=["b"],
174+
wanted_values={"b": ["a", "b", "c", "d", "e", "f"]},
174175
)
175176

176-
with pytest.warns(None):
177+
with pytest.warns(None) as warnings:
177178
transformer.fit(df)
179+
assert len(warnings) == 0
178180

179181

180182
class TestTransform(
@@ -501,13 +503,16 @@ def test_transform_output_with_wanted_values_arg(self, library):
501503
)
502504
def test_transform_no_warning_if_all_wanted_values_present(self, library):
503505
"""Test that OneHotEncodingTransformer.transform does NOT raise a warning when all levels in wanted_levels are present in the data."""
504-
df_train = d.create_df_7(library=library)
505-
df_test = d.create_df_8(library=library)
506+
df_train = d.create_df_8(library=library)
507+
df_test = d.create_df_7(library=library)
506508

507509
transformer = OneHotEncodingTransformer(
508-
columns=["b"], wanted_values={"b": ["x", "z", "y"]}
510+
columns=["b"],
511+
wanted_values={"b": ["z", "y", "x"]},
509512
)
510513
transformer.fit(df_train)
511514

512-
with pytest.warns(None):
515+
with pytest.warns(None) as warnings:
513516
transformer.transform(df_test)
517+
assert len(warnings) == 0
518+

0 commit comments

Comments
 (0)