def test_one_entry_next_date(self): testdate = util.get_date_string(date.today()) schedule_lines = [ '{date} lightning energy'.format(date=testdate), ' ;; schedule ; monthly', ' blah blah blah', ] schedule_thing = ScheduleThing(schedule_lines) schedule_thing.get_scheduled_entries() expected = date.today() + relativedelta(months=1) actual = schedule_thing.thing_date self.assertEqual(expected, actual)
def test_one_entry_count(self): testdate = util.get_date_string(date.today()) schedule_lines = [ '{date} lightning energy'.format(date=testdate), ' ;; schedule ; monthly', ' blah blah blah', ] schedule_thing = ScheduleThing(schedule_lines) expected = 1 actual = len(schedule_thing.get_scheduled_entries()) self.assertEqual(expected, actual)