wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-04 21:02:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleaned up unittests in response to Eddie's comments.

Browse Source
This commit is contained in:
Thomas Wiecki
2012-09-28 09:20:17 -04:00
parent 39c3cf88ce
commit c2dd51c24e
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_transforms.py
+4 -3
View File
@@ -298,7 +298,8 @@ class BatchTransformTestCase(TestCase):
algo = BatchTransformAlgorithm(sids=[0, 1])
algo.run(self.source)
assert algo.history_return_price_class[:2] == algo.history_return_price_decorator[:2] == [None, None], "First two iterations should return None"
self.assertEqual(algo.history_return_price_class[:2], [None, None], "First two iterations should return None")
self.assertEqual(algo.history_return_price_decorator[:2], [None, None], "First two iterations should return None")
# test overloaded class
for test_history in [algo.history_return_price_class, algo.history_return_price_decorator]:
@@ -308,9 +309,9 @@ class BatchTransformTestCase(TestCase):
def test_passing_of_args(self):
algo = BatchTransformAlgorithm([0, 1], 1, kwarg='str')
algo.run(self.source)
self.assertEqual(algo.args, (1,))
self.assertEqual(algo.kwargs, {'kwarg':'str'})
algo.run(self.source)
expected_item = ((1, ), {'kwarg': 'str'})
self.assertEqual(algo.history_return_args, [None, None, expected_item, expected_item, expected_item])