Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jul 18, 2024
1 parent dbad60f commit 9357e5b
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tests/data_processors/transformers/test_transformers_const.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,16 +63,16 @@ def test_const_handling_test_df(test_const_data: pd.DataFrame):
# Transform the DataFrame using the transformer.
transformed_df = const_transformer.convert(test_const_data)

assert 'age' not in transformed_df.columns
assert 'fnlwgt' not in transformed_df.columns
assert 'workclass' not in transformed_df.columns
assert "age" not in transformed_df.columns
assert "fnlwgt" not in transformed_df.columns
assert "workclass" not in transformed_df.columns

# reverse convert the df
reverse_converted_df = const_transformer.reverse_convert(transformed_df)

assert 'age' in reverse_converted_df.columns
assert 'fnlwgt' in reverse_converted_df.columns
assert 'workclass' in reverse_converted_df.columns
assert "age" in reverse_converted_df.columns
assert "fnlwgt" in reverse_converted_df.columns
assert "workclass" in reverse_converted_df.columns

assert reverse_converted_df["age"][0] == 100
assert reverse_converted_df["fnlwgt"][0] == 1.41421
Expand Down

0 comments on commit 9357e5b

Please sign in to comment.