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