Function assert_fields

tests/python/relax/test_ast_printer.py:61–71

Given a target string, ensure that the string defines the specified node and that the given mappings of fields to values are present in the string. Strips all whitespace in the target and fields. Does not assume any particular ordering for the fields.

(nodename: str, fields: dict[str, str], target: str)

def assert_fields(nodename: str, fields: dict[str, str], target: str) -> None:
    """
    Given a target string, ensure that the string defines the specified node
    and that the given mappings of fields to values are present in the string.
    Strips all whitespace in the target and fields.
    Does not assume any particular ordering for the fields.
    """
    stripped_target = strip_whitespace(target)
    assert stripped_target.startswith(f"{nodename}(")
    for field, value in fields.items():
        assert f"{field}={strip_whitespace(value)}" in stripped_target


# test cases are mostly adapted from text_expr, only testing very basic properties