test_datastructures.py

from hypothesis import given, strategies as st
from mpu.datastructures import Interval


@given(st.lists(st.integers(), min_size=4, max_size=4).map(sorted))
def test_interval_issubset(integer_list):
    a, b, c, d = integer_list
    assert Interval(b, c).issubset(Interval(a, d))


@given(st.lists(st.integers(), min_size=4, max_size=4).map(sorted))
def test_interval_issubset_not(integer_list):
    a, b, c, d = integer_list
    assert not Interval(a, b).issubset(Interval(c, d)) or c <= a <= b <= d
    assert not Interval(a, c).issubset(Interval(b, d)) or b <= a <= c <= d
    assert not Interval(c, d).issubset(Interval(a, b)) or a <= c <= d <= b
    assert not Interval(b, d).issubset(Interval(a, c)) or a <= b <= d <= c