def test_super_fit_call(self, mocker):
"""Test the call to CappingTransformer.fit."""
spy = mocker.spy(tubular.capping.CappingTransformer, "fit")
df = d.create_df_9()
x = OutOfRangeNullTransformer(quantiles={
"a": [0.1, 1],
"b": [0.5, None]
},
weights_column="c")
x.fit(df)
assert (spy.call_count == 1
), "unexpected number of calls to CappingTransformer.fit"
call_args = spy.call_args_list[0]
call_pos_args = call_args[0]
call_kwargs = call_args[1]
assert call_pos_args == (
x, ), "unexpected positional args in CappingTransformer.fit call"
expected_kwargs = {"X": d.create_df_9(), "y": None}
h.assert_equal_dispatch(
expected=expected_kwargs,
actual=call_kwargs,
msg="unexpected kwargs in CappingTransformer.fit call",
)
def test_super_init_called(self, mocker, capping_values, quantiles,
weights_column, verbose, copy):
"""Test that init calls CappingTransformer.init."""
spy = mocker.spy(tubular.capping.CappingTransformer, "__init__")
x = OutOfRangeNullTransformer(
capping_values=capping_values,
quantiles=quantiles,
weights_column=weights_column,
verbose=verbose,
copy=copy,
)
assert (spy.call_count == 1
), "unexpected number of calls to CappingTransformer.__init__"
call_args = spy.call_args_list[0]
call_pos_args = call_args[0]
call_kwargs = call_args[1]
assert call_pos_args == (
x,
), "unexpected positional args in CappingTransformer.__init__ call"
expected_kwargs = {
"capping_values": capping_values,
"quantiles": quantiles,
"weights_column": weights_column,
"verbose": verbose,
"copy": copy,
}
assert (call_kwargs == expected_kwargs
), "unexpected kwargs in CappingTransformer.__init__ call"
def test_fit_returns_self(self):
"""Test fit returns self?"""
df = d.create_df_9()
x = OutOfRangeNullTransformer(quantiles={
"a": [0.1, 1],
"b": [0.5, None]
},
weights_column="c")
x_fitted = x.fit(df)
assert (
x_fitted is x
), "Returned value from OutOfRangeNullTransformer.fit not as expected."
def test_class_methods(self, method_name):
"""Test that OutOfRangeNullTransformer has transform fit and set_replacement_values methods."""
x = OutOfRangeNullTransformer(capping_values={"a": [1, 3]})
h.test_object_method(obj=x,
expected_method=method_name,
msg=method_name)
def test_set_replacement_values_called(self, mocker):
"""Test that init calls OutOfRangeNullTransformer.set_replacement_values during fit."""
df = d.create_df_9()
x = OutOfRangeNullTransformer(quantiles={
"a": [0.1, 1],
"b": [0.5, None]
},
weights_column="c")
expected_call_args = {0: {"args": (), "kwargs": {}}}
with h.assert_function_call(
mocker,
tubular.capping.OutOfRangeNullTransformer,
"set_replacement_values",
expected_call_args,
):
x.fit(df)
def test_expected_replacement_values_set(self, value_to_set,
expected_replacement_values):
"""Test the _replacement_values attribute is modified as expected given the prior values of the attribute."""
x = OutOfRangeNullTransformer(capping_values={"a": [0, 1]})
x._replacement_values = value_to_set
x.set_replacement_values()
# also tests that capping_values is not modified
h.test_object_attributes(
obj=x,
expected_attributes={
"_replacement_values": expected_replacement_values,
"capping_values": {
"a": [0, 1]
},
},
msg=
"attributes not as expected after running set_replacement_values",
)
def test_set_replacement_values_called(self, mocker):
"""Test that init calls OutOfRangeNullTransformer.set_replacement_values during init."""
expected_call_args = {0: {"args": (), "kwargs": {}}}
with h.assert_function_call(
mocker,
tubular.capping.OutOfRangeNullTransformer,
"set_replacement_values",
expected_call_args,
):
OutOfRangeNullTransformer(quantiles={
"c": [0, 0.99],
"d": [None, 0.01]
},
verbose=True,
copy=True)
def test_inheritance(self):
"""Test that OutOfRangeNullTransformer inherits from CappingTransformer."""
x = OutOfRangeNullTransformer(capping_values={"a": [1, 3]})
h.assert_inheritance(x, tubular.capping.CappingTransformer)