def test_cancelled_orders_in_spread_calc(self): book = OrderBook() limit_buy_order = Order(ID_GENERATOR, 1, 195, "LIMIT", 100) limit_sell_order = Order(ID_GENERATOR, 2, 200, "LIMIT", -50) book.add_order(limit_sell_order) book.add_order(limit_buy_order) self.assertEqual(book.current_spread(), 5) limit_sell_order.cancel() self.assertEqual(book.current_spread(), 0)
def test_change_to_cancel(self): order = Order(ID_GENERATOR, AGENT_ID, PRICE, "LIMIT", -100) self.assertEqual(order.order_state, "ACTIVE") order.cancel() self.assertEqual(order.order_state, "CANCELLED")