def test_wrong_way_interval(self):
test1 = list(date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 6), interval=datetime.timedelta(days=-1)))
expected1 = []
self.assertEqual(test1, expected1)
test2 = list(date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 4), interval=datetime.timedelta(days=1)))
expected2 = []
self.assertEqual(test2, expected2)
def test_non_hour_intervals(self):
test1 = list(date_range(datetime.datetime(2013, 4, 5, 15),
datetime.datetime(2013, 4, 5, 17), interval=datetime.timedelta(hours=1)))
expected1 = [datetime.datetime(2013, 4, 5, 15, 0),
datetime.datetime(2013, 4, 5, 16, 0)]
self.assertEqual(test1, expected1)
test2 = list(date_range(datetime.datetime(2013, 4, 5, 15, 5),
datetime.datetime(2013, 4, 5, 15, 2), interval=datetime.timedelta(
minutes=-1)))
expected2 = [datetime.datetime(2013, 4, 5, 15, 5), datetime.datetime(
2013, 4, 5, 15, 4), datetime.datetime(2013, 4, 5, 15, 3)]
self.assertEqual(test2, expected2)
def test_zero_interval(self):
self.assertRaises(ValueError, list, date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 6), interval=datetime.timedelta(days=0)))
self.assertRaises(ValueError, list, date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 6), interval=datetime.timedelta(hours=0)))
def test_interval_larger_than_diff(self):
test = list(date_range(datetime.datetime(2013, 4, 5, 15, 5, 20),
datetime.datetime(2013, 4, 5, 15, 5, 10), interval=datetime.timedelta(
hours=-1)))
expected = [datetime.datetime(2013, 4, 5, 15, 5, 20)]
self.assertEqual(test, expected)
def test_no_hit_end(self):
test = list(date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 10), interval=datetime.timedelta(days=2)))
expected = [datetime.date(2013, 4, 5), datetime.date(2013, 4, 7),
datetime.date(2013, 4, 9)]
self.assertEqual(test, expected)
def test_negative_interval(self):
expected = [datetime.date(2013, 4, 5), datetime.date(2013, 4, 4)]
test = list(date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 3), interval=datetime.timedelta(days=-1)))
self.assertEqual(test, expected)
def test_sanity(self):
expected = [datetime.date(2013, 4, 5)]
test = list(date_range(datetime.date(2013, 4, 5),
datetime.date(2013, 4, 6)))
self.assertEqual(test, expected)
