wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-08-12 11:50:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH: batch_transform now supports field filtering.

Browse Source
This commit is contained in:
Thomas Wiecki
2013-01-16 10:30:37 -05:00
parent b68c51afb4
commit b815a57430
3 changed files with 42 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_transforms.py
+8
View File
@@ -346,6 +346,14 @@ class TestBatchTransform(TestCase):
self.assertIn(0, data.columns)
self.assertNotIn(1, data.columns)
for data in algo.history_return_field_filter[wl:]:
self.assertIn('price', data.items)
self.assertNotIn('ignore', data.items)
for data in algo.history_return_field_no_filter[wl:]:
self.assertIn('price', data.items)
self.assertIn('ignore', data.items)
# test overloaded class
for test_history in [algo.history_return_price_class,
algo.history_return_price_decorator]: