def test_get_right_is_leaf():
gt = Segment(
[TaggedValue(2, "N"),
TaggedValue(1, "L"),
TaggedValue(2, "L")])
i = 1
with pytest.raises(AssertionError):
gt.get_right(i)
def test_get_right_has_critical():
gt = Segment(
[TaggedValue(2, "N"),
TaggedValue(1, "C"),
TaggedValue(2, "L")])
i = 0
with pytest.raises(AssertionError):
gt.get_right(i)
def test_get_right_direct():
gt = Segment(
[TaggedValue(2, "N"),
TaggedValue(1, "L"),
TaggedValue(2, "L")])
i = 0
res = gt.get_right(i)
exp = TaggedValue(2, "L")
assert res == exp