Skip to content

Commit

Permalink
refactor: black it
Browse files Browse the repository at this point in the history
  • Loading branch information
Chris Lemke committed Jan 20, 2023
1 parent 8373acb commit e414da6
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/test_transformer/test_generic_transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,12 +61,16 @@ def test_column_eval_transformer_with_eval_func_to_multiple_columns(X_strings) -
"""
)


def test_column_eval_transformer_for_value_error(X_strings) -> None:
with pytest.raises(ValueError) as error:
transformer = ColumnEvalTransformer([("email", "astype(int)")])
_ = transformer.fit_transform(X_strings)
str(error.value)
assert str(error.value) == "invalid literal for int() with base 10: 'test@test1.com'"
assert (
str(error.value) == "invalid literal for int() with base 10: 'test@test1.com'"
)


def test_column_eval_transformer_with_warning(X_numbers) -> None:
with pytest.raises(Warning) as warning:
Expand Down

0 comments on commit e414da6

Please sign in to comment.