def test_mocker_patch_object_call(mocker):
"""Test the mocker.patch.object call."""
mocked = mocker.spy(mocker.patch, "object")
with h.assert_function_call_count(
mocker,
tubular.base.BaseTransformer,
"__init__",
1,
return_value=None,
):
tubular.imputers.BaseImputer("a", other=1)
assert mocked.call_count == 1, "unexpected number of calls to mocker.patch.object"
mocker_patch_object_call = mocked.call_args_list[0]
call_pos_args = mocker_patch_object_call[0]
call_kwargs = mocker_patch_object_call[1]
assert call_pos_args == (
tubular.base.BaseTransformer,
"__init__",
), "unexpected positional args in mocker.patch.object call"
assert call_kwargs == {
"return_value": None
}, "unexpected kwargs in mocker.patch.object call"
def test_successful_usage(mocker):
"""Test an example of successful run of h.assert_function_call_count."""
df = d.create_df_1()
x = tubular.base.BaseTransformer(columns="a")
with h.assert_function_call_count(mocker, tubular.base.BaseTransformer,
"columns_set_or_check", 1):
x.fit(X=df)
def test_check_is_fitted_call_count(self, mocker):
"""Test there are 2 calls to BaseTransformer check_is_fitted in transform."""
df = d.create_df_3()
x = CappingTransformer(capping_values={"a": [2, 5], "b": [-1, 8]})
with h.assert_function_call_count(
mocker, tubular.base.BaseTransformer, "check_is_fitted", 2
):
x.transform(df)
def test_mocker_arg_not_mocker_fixture_error():
"""Test an exception is raised if mocker is not pytest_mock.plugin.MockerFixture type."""
with pytest.raises(
TypeError,
match="mocker should be the pytest_mock mocker fixture"):
df = d.create_df_1()
x = tubular.base.BaseTransformer(columns="a")
with h.assert_function_call_count("aaaaaa",
tubular.base.BaseTransformer,
"columns_set_or_check", 1):
x.fit(X=df)
def test_exception_raised_more_calls_expected2(mocker):
"""Test an exception is raised in the case more calls to a function are expected than happen."""
with pytest.raises(
AssertionError,
match="incorrect number of calls to __init__, expected 4 but got 0",
):
df = d.create_df_1()
x = tubular.base.BaseTransformer(columns="a")
with h.assert_function_call_count(mocker, tubular.base.BaseTransformer,
"__init__", 4):
x.fit(X=df)
def test_exception_raised_less_calls_expected2(mocker):
"""Test an exception is raised in the case less calls to a function are expected than happen."""
with pytest.raises(
AssertionError,
match=
"incorrect number of calls to columns_set_or_check, expected 0 but got 1",
):
df = d.create_df_1()
x = tubular.base.BaseTransformer(columns="a")
with h.assert_function_call_count(mocker, tubular.base.BaseTransformer,
"columns_set_or_check", 0):
x.fit(X=df)