mirror of
https://github.com/zekexiao/pocketlang.git
synced 2025-02-06 04:37:47 +08:00
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import subprocess, os, sys
|
|
import json, re
|
|
from os.path import join
|
|
|
|
## FIXME: this file needs to be refactored.
|
|
|
|
## All benchmark files. ## TODO: pass args for iterations.
|
|
benchmarks = {
|
|
"factors" : ['.pk', '.py', '.rb', '.wren'],
|
|
"fib" : ['.pk', '.py', '.rb', '.wren'],
|
|
"list" : ['.pk', '.py', '.rb'],
|
|
"loop" : ['.pk', '.py', '.rb', ".wren"],
|
|
"primes" : ['.pk', '.py', '.rb', ".wren"],
|
|
}
|
|
|
|
def main():
|
|
run_all_benchmarks()
|
|
|
|
def run_all_benchmarks():
|
|
print_title("BENCHMARKS")
|
|
|
|
def get_interpreter(file):
|
|
if file.endswith('.pk' ) : return 'pocket'
|
|
if file.endswith('.py' ) : return 'python'
|
|
if file.endswith('.rb' ) : return 'ruby'
|
|
if file.endswith('.wren') : return 'wren'
|
|
assert False
|
|
|
|
for bm_name in benchmarks:
|
|
print(bm_name + ":")
|
|
for ext in benchmarks[bm_name]:
|
|
file = join(bm_name, bm_name + ext)
|
|
interpreter = get_interpreter(file)
|
|
print(' %10s: '%interpreter, end=''); sys.stdout.flush()
|
|
result = run_command([interpreter, file])
|
|
time = re.findall(r'elapsed:\s*([0-9\.]+)\s*s',
|
|
result.stdout.decode('utf8'),
|
|
re.MULTILINE)
|
|
assert len(time) == 1, r'elapsed:\s*([0-9\.]+)\s*s --> no mach found.'
|
|
print('%10ss'%time[0])
|
|
|
|
def run_command(command):
|
|
return subprocess.run(command,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
|
|
def print_title(title):
|
|
print("--------------------------------")
|
|
print(" %s " % title)
|
|
print("--------------------------------")
|
|
|
|
if __name__ == '__main__':
|
|
main()
|