mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 00:56:05 +00:00
f98ee40f4b
This is an ongoing series of commits that are reformatting our Python code. This catches the last of the python files to reformat. Since they where so few I bunched them together. Reformatting is done with `black`. If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run git checkout --ours <yourfile> and then reformat it with black. If you run into any problems, post to discourse about it and we will try to help. RFC Thread below: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Reviewed By: jhenderson, #libc, Mordante, sivachandra Differential Revision: https://reviews.llvm.org/D150784
99 lines
3.0 KiB
Python
Executable File
99 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
"""Compiles a source file and compares generated .mod files against expected.
|
|
|
|
Parameters:
|
|
sys.argv[1]: a source file with contains the input and expected output
|
|
sys.argv[2]: the Flang frontend driver
|
|
sys.argv[3:]: Optional arguments to the Flang frontend driver"""
|
|
|
|
import sys
|
|
import re
|
|
import os
|
|
import tempfile
|
|
import subprocess
|
|
import glob
|
|
import common as cm
|
|
|
|
from pathlib import Path
|
|
from difflib import unified_diff
|
|
|
|
cm.check_args_long(sys.argv)
|
|
srcdir = Path(sys.argv[1])
|
|
sources = list(glob.iglob(str(srcdir)))
|
|
sources = sorted(sources)
|
|
diffs = ""
|
|
|
|
flang_fc1 = cm.set_executable(sys.argv[2])
|
|
flang_fc_args = sys.argv[3:]
|
|
flang_fc1_options = "-fsyntax-only"
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
for src in sources:
|
|
src = Path(src).resolve()
|
|
actual = ""
|
|
expect = ""
|
|
expected_files = set()
|
|
actual_files = set()
|
|
|
|
if not src.is_file():
|
|
cm.die(src)
|
|
|
|
prev_files = set(os.listdir(tmpdir))
|
|
cmd = [flang_fc1, *flang_fc_args, flang_fc1_options, str(src)]
|
|
proc = subprocess.check_output(
|
|
cmd, stderr=subprocess.PIPE, universal_newlines=True, cwd=tmpdir
|
|
)
|
|
actual_files = set(os.listdir(tmpdir)).difference(prev_files)
|
|
|
|
# The first 3 bytes of the files are an UTF-8 BOM
|
|
with open(src, "r", encoding="utf-8", errors="strict") as f:
|
|
for line in f:
|
|
m = re.search(r"^!Expect: (.*)", line)
|
|
if m:
|
|
expected_files.add(m.group(1))
|
|
|
|
extra_files = actual_files.difference(expected_files)
|
|
if extra_files:
|
|
print(f"Unexpected .mod files produced: {extra_files}")
|
|
sys.exit(1)
|
|
|
|
for mod in expected_files:
|
|
mod = Path(tmpdir).joinpath(mod)
|
|
if not mod.is_file():
|
|
print(f"Compilation did not produce expected mod file: {mod}")
|
|
sys.exit(1)
|
|
with open(mod, "r", encoding="utf-8", errors="strict") as f:
|
|
for line in f:
|
|
if "!mod$" in line:
|
|
continue
|
|
actual += line
|
|
|
|
with open(src, "r", encoding="utf-8", errors="strict") as f:
|
|
for line in f:
|
|
if f"!Expect: {mod.name}" in line:
|
|
for line in f:
|
|
if re.match(r"^$", line):
|
|
break
|
|
m = re.sub(r"^!", "", line.lstrip())
|
|
expect += m
|
|
|
|
diffs = "\n".join(
|
|
unified_diff(
|
|
actual.replace(" ", "").split("\n"),
|
|
expect.replace(" ", "").split("\n"),
|
|
fromfile=mod.name,
|
|
tofile="Expect",
|
|
n=999999,
|
|
)
|
|
)
|
|
|
|
if diffs != "":
|
|
print(diffs)
|
|
print()
|
|
print("FAIL")
|
|
sys.exit(1)
|
|
|
|
print()
|
|
print("PASS")
|