def test_inject_feature_simple(self):
f = Map(Map(F('a'), np.abs), np.abs)
self.assertEqual(str(f), 'absolute(absolute(a))')
new_f = inject_feature(f, Map, function=lambda x: x + 1000)
self.assertEqual(str(new_f), 'absolute(absolute(<lambda>(a)))')
d, ff = build_feature_safe(new_f, self.data)
assert_almost_equal(d[d.columns[0]].values, (self.data['a'] + 1000).values)
def test_inject_feature_combo(self):
f = Map(F('a') + F('b'), np.abs)
self.assertEqual(str(f), 'absolute(Add(a, b))')
new_f = inject_feature(f, Map, function=lambda x: x + 1000)
# Should inject in
self.assertEqual(str(new_f), 'absolute(Add(<lambda>(a), <lambda>(b)))')
def test_inject_feature_simple(self):
f = Map(Map(F('a'), np.abs), np.abs)
self.assertEqual(str(f), 'absolute(absolute(a))')
new_f = inject_feature(f, Map, function=lambda x: x + 1000)
self.assertEqual(str(new_f), 'absolute(absolute(<lambda>(a)))')