def test__get_columns_none(self): data = pd.DataFrame({ 'a': [1, 2, 3], }) returned = HyperTransformer._get_columns(data, 'b') assert returned.empty
def test__get_columns_one(self): data = pd.DataFrame({ 'a': [1, 2, 3], }) returned = HyperTransformer._get_columns(data, 'a') np.testing.assert_equal(returned, np.array([1, 2, 3]))
def test__get_columns_two(self): data = pd.DataFrame({ 'b': [4, 5, 6], 'b#1': [7, 8, 9], }) returned = HyperTransformer._get_columns(data, 'b') expected = np.array([[4, 7], [5, 8], [6, 9]]) np.testing.assert_equal(returned, expected)
def test__get_columns_regex(self): data = pd.DataFrame({ 'a(b)': [4, 5, 6], 'a(b)#1': [7, 8, 9], 'b(b)': [4, 5, 6], 'b(b)#1': [7, 8, 9], }) returned = HyperTransformer._get_columns(data, 'a(b)') expected = np.array([ 'a(b)', 'a(b)#1', ]) np.testing.assert_equal(returned, expected)
def test__get_columns_error(self): data = pd.DataFrame({}) with pytest.raises(ValueError): HyperTransformer._get_columns(data, 'a')