def testYearly(self): datestring = 'Yearly' start_date = datetime.datetime.now() hydro = Hydro(start_date, datestring) hydro.tokenize() self.assertEqual(len(hydro.tokens['all'].tokens), 1)
def testLimit(self): datestring = "Daily for 100 times" start_date = datetime.datetime.now() hydro = Hydro(start_date, datestring) hydro.tokenize() self.assertEqual(len(hydro.all_tokens.tokens), 4) dates = hydro.generate_dates() self.assertEqual(len(dates), 100)
def testDaily(self): datestring = 'Daily' start_date = datetime.datetime.now() hydro = Hydro(start_date, datestring) hydro.tokenize() self.assertEqual(len(hydro.tokens['all'].tokens), 1) dates = hydro.generate_dates() self.assertEqual(len(dates), 365)
def testMaximium(self): """ We're only allowed to return 365 events for now """ datestring = "Daily for 1201 times" start_date = datetime.datetime.now() hydro = Hydro(start_date, datestring) hydro.tokenize() self.assertEqual(len(hydro.all_tokens.tokens), 4) dates = hydro.generate_dates() self.assertEqual(len(dates), 365)