def test_expected_output(self, df, expected):
"""Test that transform then inverse_transform gets back to the original df."""
x = NominalToIntegerTransformer(columns=["a", "b"])
# set the mapping dict directly rather than fitting x on df so test works with helpers
x.mappings = {
"a": {
1: 0,
2: 1,
3: 2,
4: 3,
5: 4,
6: 5
},
"b": {
"a": 0,
"b": 1,
"c": 2,
"d": 3,
"e": 4,
"f": 5
},
}
df_transformed = x.transform(df)
df_transformed_back = x.inverse_transform(df_transformed)
h.assert_frame_equal_msg(
actual=df_transformed_back,
expected=expected,
msg_tag="transform reverse does not get back to original",
)
def test_expected_output(self, df, expected):
"""Test that the output is expected from transform."""
x = NominalToIntegerTransformer(columns=["a", "b"])
# set the mapping dict directly rather than fitting x on df so test works with helpers
x.mappings = {
"a": {
1: 0,
2: 1,
3: 2,
4: 3,
5: 4,
6: 5
},
"b": {
"a": 0,
"b": 1,
"c": 2,
"d": 3,
"e": 4,
"f": 5
},
}
df_transformed = x.transform(df)
h.assert_frame_equal_msg(
actual=df_transformed,
expected=expected,
msg_tag=
"Unexpected values in NominalToIntegerTransformer.transform",
)