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)
Example #2
0
    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)