TST: Group test trades for transaction simulator by date.

Group by date to prepare for other values to be added per dt.
This commit is contained in:
Eddie Hebert
2013-04-12 11:13:23 -04:00
parent 3883f79ece
commit 48a691e9a7
+10 -5
View File
@@ -16,6 +16,9 @@
"""
Tests for the zipline.finance package
"""
import itertools
import operator
import pytz
from unittest import TestCase
@@ -334,12 +337,14 @@ class FinanceTestCase(TestCase):
# this approximates the loop inside TradingSimulationClient
transactions = []
for trade in generated_trades:
trade_sim.update(trade)
if trade.TRANSACTION:
transactions.append(trade.TRANSACTION)
for dt, trades in itertools.groupby(generated_trades,
operator.attrgetter('dt')):
for trade in trades:
trade_sim.update(trade)
if trade.TRANSACTION:
transactions.append(trade.TRANSACTION)
tracker.process_event(trade)
tracker.process_event(trade)
if complete_fill:
self.assertEqual(len(transactions), len(order_list))