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