def test_precision_finer_than_offset(self): # GH 9907 result1 = DatetimeIndex(start='2015-04-15 00:00:03', end='2016-04-22 00:00:00', freq='Q') result2 = DatetimeIndex(start='2015-04-15 00:00:03', end='2015-06-22 00:00:04', freq='W') expected1_list = [ '2015-06-30 00:00:03', '2015-09-30 00:00:03', '2015-12-31 00:00:03', '2016-03-31 00:00:03' ] expected2_list = [ '2015-04-19 00:00:03', '2015-04-26 00:00:03', '2015-05-03 00:00:03', '2015-05-10 00:00:03', '2015-05-17 00:00:03', '2015-05-24 00:00:03', '2015-05-31 00:00:03', '2015-06-07 00:00:03', '2015-06-14 00:00:03', '2015-06-21 00:00:03' ] expected1 = DatetimeIndex(expected1_list, dtype='datetime64[ns]', freq='Q-DEC', tz=None) expected2 = DatetimeIndex(expected2_list, dtype='datetime64[ns]', freq='W-SUN', tz=None) self.assertTrue(result1.equals(expected1)) self.assertTrue(result2.equals(expected2))
def test_cdaterange_weekmask_and_holidays(self): rng = cdate_range('2013-05-01', periods=3, weekmask='Sun Mon Tue Wed Thu', holidays=['2013-05-01']) xp = DatetimeIndex(['2013-05-02', '2013-05-05', '2013-05-06']) self.assert_(xp.equals(rng))
def test_precision_finer_than_offset(self): # GH 9907 result1 = DatetimeIndex(start='2015-04-15 00:00:03', end='2016-04-22 00:00:00', freq='Q') result2 = DatetimeIndex(start='2015-04-15 00:00:03', end='2015-06-22 00:00:04', freq='W') expected1_list = ['2015-06-30 00:00:03', '2015-09-30 00:00:03', '2015-12-31 00:00:03', '2016-03-31 00:00:03'] expected2_list = ['2015-04-19 00:00:03', '2015-04-26 00:00:03', '2015-05-03 00:00:03', '2015-05-10 00:00:03', '2015-05-17 00:00:03', '2015-05-24 00:00:03', '2015-05-31 00:00:03', '2015-06-07 00:00:03', '2015-06-14 00:00:03', '2015-06-21 00:00:03'] expected1 = DatetimeIndex(expected1_list, dtype='datetime64[ns]', freq='Q-DEC', tz=None) expected2 = DatetimeIndex(expected2_list, dtype='datetime64[ns]', freq='W-SUN', tz=None) self.assertTrue(result1.equals(expected1)) self.assertTrue(result2.equals(expected2))
def test_cdaterange_weekmask_and_holidays(self): rng = cdate_range('2013-05-01', periods=3, weekmask='Sun Mon Tue Wed Thu', holidays=['2013-05-01']) xp = DatetimeIndex(['2013-05-02', '2013-05-05', '2013-05-06']) self.assertTrue(xp.equals(rng))
def test_cdaterange_holidays(self): rng = cdate_range('2013-05-01', periods=3, holidays=['2013-05-01']) xp = DatetimeIndex(['2013-05-02', '2013-05-03', '2013-05-06']) self.assertTrue(xp.equals(rng))
def test_cdaterange(self): rng = cdate_range('2013-05-01', periods=3) xp = DatetimeIndex(['2013-05-01', '2013-05-02', '2013-05-03']) self.assertTrue(xp.equals(rng))
def test_cdaterange_weekmask(self): rng = cdate_range('2013-05-01', periods=3, weekmask='Sun Mon Tue Wed Thu') xp = DatetimeIndex(['2013-05-01', '2013-05-02', '2013-05-05']) self.assertTrue(xp.equals(rng))
def test_cdaterange_weekmask(self): rng = cdate_range('2013-05-01', periods=3, weekmask='Sun Mon Tue Wed Thu') xp = DatetimeIndex(['2013-05-01', '2013-05-02', '2013-05-05']) self.assert_(xp.equals(rng))
def test_cdaterange_weekmask_and_holidays(self): rng = cdate_range("2013-05-01", periods=3, weekmask="Sun Mon Tue Wed Thu", holidays=["2013-05-01"]) xp = DatetimeIndex(["2013-05-02", "2013-05-05", "2013-05-06"]) self.assert_(xp.equals(rng))
def test_cdaterange_holidays(self): rng = cdate_range("2013-05-01", periods=3, holidays=["2013-05-01"]) xp = DatetimeIndex(["2013-05-02", "2013-05-03", "2013-05-06"]) self.assert_(xp.equals(rng))
def test_cdaterange(self): rng = cdate_range("2013-05-01", periods=3) xp = DatetimeIndex(["2013-05-01", "2013-05-02", "2013-05-03"]) self.assert_(xp.equals(rng))