From 1ea68eb4f3072bd8b91e497bd2c4ec9e5b3971dc Mon Sep 17 00:00:00 2001 From: Stephen Diehl Date: Tue, 3 Jul 2012 11:39:00 -0400 Subject: [PATCH] Logging compat for test_transforms. --- tests/test_transforms.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_transforms.py b/tests/test_transforms.py index 0d267e60..04a434ea 100644 --- a/tests/test_transforms.py +++ b/tests/test_transforms.py @@ -2,6 +2,8 @@ from datetime import timedelta from collections import defaultdict from unittest2 import TestCase +from logbook.compat import LoggingHandler + import zipline.utils.factory as factory from zipline.finance.vwap import DailyVWAP, VWAPTransform from zipline.finance.returns import ReturnsFromPriorClose @@ -22,6 +24,11 @@ class ZiplineWithTransformsTestCase(TestCase): 'allocator':allocator, 'sid':133 } + self.log_handler = LoggingHandler() + self.log_handler.push_application() + + def tearDown(self): + self.log_handler.pop_application() def test_vwap_tnfm(self): zipline = SimulatedTrading.create_test_zipline(