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

ENH: Accept simulation parameters and benchmark source args to create_test_zipline

Browse Source
This commit is contained in:
Jonathan Kamens
2013-10-29 11:39:07 -04:00
parent e36f70f541
commit c3ddbc258d
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/utils/simfactory.py
+9 -1
View File
@@ -61,7 +61,8 @@ def create_test_zipline(**config):
sid,
order_amount,
order_count,
sim_params=factory.create_simulation_parameters()
sim_params=config.get('sim_params',
factory.create_simulation_parameters())
)
#-------------------
@@ -79,6 +80,13 @@ def create_test_zipline(**config):
if trade_source:
test_algo.set_sources([trade_source])
#-------------------
# Benchmark source
#-------------------
test_algo.benchmark_return_source = config.get('benchmark_source', None)
#-------------------
# Transforms
#-------------------