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

TST: Fixes test_order_methods_for_future

Browse Source
This commit is contained in:
jfkirk
2015-11-16 11:47:00 -05:00
parent abbda17239
commit 11b7cd2dd0
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_algorithm.py
+6 -4
View File
@@ -645,10 +645,12 @@ class TestTransformAlgorithm(TestCase):
@classmethod
def setUpClass(cls):
futures_metadata = {3: {'contract_multiplier': 10}}
cls.env = TradingEnvironment()
cls.env.write_data(equities_identifiers=[0, 1, 133],
futures_data=futures_metadata)
cls.env.write_data(equities_identifiers=[0, 1, 133])
futures_metadata = {0: {'contract_multiplier': 10}}
cls.futures_env = TradingEnvironment()
cls.futures_env.write_data(futures_data=futures_metadata)
@classmethod
def tearDownClass(cls):
@@ -807,7 +809,7 @@ class TestTransformAlgorithm(TestCase):
def test_order_methods_for_future(self, algo_class):
algo = algo_class(
sim_params=self.sim_params,
env=self.env,
env=self.futures_env,
)
algo.run(self.df)