def test_summary(self):
self.rng.summary()
self.rng[2:2].summary()
try:
bdate_range('1/1/2005', '1/1/2009', tz=pytz.utc).summary()
except Exception:
pass
def test_with_tz(self):
_skip_if_no_pytz()
tz = pytz.timezone('US/Central')
# just want it to work
start = datetime(2011, 3, 12, tzinfo=pytz.utc)
dr = bdate_range(start, periods=50, freq=datetools.Hour())
self.assert_(dr.tz is not None)
self.assert_(dr.tz is start.tzinfo)
# DateRange with naive datetimes
dr = bdate_range('1/1/2005', '1/1/2009', tz=pytz.utc)
dr = bdate_range('1/1/2005', '1/1/2009', tz=tz)
# normalized
central = dr.tz_normalize(tz)
self.assert_(central.tz is tz)
self.assert_(central[0].tz is tz)
# datetimes with tzinfo set
dr = bdate_range(datetime(2005, 1, 1, tzinfo=pytz.utc),
'1/1/2009', tz=pytz.utc)
self.assertRaises(Exception, bdate_range,
datetime(2005, 1, 1, tzinfo=pytz.utc),
'1/1/2009', tz=tz)
def test_daterange_bug_456(self):
# GH #456
rng1 = bdate_range('12/5/2011', '12/5/2011')
rng2 = bdate_range('12/2/2011', '12/5/2011')
rng2.offset = datetools.BDay()
result = rng1.union(rng2)
self.assert_(isinstance(result, DatetimeIndex))
def test_misc(self):
end = datetime(2009, 5, 13)
dr = bdate_range(end=end, periods=20)
firstDate = end - 19 * datetools.bday
assert len(dr) == 20
assert dr[0] == firstDate
assert dr[-1] == end
def test_constructor(self):
rng = bdate_range(START, END, freq=datetools.bday)
rng = bdate_range(START, periods=20, freq=datetools.bday)
rng = bdate_range(end=START, periods=20, freq=datetools.bday)
def setUp(self):
self.rng = bdate_range(START, END)
def test_tz_localize(self):
_skip_if_no_pytz()
dr = bdate_range('1/1/2009', '1/1/2010')
dr_utc = bdate_range('1/1/2009', '1/1/2010', tz=pytz.utc)
localized = dr.tz_localize(pytz.utc)
self.assert_(np.array_equal(dr_utc, localized))
def test_intersection_bug(self):
# GH #771
a = bdate_range('11/30/2011','12/31/2011')
b = bdate_range('12/10/2011','12/20/2011')
result = a.intersection(b)
self.assert_(result.equals(b))
