diff --git a/tests/test_data_portal.py b/tests/test_data_portal.py index dc78a0b16d..e479fe2c3f 100644 --- a/tests/test_data_portal.py +++ b/tests/test_data_portal.py @@ -38,8 +38,7 @@ class DataPortalTestBase(WithDataPortal, - WithTradingSessions, - ZiplineTestCase): + WithTradingSessions): ASSET_FINDER_EQUITY_SIDS = (1, 2, 3) DIVIDEND_ASSET_SID = 3 @@ -578,11 +577,13 @@ def test_price_rounding(self, frequency, field): assert_equal(result, expected_result) -class TestDataPortal(DataPortalTestBase): +class TestDataPortal(DataPortalTestBase, + ZiplineTestCase): DATA_PORTAL_LAST_AVAILABLE_SESSION = None DATA_PORTAL_LAST_AVAILABLE_MINUTE = None -class TestDataPortalExplicitLastAvailable(DataPortalTestBase): +class TestDataPortalExplicitLastAvailable(DataPortalTestBase, + ZiplineTestCase): DATA_PORTAL_LAST_AVAILABLE_SESSION = alias('START_DATE') DATA_PORTAL_LAST_AVAILABLE_MINUTE = alias('END_DATE')