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

adding type as variable to create_test_panel

Browse Source
This commit is contained in:
warren-oneill
2015-06-04 15:50:41 +02:00
parent 44fbdff4ac
commit 77fb100ae6
3 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_sources.py
+3 -1
View File
@@ -47,14 +47,16 @@ class TestDataFrameSource(TestCase):
"DataFrameSource should only stream selected sid 0, not sid 1."
def test_panel_source(self):
source, panel = factory.create_test_panel_source()
source, panel = factory.create_test_panel_source(source_type=5)
assert isinstance(source.start, pd.lib.Timestamp)
assert isinstance(source.end, pd.lib.Timestamp)
for event in source:
self.assertTrue('sid' in event)
self.assertTrue('arbitrary' in event)
self.assertTrue('type' in event)
self.assertTrue(hasattr(event, 'volume'))
self.assertTrue(hasattr(event, 'price'))
self.assertEquals(event['type'], 5)
self.assertEquals(event['arbitrary'], 1.)
self.assertEquals(event['sid'], 0)
self.assertTrue(isinstance(event['volume'], int))