Skip to content

Commit 1ffe9ed

Browse files
committed
chore: assert the actual result
1 parent 3759dc8 commit 1ffe9ed

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/preprocessing/test_preprocessor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,10 +170,11 @@ def test_mutable_train_data_fit_transform(self, mocker: MockerFixture):
170170
preprocessor._target_encoder = MagicMock()
171171
preprocessor._target_encoder.transform = self.mock_transform
172172

173-
_ = preprocessor.fit_transform(
173+
result = preprocessor.fit_transform(
174174
train_data,
175175
continuous_vars=["foo"],
176176
discrete_vars=["bar"],
177177
target_column_name=["baz"]
178178
)
179179
assert "new_column" not in train_data.columns
180+
assert "new_column" in result.columns

0 commit comments

Comments
 (0)