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

Merge pull request #1317 from quantopian/reduce-number-of-days-for-test-misc-api

Browse Source 
TST: Speed up TestMiscellaneousAPI
This commit is contained in:
Eddie Hebert
2016-07-11 15:47:51 -04:00
committed by GitHub
parent 7c170274a3 4840e124af
commit db4e06055c
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_algorithm.py
+7 -4
View File
@@ -194,6 +194,9 @@ class TestMiscellaneousAPI(WithLogger,
WithSimParams, WithSimParams,
WithDataPortal, WithDataPortal,
ZiplineTestCase): ZiplineTestCase):
START_DATE = pd.Timestamp('2006-01-03', tz='UTC')
END_DATE = pd.Timestamp('2006-01-04', tz='UTC')
SIM_PARAMS_DATA_FREQUENCY = 'minute' SIM_PARAMS_DATA_FREQUENCY = 'minute'
sids = 1, 2 sids = 1, 2
@@ -322,7 +325,7 @@ def handle_data(context, data):
'arena': 'backtest', 'arena': 'backtest',
'data_frequency': 'minute', 'data_frequency': 'minute',
'start': pd.Timestamp('2006-01-03 14:31:00+0000', tz='utc'), 'start': pd.Timestamp('2006-01-03 14:31:00+0000', tz='utc'),
'end': pd.Timestamp('2006-12-29 21:00:00+0000', tz='utc'), 'end': pd.Timestamp('2006-01-04 21:00:00+0000', tz='utc'),
'capital_base': 100000.0, 'capital_base': 100000.0,
'platform': 'zipline' 'platform': 'zipline'
} }
@@ -471,14 +474,14 @@ def handle_data(context, data):
) )
algo.run(self.data_portal) algo.run(self.data_portal)
self.assertEqual(len(expected_data), 97530) self.assertEqual(len(expected_data), 780)
self.assertEqual(collected_data_pre, expected_data) self.assertEqual(collected_data_pre, expected_data)
self.assertEqual(collected_data_post, expected_data) self.assertEqual(collected_data_post, expected_data)
self.assertEqual( self.assertEqual(
len(function_stack), len(function_stack),
97530 * 5, 3900,
'Incorrect number of functions called: %s != 780' % 'Incorrect number of functions called: %s != 3900' %
len(function_stack), len(function_stack),
) )
expected_functions = [pre, handle_data, f, g, post] * 97530 expected_functions = [pre, handle_data, f, g, post] * 97530