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

TST: Adds basic test coverage for FutureChain.__iter__

Browse Source
This commit is contained in:
Andrew Daniels
2015-10-08 14:35:00 -04:00
parent 9ecd7e18d0
commit 2e85e08768
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_assets.py
+12
View File
@@ -888,6 +888,18 @@ class TestFutureChain(TestCase):
cl = FutureChain(self.asset_finder, lambda: '2006-02-01', 'CL')
self.assertEqual(cl[-1], 3)
def test_iter(self):
""" Test the __iter__ method of FutureChain.
"""
cl = FutureChain(self.asset_finder, lambda: '2005-12-01', 'CL')
for i, contract in enumerate(cl):
self.assertEqual(contract, i)
# First contract is now invalid, so sids will be offset by one
cl = FutureChain(self.asset_finder, lambda: '2005-12-21', 'CL')
for i, contract in enumerate(cl):
self.assertEqual(contract, i + 1)
def test_root_symbols(self):
""" Test that different variations on root symbols are handled
as expected.