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))