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_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_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(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.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))
