108 lines
3.2 KiB
Python
Executable File
108 lines
3.2 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import argparse
|
|
import re
|
|
from glob import glob
|
|
from collections import defaultdict
|
|
|
|
# Tests require at least CPython 3.3. If your default python3 executable
|
|
# is of lower version, you can point MICROPY_CPYTHON3 environment var
|
|
# to the correct executable.
|
|
if os.name == "nt":
|
|
CPYTHON3 = os.getenv("MICROPY_CPYTHON3", "python3.exe")
|
|
MICROPYTHON = os.getenv("MICROPY_MICROPYTHON", "../ports/windows/micropython.exe")
|
|
else:
|
|
CPYTHON3 = os.getenv("MICROPY_CPYTHON3", "python3")
|
|
MICROPYTHON = os.getenv("MICROPY_MICROPYTHON", "../ports/unix/micropython")
|
|
|
|
|
|
def run_tests(pyb, test_dict):
|
|
test_count = 0
|
|
testcase_count = 0
|
|
|
|
for base_test, tests in sorted(test_dict.items()):
|
|
print(base_test + ":")
|
|
for test_file in tests:
|
|
|
|
# run MicroPython
|
|
if pyb is None:
|
|
# run on PC
|
|
try:
|
|
output_mupy = subprocess.check_output(
|
|
[MICROPYTHON, "-X", "emit=bytecode", test_file[0]]
|
|
)
|
|
except subprocess.CalledProcessError:
|
|
output_mupy = b"CRASH"
|
|
else:
|
|
# run on pyboard
|
|
pyb.enter_raw_repl()
|
|
try:
|
|
output_mupy = pyb.execfile(test_file).replace(b"\r\n", b"\n")
|
|
except pyboard.PyboardError:
|
|
output_mupy = b"CRASH"
|
|
|
|
output_mupy = float(output_mupy.strip())
|
|
test_file[1] = output_mupy
|
|
testcase_count += 1
|
|
|
|
test_count += 1
|
|
baseline = None
|
|
for t in tests:
|
|
if baseline is None:
|
|
baseline = t[1]
|
|
print(" %.3fs (%+06.2f%%) %s" % (t[1], (t[1] * 100 / baseline) - 100, t[0]))
|
|
|
|
print("{} tests performed ({} individual testcases)".format(test_count, testcase_count))
|
|
|
|
# all tests succeeded
|
|
return True
|
|
|
|
|
|
def main():
|
|
cmd_parser = argparse.ArgumentParser(description="Run tests for MicroPython.")
|
|
cmd_parser.add_argument("--pyboard", action="store_true", help="run the tests on the pyboard")
|
|
cmd_parser.add_argument("files", nargs="*", help="input test files")
|
|
args = cmd_parser.parse_args()
|
|
|
|
# Note pyboard support is copied over from run-tests.py, not tests, and likely needs revamping
|
|
if args.pyboard:
|
|
import pyboard
|
|
|
|
pyb = pyboard.Pyboard("/dev/ttyACM0")
|
|
pyb.enter_raw_repl()
|
|
else:
|
|
pyb = None
|
|
|
|
if len(args.files) == 0:
|
|
if pyb is None:
|
|
# run PC tests
|
|
test_dirs = ("internal_bench",)
|
|
else:
|
|
# run pyboard tests
|
|
test_dirs = ("basics", "float", "pyb")
|
|
tests = sorted(
|
|
test_file
|
|
for test_files in (glob("{}/*.py".format(dir)) for dir in test_dirs)
|
|
for test_file in test_files
|
|
)
|
|
else:
|
|
# tests explicitly given
|
|
tests = sorted(args.files)
|
|
|
|
test_dict = defaultdict(lambda: [])
|
|
for t in tests:
|
|
m = re.match(r"(.+?)-(.+)\.py", t)
|
|
if not m:
|
|
continue
|
|
test_dict[m.group(1)].append([t, None])
|
|
|
|
if not run_tests(pyb, test_dict):
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|