rpnifs/tests/benchmark.py

#!/usr/bin/python3

import sys
import random
import time

try:
    import pyrpn
except (ImportError, NameError) as e:
    print("Error importing pyrpn. Try to run make.",
          file=sys.stderr)
    raise e


expr_count = 0x200
max_iter = 0x3000
argc = 2
sz = 0x10

if len(sys.argv) > 1:
    expr_count = int(sys.argv[1], 0)
if len(sys.argv) > 2:
    max_iter = int(sys.argv[2], 0)
if len(sys.argv) > 3:
    argc = int(sys.argv[3], 0)
if len(sys.argv) > 4:
    sz = int(sys.argv[4], 0)

tot = 0
time_op = 0
start = time.time()
IMAX = (1<<63)-1
samples = 8
rnd_samples = [[[random.randint(0, IMAX) for _ in range(argc)] for _ in range(max_iter)]
               for _ in range(samples)]
for i in range(expr_count):
    rnd_expr = pyrpn.random_expr(argc, sz)
    all_start = time.time()
    expr = pyrpn.RPNExpr(rnd_expr, argc)
    start_op = time.time()
    rnds = random.choice(rnd_samples)
    for rnd in rnds:
        expr.eval(*rnd)
    time_op += time.time() - start_op
    tot += time.time() - all_start
    sys.stderr.write(".")
    sys.stderr.flush()
sys.stderr.write("\n")
total = time.time() - start
print("Runned %.2fs" % total)
print("%.1fK iter/s %.1fms per expr" % (max_iter*expr_count/time_op/1000, tot/expr_count*1000))