micropython/tests/run-tests

88 wiersze
2.6 KiB
Python
Executable File

#! /usr/bin/env python3
import os
import subprocess
import sys
from glob import glob
# 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')
MP_PY = '../windows/micropython.exe'
else:
CPYTHON3 = os.getenv('MICROPY_CPYTHON3', 'python3')
MP_PY = '../unix/micropython'
def rm_f(fname):
if os.path.exists(fname):
os.remove(fname)
test_count = 0
testcase_count = 0
passed_count = 0
failed_tests = []
tests = []
if not sys.argv[1:]:
tests = sorted(glob('basics/*.py') + glob('io/*.py') + glob('misc/*.py'))
else:
tests = sys.argv[1:]
test_on_pyboard = False
if test_on_pyboard:
import pyboard
pyb = pyboard.Pyboard('/dev/ttyACM0')
pyb.enter_raw_repl()
for test_file in tests:
test_name = os.path.splitext(os.path.basename(test_file))[0]
# run CPython
try:
output_expected = subprocess.check_output([CPYTHON3, '-B', test_file])
except subprocess.CalledProcessError:
output_expected = b'CPYTHON3 CRASH'
# run Micro Python
if test_on_pyboard:
pyb.enter_raw_repl()
try:
if test_file == 'basics/memoryerror.py':
# this test crashes the pyboard
output_mupy = b'CRASH'
else:
output_mupy = pyb.execfile(test_file).replace(b'\r\n', b'\n')
except pyboard.PyboardError:
output_mupy = b'CRASH\n' + output_mupy
else:
try:
output_mupy = subprocess.check_output([MP_PY, '-X', 'emit=bytecode', test_file])
except subprocess.CalledProcessError:
output_mupy = b'CRASH'
testcase_count += len(output_expected.splitlines())
if output_expected == output_mupy:
print("pass ", test_file)
passed_count += 1
rm_f(os.path.basename(test_file + ".exp"))
rm_f(os.path.basename(test_file + ".out"))
else:
with open(os.path.basename(test_file + ".exp"), "w") as f:
f.write(str(output_expected, "ascii"))
with open(os.path.basename(test_file + ".out"), "w") as f:
f.write(str(output_mupy, "ascii"))
print("FAIL ", test_file)
failed_tests.append(test_name)
test_count += 1
print("{} tests performed ({} individual testcases)".format(test_count, testcase_count))
print("{} tests passed".format(passed_count))
if len(failed_tests) > 0:
print("{} tests failed: {}".format(len(failed_tests), ' '.join(failed_tests)))
sys.exit(1)