def test_transform_array(self):
"""Test transform numpy.array"""
# Setup
data = np.array([False, True, None, True, False])
# Run
transformer = Mock()
BooleanTransformer.transform(transformer, data)
# Asserts
expect_call_count = 1
expect_call_args = pd.Series([0, 1, None, 1, 0], dtype=object)
self.assertEqual(transformer.null_transformer.transform.call_count,
expect_call_count,
"NullTransformer.transform must be called one time")
pd.testing.assert_series_equal(
transformer.null_transformer.transform.call_args[0][0],
expect_call_args)
def test_transform_series(self):
"""Test transform pandas.Series"""
# Setup
data = pd.Series([False, True, None, True, False])
# Run
transformer = Mock()
BooleanTransformer.transform(transformer, data)
# Asserts
expect_call_count = 1
expect_call_args = pd.Series([0., 1., None, 1., 0.], dtype=float)
self.assertEqual(transformer.null_transformer.transform.call_count,
expect_call_count,
"NullTransformer.transform must be called one time")
pd.testing.assert_series_equal(
transformer.null_transformer.transform.call_args[0][0],
expect_call_args)
