Function test_occurrences

tests/tests/test_timeline.py:70–122

(make_timeline)

@pytest.mark.timeout(1)
def test_occurrences(make_timeline):
    timeline, initial_history = make_timeline()

    mark1 = timeline.mark("dum")
    mark2 = timeline.mark("dee")
    mark3 = timeline.mark("dum")

    assert mark1 == Mark("dum")
    assert mark1.id == "dum"
    assert mark2 == Mark("dee")
    assert mark2.id == "dee"
    assert mark3 == Mark("dum")
    assert mark3.id == "dum"

    with timeline.frozen():
        assert timeline.history() == initial_history + [
            some.object.same_as(mark1),
            some.object.same_as(mark2),
            some.object.same_as(mark3),
        ]
        timeline.finalize()

    assert timeline.all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark1),
        some.object.same_as(mark3),
    )
    assert timeline.all_occurrences_of(Mark("dee")) == (some.object.same_as(mark2),)

    assert timeline[:].all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark1),
        some.object.same_as(mark3),
    )

    # Lower boundary is inclusive.
    assert timeline[mark1:].all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark1),
        some.object.same_as(mark3),
    )
    assert timeline[mark2:].all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark3),
    )
    assert timeline[mark3:].all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark3),
    )

    # Upper boundary is exclusive.
    assert timeline[:mark1].all_occurrences_of(Mark("dum")) == ()
    assert timeline[:mark2].all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark1),
    )
    assert timeline[:mark3].all_occurrences_of(Mark("dum")) == (
        some.object.same_as(mark1),
    )


def expectation_of(message):