Function assert_event_order

tests/integration/utils.py:81–111

Verify that the actual event order matches the expected event order, accounting for sets in the expected order. Args: actual_event_order: the actual events recorded in the State. exp_event_order: the expected events recorded in the State, where some entries may be sets of events

(
    actual_event_order: list[str], exp_event_order: Sequence[str | set[str]]
)

def assert_event_order(
    actual_event_order: list[str], exp_event_order: Sequence[str | set[str]]
) -> None:
    """Verify that the actual event order matches the expected event order, accounting for sets in the expected order.

    Args:
        actual_event_order: the actual events recorded in the State.
        exp_event_order: the expected events recorded in the State, where some entries may be sets of events that can occur in any order.

    Raises:
        AssertionError: if the actual event order does not match the expected event order.
    """
    actual_idx = 0
    for expected in exp_event_order:
        if isinstance(expected, str):
            assert actual_event_order[actual_idx] == expected, (
                f"Expected event '{expected}' at position {actual_idx}, but got '{actual_event_order[actual_idx]}'."
            )
            actual_idx += 1
        else:  # expected is a set of events that can occur in any order
            expected_events = set(expected)
            actual_events = set(
                actual_event_order[actual_idx : actual_idx + len(expected_events)]
            )
            assert actual_events == expected_events, (
                f"Expected events {expected_events} at positions {actual_idx} to {actual_idx + len(expected_events) - 1}, but got {actual_events}."
            )
            actual_idx += len(expected_events)
    assert actual_idx == len(actual_event_order), (
        f"Expected {actual_idx} events, but got {len(actual_event_order)}: {actual_event_order[actual_idx:]} remain."
    )


def poll_assert_event_order(