From 0f4964fdfaccafd3344d8f64ffae2217b57c25c8 Mon Sep 17 00:00:00 2001 From: Scott Sanderson Date: Mon, 15 Aug 2016 22:56:45 -0400 Subject: [PATCH] MAINT: Clean up WithTradingSessions. - Add WithDefaultDateBounds, since we use alias('START_DATE') and alias ('END_DATE'). - Fix copypasta in assertion. --- zipline/testing/fixtures.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/zipline/testing/fixtures.py b/zipline/testing/fixtures.py index b2d440c3..88af7f1d 100644 --- a/zipline/testing/fixtures.py +++ b/zipline/testing/fixtures.py @@ -527,7 +527,7 @@ class WithSimParams(WithTradingEnvironment): cls.sim_params = cls.make_simparams() -class WithTradingSessions(WithTradingCalendars): +class WithTradingSessions(WithTradingCalendars, WithDefaultDateBounds): """ ZiplineTestCase mixin providing cls.trading_days, cls.all_trading_sessions as a class-level fixture. @@ -1212,7 +1212,7 @@ class WithSeededRandomPipelineEngine(WithTradingSessions, WithAssetFinder): if start_date not in self.trading_days: raise AssertionError("Start date not in calendar: %s" % start_date) if end_date not in self.trading_days: - raise AssertionError("Start date not in calendar: %s" % start_date) + raise AssertionError("End date not in calendar: %s" % end_date) return self.seeded_random_engine.run_pipeline( pipeline, start_date,