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)