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