From d88b1756726f790dd0c932b74c997a081b88e7fc Mon Sep 17 00:00:00 2001 From: fawce Date: Fri, 10 Aug 2012 11:55:30 -0400 Subject: [PATCH] SimulatedTrading is now a generator. --- zipline/lines.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/zipline/lines.py b/zipline/lines.py index 9bc8b3ac..2a59190f 100644 --- a/zipline/lines.py +++ b/zipline/lines.py @@ -244,10 +244,17 @@ class SimulatedTrading(object): else: return [] + def __iter__(self): + return self + + def next(self): + return self.gen.next() + @staticmethod def create_test_zipline(**config): """ - :param config: A configuration object that is a dict with: + :param config: A configuration object that is a dict with + (all optional): - environment - a \ :py:class:`zipline.finance.trading.TradingEnvironment` @@ -269,7 +276,7 @@ class SimulatedTrading(object): of StatefulTransform objects. """ assert isinstance(config, dict) - sid = config['sid'] + sid = config.get('sid', 133) #-------------------- # Trading Environment