forked from OSchip/llvm-project
45 lines
1.2 KiB
Python
Executable File
45 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)
|
|
|