test.py

# detect if running on a supported beagle or not
supported_models = [
    'TI AM335x BeagleBone Black Wireless',
    'TI AM335x BeagleBone Black',
    'TI AM335x PocketBeagle',
    'BeagleBoard.org BeagleBone AI'
]

running_on_beagle = False

try:
    with open("/proc/device-tree/model") as file:
        model = file.read(100).strip()
        if model in supported_models:
            print("Detected: ", model)
            running_on_beagle = True
except OSError:
    print("Not running on a BeagleBone")

# run_anywhere tests
import subprocess
from fnmatch import fnmatch
import os
import sys

for file in os.listdir("tests/run_anywhere"):
    if fnmatch(file, "*.sim"):
        print(f"    ")
        print("Running test ", file)

        transpile_output = subprocess.run(f"bin/simppru --preprocess -t tests/run_anywhere/{file}", shell=True, capture_output=True)

        if transpile_output.returncode != 0:
            print(f"**** ****TEST FAILED**** ****: {file}")
            print("stdout:\n", transpile_output.stdout.decode("utf-8"), sep='')
            print("stderr:\n", transpile_output.stderr.decode("utf-8"), sep='')
            sys.exit(1)

        compile_output = subprocess.run(f"gcc -o tests/run_anywhere/{file}.out /tmp/temp.c", shell=True, capture_output=True)

        if compile_output.returncode != 0:
            print(f"**** ****TEST FAILED**** ****: {file}")
            print("stdout:\n", compile_output.stdout.decode("utf-8"), sep='')
            print("stderr:\n", compile_output.stderr.decode("utf-8"), sep='')
            sys.exit(1)

        run_output = subprocess.run(f"tests/run_anywhere/{file}.out", shell=True, capture_output=True)

        if run_output.returncode != 0:
            print(f"**** ****TEST FAILED**** ****: {file}")
            print("stdout:\n", run_output.stdout.decode("utf-8"), sep='')
            print("stderr:\n", run_output.stderr.decode("utf-8"), sep='')
            sys.exit(1)
        else:
            output = run_output.stdout
        with open(f"tests/run_anywhere/{file}.output", 'r+b') as out:
            expected_output = out.read()

        if output != expected_output:
            print(f"**** ****TEST FAILED**** ****: {file}")
            print(f"Standard Output:{output}")
            print(f"Expected Output:{expected_output}")
            sys.exit(1)
        else:
            print(f"Test {file} passed")
            print(f"Standard Output:{output}")
            print(f"Expected Output:{expected_output}")