def main():
parser = argparse.ArgumentParser(description="LFortran Test Suite")
parser.add_argument("-u",
"--update",
action="store_true",
help="update all reference results")
parser.add_argument("-l",
"--list",
action="store_true",
help="list all tests")
parser.add_argument("-t", metavar="TEST", help="Run a specific test")
parser.add_argument("-v",
"--verbose",
action="store_true",
help="increase test verbosity")
parser.add_argument("--no-llvm",
action="store_true",
help="Skip LLVM tests")
args = parser.parse_args()
update_reference = args.update
list_tests = args.list
specific_test = args.t
verbose = args.verbose
no_llvm = args.no_llvm
# So that the tests find the `lfortran` executable
os.environ["PATH"] = os.path.join(os.getcwd(), "src", "bin") \
+ os.pathsep + os.environ["PATH"]
d = toml.load(open("tests/tests.toml"))
for test in d["test"]:
filename = test["filename"]
if specific_test and filename != specific_test:
continue
tokens = test.get("tokens", False)
ast = test.get("ast", False)
ast_indent = test.get("ast_indent", False)
ast_f90 = test.get("ast_f90", False)
ast_cpp = test.get("ast_cpp", False)
ast_cpp_hip = test.get("ast_cpp_hip", False)
ast_openmp = test.get("ast_openmp", False)
asr = test.get("asr", False)
asr_preprocess = test.get("asr_preprocess", False)
asr_indent = test.get("asr_indent", False)
mod_to_asr = test.get("mod_to_asr", False)
llvm = test.get("llvm", False)
cpp = test.get("cpp", False)
obj = test.get("obj", False)
x86 = test.get("x86", False)
bin_ = test.get("bin", False)
pass_ = test.get("pass", None)
optimization_passes = ["flip_sign", "div_to_mul", "fma"]
if pass_ and (pass_ not in ["do_loops", "global_stmts"]
and pass_ not in optimization_passes):
raise Exception("Unknown pass: %s" % pass_)
print(color(style.bold) + "TEST:" + color(style.reset), filename)
extra_args = "--no-error-banner"
if tokens:
run_test(
"tokens",
"lfortran --no-color --show-tokens {infile} -o {outfile}",
filename, update_reference, extra_args)
if ast:
if filename.endswith(".f"):
# Use fixed form
run_test(
"ast",
"lfortran --fixed-form --show-ast --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
else:
# Use free form
run_test(
"ast",
"lfortran --show-ast --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
if ast_indent:
run_test(
"ast_indent",
"lfortran --show-ast --indent --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
if ast_f90:
if filename.endswith(".f"):
# Use fixed form
run_test(
"ast_f90",
"lfortran --fixed-form --show-ast-f90 --no-color {infile}",
filename, update_reference, extra_args)
else:
# Use free form
run_test("ast_f90",
"lfortran --show-ast-f90 --no-color {infile}",
filename, update_reference, extra_args)
if ast_openmp:
run_test("ast_openmp", "cpptranslate --show-ast-openmp {infile}",
filename, update_reference)
if asr:
run_test("asr",
"lfortran --show-asr --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
if asr_preprocess:
run_test(
"asr_preprocess",
"lfortran --cpp --show-asr --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
if asr_indent:
run_test(
"asr_indent",
"lfortran --show-asr --indent --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
if mod_to_asr:
run_test("mod_to_asr",
"lfortran mod --show-asr --no-color {infile}", filename,
update_reference)
if pass_ is not None:
cmd = "lfortran --pass="******" --show-asr --no-color {infile} -o {outfile}"
run_test("pass_{}".format(pass_), cmd, filename, update_reference,
extra_args)
if llvm:
if no_llvm:
print(" * llvm SKIPPED as requested")
else:
run_test(
"llvm",
"lfortran --no-color --show-llvm {infile} -o {outfile}",
filename, update_reference, extra_args)
if cpp:
run_test("cpp", "lfortran --no-color --show-cpp {infile}",
filename, update_reference, extra_args)
if obj:
if no_llvm:
print(" * obj SKIPPED as requested")
else:
run_test("obj", "lfortran --no-color -c {infile} -o output.o",
filename, update_reference, extra_args)
if x86:
run_test("x86",
"lfortran --no-color --backend=x86 {infile} -o output",
filename, update_reference, extra_args)
if bin_:
run_test("bin", "lfortran --no-color {infile} -o {outfile}",
filename, update_reference, extra_args)
print()
if list_tests:
return
if update_reference:
print("Reference tests updated.")
else:
print("%sTESTS PASSED%s" % (color(fg.green) + color(style.bold),
color(fg.reset) + color(style.reset)))
def main():
parser = argparse.ArgumentParser(description="LFortran Test Suite")
parser.add_argument("-u",
"--update",
action="store_true",
help="update all reference results")
parser.add_argument("-l",
"--list",
action="store_true",
help="list all tests")
parser.add_argument("-t", metavar="TEST", help="Run a specific test")
parser.add_argument("-v",
"--verbose",
action="store_true",
help="increase test verbosity")
parser.add_argument("--no-llvm",
action="store_true",
help="Skip LLVM tests")
args = parser.parse_args()
update_reference = args.update
list_tests = args.list
specific_test = args.t
verbose = args.verbose
no_llvm = args.no_llvm
# So that the tests find the `lfortran` executable
os.environ["PATH"] = os.path.join(os.getcwd(), "src", "bin") \
+ os.pathsep + os.environ["PATH"]
d = toml.load(open("tests/tests.toml"))
for test in d["test"]:
filename = test["filename"]
if specific_test and filename != specific_test:
continue
tokens = test.get("tokens", False)
ast = test.get("ast", False)
ast_f90 = test.get("ast_f90", False)
ast_cpp = test.get("ast_cpp", False)
asr = test.get("asr", False)
llvm = test.get("llvm", False)
bin_ = test.get("bin", False)
print(color(style.bold) + "TEST:" + color(style.reset), filename)
if tokens:
run_test("tokens", "lfortran --show-tokens {infile} -o {outfile}",
filename, update_reference)
if ast:
run_test("ast", "lfortran --show-ast {infile} -o {outfile}",
filename, update_reference)
if ast_f90:
run_test("ast_f90", "cpptranslate --show-ast-f90 {infile}",
filename, update_reference)
if ast_cpp:
run_test("ast_cpp", "cpptranslate --show-ast-cpp {infile}",
filename, update_reference)
if asr:
run_test("asr", "lfortran --show-asr {infile} -o {outfile}",
filename, update_reference)
if llvm:
if no_llvm:
print(" * llvm SKIPPED as requested")
else:
run_test("llvm", "lfortran --show-llvm {infile} -o {outfile}",
filename, update_reference)
if bin_:
run_test("bin", "lfortran {infile} -o {outfile}", filename,
update_reference)
print()
if list_tests:
return
if update_reference:
print("Reference tests updated.")
else:
print("%sTESTS PASSED%s" % (color(fg.green) + color(style.bold),
color(fg.reset) + color(style.reset)))
def main():
parser = argparse.ArgumentParser(description="LFortran Test Suite")
parser.add_argument("-u",
"--update",
action="store_true",
help="update all reference results")
parser.add_argument("-l",
"--list",
action="store_true",
help="list all tests")
parser.add_argument("-t", metavar="TEST", help="Run a specific test")
parser.add_argument("-v",
"--verbose",
action="store_true",
help="increase test verbosity")
parser.add_argument("--no-llvm",
action="store_true",
help="Skip LLVM tests")
args = parser.parse_args()
update_reference = args.update
list_tests = args.list
specific_test = args.t
verbose = args.verbose
no_llvm = args.no_llvm
# So that the tests find the `lfortran` executable
os.environ["PATH"] = os.path.join(os.getcwd(), "src", "bin") \
+ os.pathsep + os.environ["PATH"]
d = toml.load(open("tests/tests.toml"))
for test in d["test"]:
filename = test["filename"]
if specific_test and filename != specific_test:
continue
tokens = test.get("tokens", False)
ast = test.get("ast", False)
ast_f90 = test.get("ast_f90", False)
ast_cpp = test.get("ast_cpp", False)
ast_cpp_hip = test.get("ast_cpp_hip", False)
ast_openmp = test.get("ast_openmp", False)
asr = test.get("asr", False)
llvm = test.get("llvm", False)
cpp = test.get("cpp", False)
obj = test.get("obj", False)
x86 = test.get("x86", False)
bin_ = test.get("bin", False)
pass_ = test.get("pass", None)
if pass_ and pass_ not in ["do_loops", "global_stmts"]:
raise Exception("Unknown pass: %s" % pass_)
print(color(style.bold) + "TEST:" + color(style.reset), filename)
if tokens:
run_test("tokens", "lfortran --show-tokens {infile} -o {outfile}",
filename, update_reference)
if ast:
run_test("ast",
"lfortran --show-ast --no-color {infile} -o {outfile}",
filename, update_reference)
if ast_f90:
run_test("ast_f90", "lfortran --show-ast-f90 --no-color {infile}",
filename, update_reference)
if ast_cpp:
run_test("ast_cpp", "cpptranslate --show-ast-cpp {infile}",
filename, update_reference)
if ast_cpp_hip:
run_test("ast_cpp_hip", "cpptranslate --show-ast-cpp-hip {infile}",
filename, update_reference)
if ast_openmp:
run_test("ast_openmp", "cpptranslate --show-ast-openmp {infile}",
filename, update_reference)
if asr:
run_test("asr",
"lfortran --show-asr --no-color {infile} -o {outfile}",
filename, update_reference)
if pass_ == "do_loops":
run_test(
"pass_do_loops",
"lfortran --pass=do_loops --show-asr --no-color {infile} -o {outfile}",
filename, update_reference)
if pass_ == "global_stmts":
run_test(
"pass_global_stmts",
"lfortran --pass=global_stmts --show-asr --no-color {infile} -o {outfile}",
filename, update_reference)
if llvm:
if no_llvm:
print(" * llvm SKIPPED as requested")
else:
run_test("llvm", "lfortran --show-llvm {infile} -o {outfile}",
filename, update_reference)
if cpp:
run_test("cpp", "lfortran --show-cpp {infile}", filename,
update_reference)
if obj:
if no_llvm:
print(" * obj SKIPPED as requested")
else:
run_test("obj", "lfortran -c {infile} -o output.o", filename,
update_reference)
if x86:
run_test("x86", "lfortran --backend=x86 {infile} -o output",
filename, update_reference)
if bin_:
run_test("bin", "lfortran {infile} -o {outfile}", filename,
update_reference)
print()
if list_tests:
return
if update_reference:
print("Reference tests updated.")
else:
print("%sTESTS PASSED%s" % (color(fg.green) + color(style.bold),
color(fg.reset) + color(style.reset)))