def __init__(self, start, end): start, end = calcs.get_start_end(start, end) self.start = start self.end = end if end < start: self.end = start self.start = end assert self.start <= self.end
def test_get_start_end(self): end = datetime.datetime.now() start = end - datetime.timedelta(days=1) new_start, new_end = calcs.get_start_end(start, end) self.assertEqual(start, new_start) self.assertEqual(end, new_end) self.assertTrue(new_start < new_end) new_start, new_end = calcs.get_start_end(start, end.date()) self.assertTrue(new_start < new_end) self.assertEqual(start.date(), new_start) self.assertEqual(end.date(), new_end) new_start, new_end = calcs.get_start_end(start.date(), end) self.assertTrue(new_start < new_end) self.assertEqual(start.date(), new_start) self.assertEqual(end.date(), new_end)