Function expect_token

tests/test_tokenutil.py:12–39

If cursor_pos is `None`, look for `|` as the cursor position. Assert that the token at the cursor position is `expected`.

(expected: str, cell: str, cursor_pos: int | None = None)

def expect_token(expected: str, cell: str, cursor_pos: int | None = None) -> None:
    """
    If cursor_pos is `None`, look for `|` as the cursor position.
    Assert that the token at the cursor position is `expected`.
    """
    if cursor_pos is None:
        assert (
            cursor_count := cell.count("|")
        ) == 1, (
            f"Cursor position not specified and found {cursor_count} instance(s) of '|'"
        )
        cursor_pos = cell.index("|")
        cell = cell.replace("|", "")
    token = token_at_cursor(cell, cursor_pos)
    offset = 0
    for line in cell.splitlines():
        if offset + len(line) >= cursor_pos:
            break
        else:
            offset += len(line) + 1
    column = cursor_pos - offset
    line_with_cursor = "%s|%s" % (line[:column], line[column:])
    assert token == expected, "Expected %r, got %r in: %r (pos %i)" % (
        expected,
        token,
        line_with_cursor,
        cursor_pos,
    )


def test_simple():