mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-04 00:48:24 +08:00
MAINT: Use "difference" instead of "-" to make Pandas happy
This commit is contained in:
@@ -81,14 +81,14 @@ class TestTradingCalendar(TestCase):
|
||||
self.check_days(env_days, cal_days)
|
||||
|
||||
def check_days(self, env_days, cal_days):
|
||||
diff = env_days - cal_days
|
||||
diff = env_days.difference(cal_days)
|
||||
self.assertEqual(
|
||||
len(diff),
|
||||
0,
|
||||
"{diff} should be empty".format(diff=diff)
|
||||
)
|
||||
|
||||
diff2 = cal_days - env_days
|
||||
diff2 = cal_days.difference(env_days)
|
||||
self.assertEqual(
|
||||
len(diff2),
|
||||
0,
|
||||
|
||||
@@ -183,4 +183,4 @@ non_trading_day_index = pd.DatetimeIndex(sorted(non_trading_days))
|
||||
business_days = pd.DatetimeIndex(start=start, end=end,
|
||||
freq=pd.datetools.BDay())
|
||||
|
||||
trading_days = business_days - non_trading_day_index
|
||||
trading_days = business_days.difference(non_trading_day_index)
|
||||
|
||||
Reference in New Issue
Block a user