diff --git a/tests/resources/example_data.tar.gz b/tests/resources/example_data.tar.gz index 23e13495..443387c2 100644 Binary files a/tests/resources/example_data.tar.gz and b/tests/resources/example_data.tar.gz differ diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py index e314d975..0968df6f 100644 --- a/tests/test_algorithm.py +++ b/tests/test_algorithm.py @@ -163,7 +163,7 @@ from zipline.test_algorithms import ( no_handle_data, ) from zipline.utils.api_support import ZiplineAPI, set_algo_instance -from zipline.utils.calendars import get_calendar +from zipline.utils.calendars import get_calendar, register_calendar from zipline.utils.context_tricks import CallbackManager from zipline.utils.control_flow import nullctx import zipline.utils.events @@ -1453,6 +1453,8 @@ class TestAlgoScript(WithLogger, @classmethod def make_equity_info(cls): + register_calendar("TEST", get_calendar("NYSE"), force=True) + data = make_simple_equity_info( cls.sids, cls.START_DATE, @@ -1787,6 +1789,7 @@ def handle_data(context, data): Test that api methods on the data object can be called with positional arguments. """ + params = SimulationParameters( start_session=pd.Timestamp("2006-01-10", tz='UTC'), end_session=pd.Timestamp("2006-01-11", tz='UTC'), diff --git a/tests/test_assets.py b/tests/test_assets.py index a92d577d..f1ba7991 100644 --- a/tests/test_assets.py +++ b/tests/test_assets.py @@ -431,6 +431,7 @@ class AssetFinderTestCase(WithTradingCalendar, ZiplineTestCase): 'symbol': 'TEST.%d' % sid, 'start_date': as_of.value, 'end_date': as_of.value, + 'exchange': uuid.uuid4().hex } for sid in sids ] @@ -480,8 +481,8 @@ class AssetFinderTestCase(WithTradingCalendar, ZiplineTestCase): def test_lookup_symbol_fuzzy(self): metadata = pd.DataFrame.from_records([ {'symbol': 'PRTY_HRD', 'exchange': "TEST"}, - {'symbol': 'BRKA', 'exchange': "TEST",}, - {'symbol': 'BRK_A', 'exchange': "TEST",}, + {'symbol': 'BRKA', 'exchange': "TEST"}, + {'symbol': 'BRK_A', 'exchange': "TEST"}, ]) self.write_assets(equities=metadata) finder = self.asset_finder @@ -524,11 +525,13 @@ class AssetFinderTestCase(WithTradingCalendar, ZiplineTestCase): 'symbol': 'A', 'start_date': T('2014-01-01'), 'end_date': T('2014-01-05'), + 'exchange': "TEST", }, { 'symbol': 'B', 'start_date': T('2014-01-06'), 'end_date': T('2014-01-10'), + 'exchange': "TEST", }, # sid 1 @@ -536,11 +539,13 @@ class AssetFinderTestCase(WithTradingCalendar, ZiplineTestCase): 'symbol': 'C', 'start_date': T('2014-01-01'), 'end_date': T('2014-01-05'), + 'exchange': "TEST", }, { 'symbol': 'A', # claiming the unused symbol 'A' 'start_date': T('2014-01-06'), 'end_date': T('2014-01-10'), + 'exchange': "TEST", }, ], index=[0, 0, 1, 1],