From c3ddbc258d6b2572432c57f546ef1074579dc9c0 Mon Sep 17 00:00:00 2001
From: Jonathan Kamens <jik@quantopian.com>
Date: Tue, 29 Oct 2013 11:39:07 -0400
Subject: [PATCH] ENH: Accept simulation parameters and benchmark source args
 to create_test_zipline

---
 zipline/utils/simfactory.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/zipline/utils/simfactory.py b/zipline/utils/simfactory.py
index eb8008e7..8df34765 100644
--- a/zipline/utils/simfactory.py
+++ b/zipline/utils/simfactory.py
@@ -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
     #-------------------