mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-20 03:20:18 +00:00
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
|
|
def test_file(filepath, cpython=False):
|
|
if cpython:
|
|
return os.system("python3 " + filepath) == 0
|
|
if sys.platform == 'win32':
|
|
return os.system("main.exe " + filepath) == 0
|
|
else:
|
|
return os.system("./main " + filepath) == 0
|
|
|
|
def test_dir(path):
|
|
print("Testing directory:", path)
|
|
for filename in sorted(os.listdir(path)):
|
|
if not filename.endswith('.py'):
|
|
continue
|
|
filepath = os.path.join(path, filename)
|
|
print("> " + filepath, flush=True)
|
|
|
|
if path == 'benchmarks/':
|
|
_0 = time.time()
|
|
if not test_file(filepath, cpython=True):
|
|
print('cpython run failed')
|
|
continue
|
|
_1 = time.time()
|
|
if not test_file(filepath): exit(1)
|
|
_2 = time.time()
|
|
print(f' cpython: {_1 - _0:.6f}s (100%)')
|
|
print(f' pocketpy: {_2 - _1:.6f}s ({(_2 - _1) / (_1 - _0) * 100:.2f}%)')
|
|
else:
|
|
if not test_file(filepath):
|
|
print('-' * 50)
|
|
print("TEST FAILED! Press any key to continue...")
|
|
input()
|
|
|
|
|
|
if len(sys.argv) == 2:
|
|
assert 'benchmark' in sys.argv[1]
|
|
d = 'benchmarks/'
|
|
else:
|
|
d = 'tests/'
|
|
test_dir(d)
|
|
print("ALL TESTS PASSED") |