def test_update(self):
"""Test _FeatureExtractor update method."""
hook = Hook(key='key', func=lambda: None)
_prep = _FeatureExtractor().update(hook)
self.assertTrue('key' in _prep.keys)
def test_feed(self):
"""Test _FeatureExtractor feed method."""
hook = Hook(key='key', func=lambda x: x)
_prep = _FeatureExtractor().update(hook)
# feed the extractor with skip=True
result = _prep.feed({'x': 'test'}, skip_unfed_hooks=True)
self.assertIsInstance(result, dict)
key, value = list(*result.items())
self.assertEqual(key, 'key')
self.assertEqual(value, 'test')
# feed and disable skip
with self.assertRaises(TypeError):
result = _prep.feed({'x': 'test'}, skip_unfed_hooks=False)
key, value = list(*result.items())
self.assertEqual(key, 'key')
self.assertEqual(value, 'test')
def test_init(self):
"""Test _FeatureExtractor initialization."""
_prep = _FeatureExtractor()
self.assertTrue(any(_prep._hooks)) # pylint: disable=protected-access