flang/test/Semantics/test_symbols.py

   1 #!/usr/bin/env python3
   2
   3 """Compiles a source file with "-fdebug-unparse-with-symbols' and verifies
   4 we get the right symbols in the output, i.e. the output should be
   5 the same as the input, except for the copyright comment.
   6
   7 Parameters:
   8     sys.argv[1]: a source file with contains the input and expected output
   9     sys.argv[2]: the Flang frontend driver
  10     sys.argv[3:]: Optional arguments to the Flang frontend driver"""
  11
  12 import sys
  13 import tempfile
  14 import re
  15 import subprocess
  16 import common as cm
  17
  18 from difflib import unified_diff
  19
  20 cm.check_args(sys.argv)
  21 src = cm.set_source(sys.argv[1])
  22 diff1 = ""
  23 diff2 = ""
  24
  25 flang_fc1 = cm.set_executable(sys.argv[2])
  26 flang_fc1_args = sys.argv[3:]
  27 flang_fc1_options = "-fdebug-unparse-with-symbols"
  28
  29 # Strips out blank lines and all comments except for "!DEF:", "!REF:", "!$acc" and "!$omp"
  30 with open(src, "r") as text_in:
  31     for line in text_in:
  32         text = re.sub(r"!(?![DR]EF:|\$omp|\$acc).*", "", line)
  33         text = re.sub(r"^\s*$", "", text)
  34         diff1 += text
  35
  36 # Strips out "!DEF:" and "!REF:" comments
  37 for line in diff1:
  38     text = re.sub(r"![DR]EF:.*", "", line)
  39     diff2 += text
  40
  41 # Compiles, inserting comments for symbols:
  42 cmd = [flang_fc1, *flang_fc1_args, flang_fc1_options]
  43 with tempfile.TemporaryDirectory() as tmpdir:
  44     diff3 = subprocess.check_output(
  45         cmd, input=diff2, universal_newlines=True, cwd=tmpdir
  46     )
  47
  48 # Removes all whitespace to compare differences in files
  49 diff1 = diff1.replace(" ", "")
  50 diff3 = diff3.replace(" ", "")
  51 diff_check = ""
  52
  53 # Compares the input with the output
  54 diff_check = "\n".join(
  55     unified_diff(
  56         diff1.split("\n"),
  57         diff3.split("\n"),
  58         n=999999,
  59         fromfile="Expected_output",
  60         tofile="Actual_output",
  61     )
  62 )
  63
  64 if diff_check != "":
  65     print(diff_check.replace(" ", ""))
  66     print()
  67     print("FAIL")
  68     sys.exit(1)
  69 else:
  70     print()
  71     print("PASS")