diff --git a/zipline/test_algorithms.py b/zipline/test_algorithms.py index 055c8c32..fbb9d5a9 100644 --- a/zipline/test_algorithms.py +++ b/zipline/test_algorithms.py @@ -59,7 +59,8 @@ class TestAlgorithm(): at the close of a simulation. """ - def __init__(self, sid, amount, sid_filter=None): + def __init__(self, sid, amount, order_count, sid_filter=None): + self.count = order_count self.sid = sid self.amount = amount self.incr = 0