From 88e0afc7a036dc514a53b32451d8a9947742ca90 Mon Sep 17 00:00:00 2001 From: Jonathan Kamens Date: Mon, 20 Aug 2012 16:55:37 -0400 Subject: [PATCH 1/2] Require zipline tests to end with a DONE packet --- zipline/utils/test_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/zipline/utils/test_utils.py b/zipline/utils/test_utils.py index 99175c63..ae804426 100644 --- a/zipline/utils/test_utils.py +++ b/zipline/utils/test_utils.py @@ -119,6 +119,7 @@ def drain_receiver(receiver, count=None): def assert_single_position(test, zipline, blocking=False): output, transaction_count = drain_zipline(test, zipline, p_blocking=blocking) + test.assertEqual(output[-1]['prefix'], 'DONE') test.assertEqual( test.zipline_test_config['order_count'], From a2376cb87adebc719a045888da2063395d956bc0 Mon Sep 17 00:00:00 2001 From: Jonathan Kamens Date: Mon, 20 Aug 2012 20:56:40 -0400 Subject: [PATCH 2/2] test_finance.py no longer needs drain_zipline --- tests/test_finance.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/test_finance.py b/tests/test_finance.py index 94a98da9..5562b671 100644 --- a/tests/test_finance.py +++ b/tests/test_finance.py @@ -20,7 +20,6 @@ from zipline.finance.performance import PerformanceTracker from zipline.utils.protocol_utils import ndict from zipline.finance.trading import TransactionSimulator from zipline.utils.test_utils import \ - drain_zipline, \ setup_logger, \ teardown_logger,\ assert_single_position