def test_range_bug(self):
# GH #770
offset = datetools.DateOffset(months=3)
result = date_range("2011-1-1", "2012-1-31", freq=offset)
start = datetime(2011, 1, 1)
exp_values = [start + i * offset for i in range(5)]
self.assertTrue(np.array_equal(result, DatetimeIndex(exp_values)))
def test_catch_infinite_loop(self):
offset = datetools.DateOffset(minute=5)
# blow up, don't loop forever
self.assertRaises(Exception,
DateRange,
datetime(2011, 11, 11),
datetime(2011, 11, 12),
offset=offset)
def _generateCashflow(self):
monthly_offset = 12 / self.frequency
#offset = int(365.2425/self.frequency)
dr = pd.DateRange(
self.startdate,
self.maturitydate,
offset=pdt.DateOffset(months=monthly_offset)) #bday * offset)
cf = Cashflow(cf_times=dr,
cf_amounts=self.couponrate * 100 / self.frequency)
cf.cf[0] = -self.price
cf.cf[cf.cf.count() - 1] += 100.0
return cf
def test_cached_range_bug(self):
rng = date_range('2010-09-01 05:00:00', periods=50,
freq=datetools.DateOffset(hours=6))
self.assertEqual(len(rng), 50)
self.assertEqual(rng[0], datetime(2010, 9, 1, 5))
