File: common.py

package info (click to toggle)
llvm-toolchain-15 1%3A15.0.6-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,554,644 kB
  • sloc: cpp: 5,922,452; ansic: 1,012,136; asm: 674,362; python: 191,568; objc: 73,855; f90: 42,327; lisp: 31,913; pascal: 11,973; javascript: 10,144; sh: 9,421; perl: 7,447; ml: 5,527; awk: 3,523; makefile: 2,520; xml: 885; cs: 573; fortran: 567
file content (44 lines) | stat: -rwxr-xr-x 1,253 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
"""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)