Convert gen-tests shell script to python

tests/gen_tests_makefile.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env python3
+
+import glob
+import os
+import sys
+import argparse
+
+yosys_basedir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+
+def _cwd_base():
+    return os.path.basename(os.getcwd())
+
+def generate_target(name, command):
+    target = f"{_cwd_base()}-{name}"
+    print(f"all: {target}")
+    print(f".PHONY: {target}")
+    print(f"{target}:")
+    print(f"\t@YOSYS_MAX_THREADS=4 {command}")
+    print(f"\t@echo 'Passed {target}'")
+
+def generate_ys_test(ys_file, yosys_args=""):
+    cmd = f'$(YOSYS) -ql {ys_file}.err {yosys_args} {ys_file} >/dev/null 2>&1 && mv {ys_file}.err {ys_file}.log'
+    generate_target(ys_file, cmd)
+
+def generate_tcl_test(tcl_file, yosys_args=""):
+    cmd = f'$(YOSYS) -ql {tcl_file}.err {yosys_args} {tcl_file} >/dev/null 2>&1 && mv {tcl_file}.err {tcl_file}.log'
+    generate_target(tcl_file, cmd)
+
+def generate_sv_test(sv_file, yosys_args=""):
+    base = os.path.splitext(sv_file)[0]
+    if not os.path.exists(base + ".ys"):
+        yosys_cmd = '-p "prep -top top; async2sync; sat -enable_undef -verify -prove-asserts"'
+        cmd = f'$(YOSYS) -ql {sv_file}.err {yosys_cmd} {yosys_args} {sv_file} >/dev/null 2>&1 && mv {sv_file}.err {sv_file}.log'
+        generate_target(sv_file, cmd)
+
+def generate_bash_test(sh_file):
+    cmd = f"bash -v {sh_file} >{sh_file}.err 2>&1 && mv {sh_file}.err {sh_file}.log"
+    generate_target(sh_file, cmd)
+
+def generate_tests(argv):
+    parser = argparse.ArgumentParser(add_help=False)
+    parser.add_argument("-y", "--yosys-scripts", action="store_true")
+    parser.add_argument("-t", "--tcl-scripts", action="store_true")
+    parser.add_argument("-s", "--prove-sv", action="store_true")
+    parser.add_argument("-b", "--bash", action="store_true")
+    parser.add_argument("-a", "--yosys-args", default="")
+
+    args = parser.parse_args(argv)
+
+    if not (args.yosys_scripts or args.tcl_scripts or args.prove_sv or args.bash):
+        raise RuntimeError("No file types selected")
+
+    print(f"YOSYS ?= {yosys_basedir}/yosys")
+    print()
+
+    print(".PHONY: all")
+    print("all:")
+
+    if args.yosys_scripts:
+        for f in sorted(glob.glob("*.ys")):
+            generate_ys_test(f, args.yosys_args)
+
+    if args.tcl_scripts:
+        for f in sorted(glob.glob("*.tcl")):
+            generate_tcl_test(f, args.yosys_args)
+
+    if args.prove_sv:
+        for f in sorted(glob.glob("*.sv")):
+            generate_sv_test(f, args.yosys_args)
+
+    if args.bash:
+        for f in sorted(glob.glob("*.sh")):
+            if f != "run-test.sh":
+                generate_bash_test(f)
+
+def generate(argv, extra=None):
+    with open("Makefile", "w") as f:
+        old = sys.stdout
+        sys.stdout = f
+        try:
+            if extra:
+                for line in extra:
+                    print(line)
+            generate_tests(argv)
+        finally:
+            sys.stdout = old