def test_add(self):
int1, int2, int3 = Interval(0, 3), Interval(1, 5), Interval(0, 5)
self.assertEqual(int1 + int2, IntervalList([int3]))
int4 = Interval(-1, 0)
self.assertEqual(int2 + int4, IntervalList([int2, int4]))
ints = IntervalList([int4, int2])
int5 = Interval(6, 7)
self.assertEqual(ints + int5, IntervalList([int2, int4, int5]))
int5 = Interval(0, 1)
self.assertEqual(ints + int5, IntervalList([Interval(-1, 5)]))
int5 = Interval(3, 7)
self.assertEqual(ints + int5, IntervalList([int4, Interval(1, 7)]))
def test_contain(self):
int1 = Interval(1, 3)
int2 = Interval(2, 3)
self.assertTrue(int2 in int1)
int1 = Interval(1.5, 2.5)
int2 = Interval(1.2, 2.7)
self.assertTrue(int1 in int2)
int1 = Interval(datetime(2016, 5, 18, 20), datetime(2016, 5, 18, 22))
int2 = Interval(datetime(2016, 5, 18, 18), datetime(2016, 5, 18, 23))
self.assertTrue(int1 in int2)
def test_sub(self):
int1 = Interval(1, 5)
int2 = Interval(3, 4)
ints1 = IntervalList([Interval(1, 3), Interval(4, 5)])
self.assertEqual(int2 - int1, IntervalList([]))
self.assertEqual(int1 - int2, ints1)
int1 = Interval(1, 3)
int2 = Interval(4, 6)
int3 = Interval(2, 5)
ints1 = IntervalList([int1, int2])
ints2 = IntervalList([Interval(1, 2), Interval(5, 6)])
self.assertTrue(ints1 - int3 == ints2)
int1 = Interval(1, 3)
int2 = Interval(3, 5)
self.assertEqual(int1 - int2, IntervalList([Interval(1, 3)]))
int1 = Interval(1, 3)
int2 = Interval(5, 8)
ints1 = IntervalList([int1, int2])
int3 = Interval(0, 2)
ints2 = IntervalList([Interval(0, 1)])
self.assertEqual(int3 - ints1, ints2)
self.assertEqual(int1 - None, IntervalList([int1]))
int1 = Interval(1, 5)
int2 = Interval(1, 3)
self.assertEqual((int1 - int2)[0], Interval(3, 5))
int1 = Interval(datetime(2016, 5, 18, 20), datetime(2016, 5, 18, 23))
int2 = Interval(datetime(2016, 5, 18, 21), datetime(2016, 5, 18, 22))
int3 = Interval(datetime(2016, 5, 18, 20), datetime(2016, 5, 18, 21))
int4 = Interval(datetime(2016, 5, 18, 22), datetime(2016, 5, 18, 23))
intlist = IntervalList([int4, int3])
self.assertEqual(int1 - int2, intlist)
def test_overlap(self):
int1 = Interval(1, 5)
int2 = Interval(0, 2)
int3 = Interval(3, 4)
int4 = Interval(4, 6)
self.assertTrue(int1.overlap(int2))
self.assertTrue(int2.overlap(int1))
self.assertTrue(int3.overlap(int1))
self.assertTrue(int1.overlap(int3))
self.assertTrue(int4.overlap(int1))
self.assertTrue(int1.overlap(int4))
int1 = Interval(datetime(2016, 5, 18, 17), datetime(2016, 5, 18, 19))
int2 = Interval(datetime(2016, 5, 18, 16), datetime(2016, 5, 18, 18))
self.assertTrue(int1.overlap(int2))
def test_intersect(self):
iv = Interval(1, 10)
ivl = IntervalList([Interval(0, 3), Interval(5, 6)])
res = IntervalList([Interval(1, 3), Interval(5, 6)])
self.assertEqual(res, iv.intersection(ivl))
def test_overlap(self):
int1 = Interval(1, 5)
int2 = Interval(0, 2)
int3 = Interval(3, 4)
int4 = Interval(4, 6)
self.assertTrue(int1.overlap(int2))
self.assertTrue(int2.overlap(int1))
self.assertTrue(int3.overlap(int1))
self.assertTrue(int1.overlap(int3))
self.assertTrue(int4.overlap(int1))
self.assertTrue(int1.overlap(int4))
int1 = Interval(datetime(2016, 5, 18, 17), datetime(2016, 5, 18, 19))
int2 = Interval(datetime(2016, 5, 18, 16), datetime(2016, 5, 18, 18))
self.assertTrue(int1.overlap(int2))
def test_intersect(self):
iv = Interval(1, 10)
ivl = IntervalList([Interval(0, 3), Interval(5, 6)])
res = IntervalList([Interval(1, 3), Interval(5, 6)])
self.assertEqual(res, iv.intersection(ivl))
