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")