Merge pull request #1245 from quantopian/daily-bars-first-day-attr

BUG: Fixes reading and writing of daily bars first_trading_day attr
This commit is contained in:
Andrew Daniels
2016-06-02 15:09:12 -04:00
2 changed files with 8 additions and 4 deletions
+6
View File
@@ -186,6 +186,12 @@ class BcolzDailyBarTestCase(WithBcolzDailyBarReader, ZiplineTestCase):
DatetimeIndex(result.attrs['calendar'], tz='UTC'),
)
def test_read_first_trading_day(self):
self.assertEqual(
self.bcolz_daily_bar_reader.first_trading_day,
self.trading_days[0],
)
def _check_read_results(self, columns, assets, start_date, end_date):
results = self.bcolz_daily_bar_reader.load_raw_arrays(
columns,
+2 -4
View File
@@ -358,9 +358,7 @@ class BcolzDailyBarWriter(object):
)
full_table.attrs['first_trading_day'] = (
earliest_date // 1e6
if earliest_date is not None else
iNaT
earliest_date if earliest_date is not None else iNaT
)
full_table.attrs['first_row'] = first_row
full_table.attrs['last_row'] = last_row
@@ -507,7 +505,7 @@ class BcolzDailyBarReader(DailyBarReader):
try:
return Timestamp(
self._table.attrs['first_trading_day'],
unit='ms',
unit='s',
tz='UTC'
)
except KeyError: