def test_fit_array(self):
"""Test fit with numpy.array"""
# Setup
data = np.array([False, True, True, False, True])
# Run
transformer = BooleanTransformer(nan=0)
transformer.fit(data)
# Asserts
expect_fill_value = 0
self.assertEqual(transformer.null_transformer.fill_value,
expect_fill_value, "Unexpected fill value")
def test_fit_nan_not_ignore(self):
"""Test fit nan not equal to ignore"""
# Setup
data = pd.Series([False, True, True, False, True])
# Run
transformer = BooleanTransformer(nan=0)
transformer.fit(data)
# Asserts
expect_fill_value = 0
self.assertEqual(transformer.null_transformer.fill_value,
expect_fill_value, "Unexpected fill value")
