mirror of
https://github.com/wassname/catalyst.git
synced 2026-09-09 11:19:23 +08:00
PERF: Use scalar lookups for TradingCalendar.schedule
When retrieving the open and close for a given session, we only care about the scalar values, so using DataFrame.at instead of DataFrame.loc is significantly faster.
This commit is contained in:
@@ -452,8 +452,10 @@ class ExchangeCalendarTestBase(object):
|
||||
(2, 1),
|
||||
])
|
||||
def test_minute_index_to_session_labels(self, interval, offset):
|
||||
minutes = self.calendar.minutes_for_sessions_in_range('2011-01-04',
|
||||
'2011-04-04')
|
||||
minutes = self.calendar.minutes_for_sessions_in_range(
|
||||
pd.Timestamp('2011-01-04', tz='UTC'),
|
||||
pd.Timestamp('2011-04-04', tz='UTC'),
|
||||
)
|
||||
minutes = minutes[range(offset, len(minutes), interval)]
|
||||
|
||||
np.testing.assert_array_equal(
|
||||
|
||||
Reference in New Issue
Block a user