wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-04 01:56:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

MAINT/TEST: Update default calendar smoketest.

Browse Source
This commit is contained in:
Scott Sanderson
2016-09-06 14:13:32 -04:00
parent a8a2cc1582
commit bd420f938b
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/calendars/test_trading_calendar.py
+8 -3
View File
@@ -39,8 +39,12 @@ from zipline.utils.calendars import(
deregister_calendar,
get_calendar,
)
from zipline.utils.calendars.calendar_utils import register_calendar_type, \
_default_calendar_factories
from zipline.utils.calendars.calendar_utils import (
_default_calendar_aliases,
_default_calendar_factories,
register_calendar_type,
)
from zipline.utils.calendars.trading_calendar import days_at_time, \
TradingCalendar
@@ -123,7 +127,8 @@ class CalendarRegistrationTestCase(TestCase):
class DefaultsTestCase(TestCase):
def test_default_calendars(self):
for name in concat(_default_calendar_factories):
for name in concat([_default_calendar_factories,
_default_calendar_aliases]):
self.assertIsNotNone(get_calendar(name),
"get_calendar(%r) returned None" % name)
zipline/utils/calendars/calendar_utils.py
+2 -2
View File
@@ -25,7 +25,7 @@ _default_calendar_factories = {
'TSX': TSXExchangeCalendar,
'us_futures': QuantopianUSFuturesCalendar,
}
_default_aliases = {
_default_calendar_aliases = {
'NASDAQ': 'NYSE',
'BATS': 'NYSE',
'CBOT': 'CME',
@@ -254,7 +254,7 @@ class TradingCalendarDispatcher(object):
global_calendar_dispatcher = TradingCalendarDispatcher(
calendars={},
calendar_factories=_default_calendar_factories,
aliases=_default_aliases,
aliases=_default_calendar_aliases,
)
get_calendar = global_calendar_dispatcher.get_calendar