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
50 lines
1.2 KiB
Python
Executable File
50 lines
1.2 KiB
Python
Executable File
"""Provides common functionality to the test scripts."""
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def set_source(source):
|
|
"""Checks whether the source file exists and returns its path."""
|
|
if not Path(source).is_file():
|
|
die(source)
|
|
return Path(source)
|
|
|
|
|
|
def set_executable(executable):
|
|
"""Checks whether a Flang executable has been set and returns its path."""
|
|
flang_fc1 = Path(executable)
|
|
if not flang_fc1.is_file():
|
|
die(flang_fc1)
|
|
return str(flang_fc1)
|
|
|
|
|
|
def set_temp(tmp):
|
|
"""Sets a temporary directory or creates one if it doesn't exist."""
|
|
os.makedirs(Path(tmp), exist_ok=True)
|
|
return Path(tmp)
|
|
|
|
|
|
def die(file=None):
|
|
"""Used in other functions."""
|
|
if file is None:
|
|
print(f"{sys.argv[0]}: FAIL")
|
|
else:
|
|
print(f"{sys.argv[0]}: File not found: {file}")
|
|
sys.exit(1)
|
|
|
|
|
|
def check_args(args):
|
|
"""Verifies that 2 arguments have been passed."""
|
|
if len(args) < 3:
|
|
print(f"Usage: {args[0]} <fortran-source> <flang-command>")
|
|
sys.exit(1)
|
|
|
|
|
|
def check_args_long(args):
|
|
"""Verifies that 3 arguments have been passed."""
|
|
if len(args) < 4:
|
|
print(f"Usage: {args[0]} <fortran-source> <temp-test-dir> <flang-command>")
|
|
sys.exit(1)
|