def test_transform(input, transformer): feature = Feature(input, transformer) mapper = FeatureEngineeringPipeline(feature) df = pd.util.testing.makeCustomDataframe(5, 2) df.columns = ['foo', 'bar'] mapper.fit(df) X = mapper.transform(df) assert np.shape(X) == (5, 2)
def test_transform(self): feature = Feature(self.input, self.transformer) mapper = FeatureEngineeringPipeline(feature) df = pd.util.testing.makeCustomDataframe(5, 2) df.columns = ['foo', 'bar'] mapper.fit(df) X = mapper.transform(df) self.assertEqual(np.shape(X), (5, 1))