TheRock/compiler/amd-llvm/flang/test/Semantics/common.py

"""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)