carbon-lang/toolchain/lexer/autoupdate_testdata.py

#!/usr/bin/env python3

"""Updates the CHECK: lines in tests with an AUTOUPDATE line."""

__copyright__ = """
Part of the Carbon Language project, under the Apache License v2.0 with LLVM
Exceptions. See /LICENSE for license information.
SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""

import subprocess
import sys
from pathlib import Path


def main() -> None:
    # Subprocess to the main script in order to avoid Python import behaviors.
    this_py = Path(__file__).resolve()
    autoupdate_py = this_py.parent.parent.parent.joinpath(
        "testing", "scripts", "autoupdate_testdata_base.py"
    )
    args = [
        str(autoupdate_py),
        # Flags to configure for lexer testing.
        "--tool=carbon",
        "--autoupdate_arg=dump",
        "--autoupdate_arg=tokens",
        # Ignore the resulting column of EndOfFile because it's typically the
        # end of the CHECK comment.
        "--extra_check_replacement",
        ".*'EndOfFile'",
        r"column: (?:\d+)",
        "column: {{[0-9]+}}",
        # Ignore spaces that are used to columnize lines.
        "--line_number_delta_prefix={{ *}}",
        "--line_number_pattern="
        r"(?P<prefix> line: )(?P<line> *\d+)(?P<suffix>,)",
        "--testdata=toolchain/lexer/testdata",
    ] + sys.argv[1:]
    exit(subprocess.call(args))


if __name__ == "__main__":
    main()