def test_reverse_transform_not_null_values(self): """Test reverse_transform not null values correctly""" # Setup data = np.array([1., 0., 1.]) # Run transformer = Mock() transformer.nan = None result = BooleanTransformer.reverse_transform(transformer, data) # Asserts expected = np.array([True, False, True]) assert type(result) == pd.Series np.testing.assert_equal(result.to_numpy(), expected)
def test_reverse_transform_2d_ndarray(self): """Test reverse_transform not null values correctly""" # Setup data = np.array([[1.], [0.], [1.]]) # Run transformer = Mock() transformer.nan = None result = BooleanTransformer.reverse_transform(transformer, data) # Asserts expected = np.array([True, False, True]) assert isinstance(result, pd.Series) np.testing.assert_equal(result.values, expected)
def test_reverse_transform_nan_ignore(self): """Test reverse_transform with nan equal to ignore""" # Setup data = np.array([0.0, 1.0, 0.0, 1.0, 0.0]) # Run transformer = Mock() transformer.nan = None result = BooleanTransformer.reverse_transform(transformer, data) # Asserts expect = np.array([False, True, False, True, False]) expect_call_count = 0 np.testing.assert_equal(result, expect) self.assertEqual( transformer.null_transformer.reverse_transform.call_count, expect_call_count, "NullTransformer.reverse_transform should not be called when nan is ignore" )