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))
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_fit(self):
feature = Feature(self.input, self.transformer)
mapper = FeatureEngineeringPipeline(feature)
df = pd.util.testing.makeCustomDataframe(5, 2)
df.columns = ['foo', 'bar']
mapper.fit(df)