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

updated run method to use sim_params.

Browse Source
This commit is contained in:
fawce
2013-02-18 10:24:32 -05:00
committed by Eddie Hebert
parent d67e5d7a4b
commit 1a85781170
2 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_algorithm.py
+7 -3
View File
@@ -25,6 +25,7 @@ from zipline.sources import (SpecificEquityTrades,
DataFrameSource,
DataPanelSource)
from zipline.transforms import MovingAverage
from zipline.finance.trading import SimulationParameters
class TestRecordAlgorithm(TestCase):
@@ -88,12 +89,15 @@ class TestTransformAlgorithm(TestCase):
algo.run([self.source, self.df_source])
def test_multi_source_as_input(self):
sim_params = SimulationParameters(
self.df.index[0],
self.df.index[-1]
)
algo = TestRegisterTransformAlgorithm(
self.sim_params,
sim_params=sim_params,
sids=[0, 1, 133]
)
algo.run([self.source, self.df_source],
start=self.df.index[0], end=self.df.index[-1])
algo.run([self.source, self.df_source])
self.assertEqual(len(algo.sources), 2)
def test_df_as_input(self):