BUG: Remove sid entry from open_orders when there are none.

TST: updated tests for new open_orders behavior
This commit is contained in:
Dale Jung
2015-02-17 09:32:49 -05:00
committed by Eddie Hebert
parent 760bbced73
commit b3bc7e166d
2 changed files with 10 additions and 7 deletions
+2 -4
View File
@@ -307,11 +307,9 @@ class FinanceTestCase(TestCase):
cumulative_pos = tracker.cumulative_performance.positions[sid]
self.assertEqual(total_volume, cumulative_pos.amount)
# the open orders should now be empty
# the open orders should not contain sid.
oo = blotter.open_orders
self.assertTrue(sid in oo)
order_list = oo[sid]
self.assertEqual(0, len(order_list))
self.assertNotIn(sid, oo, "Entry is removed when no open orders")
def test_blotter_processes_splits(self):
sim_params = factory.create_simulation_parameters()
+8 -3
View File
@@ -210,10 +210,15 @@ class Blotter(object):
yield txn, order
# update the open orders for the trade_event's sid
self.open_orders[trade_event.sid] = \
updated_orders = \
[order for order
in self.open_orders[trade_event.sid]
if order.open]
in self.open_orders[trade_event.sid]
if order.open]
if updated_orders:
self.open_orders[trade_event.sid] = updated_orders
else:
del self.open_orders[trade_event.sid]
def process_transactions(self, trade_event, current_orders):
for order, txn in self.transact(trade_event, current_orders):