def test_contains():
now = datetime.now()
tp1 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=1))
tp2 = TimePeriod(now, now + timedelta(hours=1))
assert tp1.contains(tp2)
assert not tp2.contains(tp1)
# if datetime is contained in period
assert now in tp1
assert now in tp2
# __contains__ operator overloading
assert tp2 in tp1
assert tp1 not in tp2
with pytest.raises(TypeError):
assert 0 in tp1
def test_intersection_entire(self):
now = datetime.now()
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(minutes=10),
now + timedelta(minutes=70))
assert tp1.intersection(tp2).duration == timedelta(minutes=60)
assert tp2.intersection(tp1).duration == timedelta(minutes=60)
def test_contains():
now = datetime.now()
tp1 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=1))
tp2 = TimePeriod(now, now + timedelta(hours=1))
assert tp1.contains(tp2)
assert not tp2.contains(tp1)
# if datetime is contained in period
assert now in tp1
assert now in tp2
# __contains__ operator overloading
assert tp2 in tp1
assert tp1 not in tp2
with pytest.raises(ValueError):
assert 0 in tp1
def _get_event_period(event: Event) -> TimePeriod:
# TODO: Better parsing of event duration
start = event.timestamp
end = start + event.duration
return TimePeriod(start, end)
def _get_event_period(event: Event) -> TimePeriod:
start = event.timestamp
end = start + event.duration
return TimePeriod(start, end)
def test_overlaps():
now = datetime.now()
# If periods are just "touching", they should not count as overlap
tp1 = TimePeriod(now - timedelta(hours=1), now)
tp2 = TimePeriod(now, now + timedelta(hours=1))
assert not tp1.overlaps(tp2)
assert not tp2.overlaps(tp1)
# If outer contains inner, or vice versa, they overlap
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=2))
assert tp1.overlaps(tp2)
assert tp2.overlaps(tp1)
# If start/end is contained in the other event, they overlap
tp1 = TimePeriod(now, now + timedelta(hours=2))
tp2 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=1))
assert tp1.overlaps(tp2)
assert tp2.overlaps(tp1)
def test_intersection_none():
now = datetime.now()
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(hours=1), now)
assert tp1.intersection(tp2) is None
assert tp2.intersection(tp1) is None
def test_overlaps(self):
now = datetime.now()
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=1))
assert tp1.overlaps(tp2)
assert tp2.overlaps(tp1)
def test_contains(self):
now = datetime.now()
tp1 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=1))
tp2 = TimePeriod(now, now + timedelta(hours=1))
assert tp1.contains(tp2)
assert not tp2.contains(tp1)
def test_union():
now = datetime.now()
# adjacent but not overlapping
tp1 = TimePeriod(now - timedelta(hours=1), now)
tp2 = TimePeriod(now, now + timedelta(hours=1))
tp_union = tp1.union(tp2)
assert tp1 in tp_union
assert tp2 in tp_union
# union with self
tp_union = tp1.union(tp1)
assert tp1 == tp_union
# overlapping
tp1 = TimePeriod(now - timedelta(hours=1), now + timedelta(minutes=30))
tp2 = TimePeriod(now, now + timedelta(hours=1))
tp_union = tp1.union(tp2)
assert tp1 in tp_union
assert tp2 in tp_union
# tp2 contained in tp1
tp1 = TimePeriod(now - timedelta(minutes=30), now + timedelta(minutes=30))
tp2 = TimePeriod(now, now + timedelta(minutes=10))
tp_union = tp1.union(tp2)
assert tp1 in tp_union
assert tp2 in tp_union
assert tp1 == tp_union
def test_adjacent():
now = datetime.now()
tp1 = TimePeriod(now - timedelta(hours=1), now)
tp2 = TimePeriod(now, now + timedelta(hours=1))
assert tp1.adjacent(tp2)
assert tp2.adjacent(tp1)
def test_intersection_start():
now = datetime.now()
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(minutes=10), now + timedelta(minutes=50))
assert tp1.intersection(tp2).duration == timedelta(minutes=50)
assert tp2.intersection(tp1).duration == timedelta(minutes=50)
def test_overlaps():
now = datetime.now()
# If periods are just "touching", they should not count as overlap
tp1 = TimePeriod(now - timedelta(hours=1), now)
tp2 = TimePeriod(now, now + timedelta(hours=1))
assert not tp1.overlaps(tp2)
assert not tp2.overlaps(tp1)
# If outer contains inner, or vice versa, they overlap
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=2))
assert tp1.overlaps(tp2)
assert tp2.overlaps(tp1)
# If start/end is contained in the other event, they overlap
tp1 = TimePeriod(now, now + timedelta(hours=2))
tp2 = TimePeriod(now - timedelta(hours=1), now + timedelta(hours=1))
assert tp1.overlaps(tp2)
assert tp2.overlaps(tp1)
def test_intersection_none():
now = datetime.now()
tp1 = TimePeriod(now, now + timedelta(hours=1))
tp2 = TimePeriod(now - timedelta(hours=1), now)
assert tp1.intersection(tp2) is None
assert tp2.intersection(tp1) is None
