From 7f1c5100b5858c03e377d964954cef86aa0aeb7d Mon Sep 17 00:00:00 2001
From: fawce <fawce@quantopian.com>
Date: Fri, 10 Aug 2012 11:55:30 -0400
Subject: [PATCH] SimulatedTrading is iterable agian.

---
 zipline/lines.py | 27 +++++++++++++++------------
 1 file changed, 15 insertions(+), 12 deletions(-)

diff --git a/zipline/lines.py b/zipline/lines.py
index b1463e38..e861706c 100644
--- a/zipline/lines.py
+++ b/zipline/lines.py
@@ -68,15 +68,16 @@ from setproctitle import setproctitle
 
 from zipline.test_algorithms import TestAlgorithm
 from zipline.finance.trading import SIMULATION_STYLE
-from zipline.utils.log_utils import ZeroMQLogHandler
+from zipline.utils.log_utils import ZeroMQLogHandler, stdout_only_pipe
 from zipline.utils import factory
 
-from zipline.gens.composites import (
-    date_sorted_sources,
-    sequential_transforms
-)
+from zipline.test_algorithms import TestAlgorithm
+
+from zipline.gens.composites import  \
+    date_sorted_sources, merged_transforms, sequential_transforms
+from zipline.gens.transform import Passthrough, StatefulTransform
 from zipline.gens.tradesimulation import TradeSimulationClient as tsc
-from logbook import Logger
+from logbook import Logger, NestedSetup, Processor
 
 import zipline.protocol as zp
 
@@ -171,8 +172,6 @@ class SimulatedTrading(object):
 
     def close(self):
         log.info("Closing Simulation: {id}".format(id=self.sim_id))
-        if self.results_socket:
-            self.results_socket.close()
         if self.proc and self.send_sighup:
             ppid = os.getppid()
             if self.success:
@@ -245,6 +244,12 @@ class SimulatedTrading(object):
         else:
             return []
 
+    def __iter__(self):
+        return self
+
+    def next(self):
+        return self.gen.next()
+
     @staticmethod
     def create_test_zipline(**config):
         """
@@ -319,13 +324,11 @@ class SimulatedTrading(object):
             trade_source = config['trade_source']
         else:
             trade_source = factory.create_daily_trade_source(
-                sid_list,
+                sids,
                 trade_count,
-                trading_environment,
-                concurrent=concurrent_trades
+                trading_environment
             )
 
-
         #-------------------
         # Transforms
         #-------------------