def test_categoricaltransformer_mixed_low_virtual_memory(psutil_mock):
"""Test the CategoricalTransformer on mixed type data with low virtual memory.
Ensure that the CategoricalTransformer can fit, transform, and reverse
transform on mixed type data, when there is low virtual memory. Expect that the
reverse transformed data is the same as the input.
Input:
- 4 rows of mixed data
Output:
- The reverse transformed data
"""
# setup
data = pd.Series([True, 'a', 1, None])
transformer = CategoricalTransformer()
virtual_memory = Mock()
virtual_memory.available = 1
psutil_mock.return_value = virtual_memory
# run
reverse = transformer.reverse_transform(transformer.fit_transform(data))
# assert
pd.testing.assert_series_equal(data, reverse)
def test_categoricaltransformer_integers():
"""Test the CategoricalTransformer on integer data.
Ensure that the CategoricalTransformer can fit, transform, and reverse
transform on integer data. Expect that the reverse transformed data is the
same as the input.
Input:
- 4 rows of int data
Output:
- The reverse transformed data
"""
# setup
data = pd.Series([1, 2, 3, 2])
transformer = CategoricalTransformer()
# run
reverse = transformer.reverse_transform(transformer.fit_transform(data))
# assert
pd.testing.assert_series_equal(data, reverse)
def test_categoricaltransformer_strings_2_categories():
"""Test the CategoricalTransformer on string data.
Ensure that the CategoricalTransformer can fit, transform, and reverse
transform on string data, when there are 2 categories of strings with
the same value counts. Expect that the reverse transformed data is the
same as the input.
Input:
- 4 rows of string data
Output:
- The reverse transformed data
"""
# setup
data = pd.Series(['a', 'b', 'a', 'b'])
transformer = CategoricalTransformer()
reverse = transformer.reverse_transform(transformer.fit_transform(data))
# assert
pd.testing.assert_series_equal(data, reverse)
def test_categoricaltransformer_mixed():
"""Test the CategoricalTransformer on mixed type data.
Ensure that the CategoricalTransformer can fit, transform, and reverse
transform on mixed type data. Expect that the reverse transformed data is
the same as the input.
Input:
- 4 rows of mixed data
Output:
- The reverse transformed data
"""
# setup
data = pd.Series([True, 'a', 1, None])
transformer = CategoricalTransformer()
# run
reverse = transformer.reverse_transform(transformer.fit_transform(data))
# assert
pd.testing.assert_series_equal(data, reverse)
