From c911dca4d248e6c250340d4a6d1b5bcb10adcb0d Mon Sep 17 00:00:00 2001 From: fawce Date: Thu, 16 Aug 2012 13:01:22 -0400 Subject: [PATCH] added a kwarg to run in blocking --- zipline/utils/test_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zipline/utils/test_utils.py b/zipline/utils/test_utils.py index 2ebbae73..99175c63 100644 --- a/zipline/utils/test_utils.py +++ b/zipline/utils/test_utils.py @@ -117,8 +117,8 @@ def drain_receiver(receiver, count=None): return output, transaction_count -def assert_single_position(test, zipline): - output, transaction_count = drain_zipline(test, zipline) +def assert_single_position(test, zipline, blocking=False): + output, transaction_count = drain_zipline(test, zipline, p_blocking=blocking) test.assertEqual( test.zipline_test_config['order_count'],