def test_multiple_compress(self):
intervals = [(datetime.date(2014, 2, 3), datetime.date(2014, 3, 10)),
(datetime.date(2014, 3, 20), datetime.date(2014, 7, 1))]
expected = [(datetime.date(2014, 2, 3), datetime.date(2014, 7, 1))]
for num_days in 11, 12, 100, 2000:
compressed = lifetimes.compress_intervals_days(intervals, num_days)
self.assertEqual(expected, compressed)
def test_multiple_no_compress(self):
intervals = [(datetime.date(2014, 2, 3), datetime.date(2014, 3, 10)),
(datetime.date(2014, 3, 20), datetime.date(2014, 7, 1))]
for num_days in 0, 1, 9, 10:
compressed = lifetimes.compress_intervals_days(intervals, num_days)
self.assertEqual(intervals, compressed)
def test_single_open(self):
intervals = [(datetime.date(2014, 2, 3), None)]
compressed = lifetimes.compress_intervals_days(intervals, 10)
self.assertEqual(intervals, compressed)
compressed = lifetimes.compress_intervals_days(intervals, 100)
self.assertEqual(intervals, compressed)