class StepsTests(TestCase): def setUp(self): self.dr = DateRange( start=datetime(1970, 1, 1), end=datetime(1970, 2, 1) ) def test_has_right_amount_for_month(self): 'returns 32 for day (since it includes the end)' steps = self.dr.steps(timedelta(days=1)) self.assertEqual(32, len(list(steps))) def test_steps_default(self): 'returns the right amount of steps by default' steps = self.dr.steps() self.assertEqual(32, len(list(steps))) def test_repr(self): 'repr should be correct' self.assertEqual( '<DateRange: datetime.datetime(1970, 1, 1, 0, 0) to datetime.datetime(1970, 1, 2, 0, 0)>', repr(DateRange( datetime(1970, 1, 1), datetime(1970, 1, 2) )) )
def setUp(self): self.dr = DateRange( start=datetime(1970, 1, 1), end=datetime(1970, 2, 1) )