From 11b7cd2dd04c10f0e52739a47552a9f8454e9f6a Mon Sep 17 00:00:00 2001 From: jfkirk Date: Mon, 16 Nov 2015 11:47:00 -0500 Subject: [PATCH] TST: Fixes test_order_methods_for_future --- tests/test_algorithm.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py index 074d947f..272d203b 100644 --- a/tests/test_algorithm.py +++ b/tests/test_algorithm.py @@ -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)