def test_X_type(self):
"""Test error is thrown if X is not the correct type."""
test_masker = Masker()
with pytest.raises(TypeError):
test_masker.fit(X=123)
def test_get_column_map_called(self, data, mocker):
"""Test that the get_column_map method is called."""
mocker.patch.object(Masker, "get_column_map")
test_masker = Masker()
test_masker.fit(X=data)
Masker.get_column_map.assert_called()