def test_fast_get_loc_ffilled(self): dts = pd.to_datetime([ '2014-01-02', '2014-01-03', # Skip 2014-01-04 '2014-01-05', '2014-01-06', ]) for dt in pd.date_range('2014-01-02', '2014-01-08'): result = zp_fixtures.fast_get_loc_ffilled(dts.values, dt.asm8) expected = dts.get_loc(dt, method='ffill') assert_equal(result, expected) with self.assertRaises(KeyError): dts.get_loc(pd.Timestamp('2014-01-01'), method='ffill') with self.assertRaises(KeyError): zp_fixtures.fast_get_loc_ffilled(dts, pd.Timestamp('2014-01-01'))
def test_fast_get_loc_ffilled(self): dts = pd.to_datetime([ "2014-01-02", "2014-01-03", # Skip 2014-01-04 "2014-01-05", "2014-01-06", ]) for dt in pd.date_range("2014-01-02", "2014-01-08"): result = zp_fixtures.fast_get_loc_ffilled(dts.values, dt.asm8) expected = dts.get_loc(dt, method="ffill") assert_equal(result, expected) with self.assertRaises(KeyError): dts.get_loc(pd.Timestamp("2014-01-01"), method="ffill") with self.assertRaises(KeyError): zp_fixtures.fast_get_loc_ffilled(dts, pd.Timestamp("2014-01-01"))