56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
|
"""Benchmark the cu2qu algorithm performance."""
|
||
|
|
|
||
|
|
from .cu2qu import *
|
||
|
|
import random
|
||
|
|
import timeit
|
||
|
|
|
||
|
|
MAX_ERR = 0.05
|
||
|
|
|
||
|
|
|
||
|
|
def generate_curve():
|
||
|
|
return [
|
||
|
|
tuple(float(random.randint(0, 2048)) for coord in range(2))
|
||
|
|
for point in range(4)
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def setup_curve_to_quadratic():
|
||
|
|
return generate_curve(), MAX_ERR
|
||
|
|
|
||
|
|
|
||
|
|
def setup_curves_to_quadratic():
|
||
|
|
num_curves = 3
|
||
|
|
return ([generate_curve() for curve in range(num_curves)], [MAX_ERR] * num_curves)
|
||
|
|
|
||
|
|
|
||
|
|
def run_benchmark(module, function, setup_suffix="", repeat=5, number=1000):
|
||
|
|
setup_func = "setup_" + function
|
||
|
|
if setup_suffix:
|
||
|
|
print("%s with %s:" % (function, setup_suffix), end="")
|
||
|
|
setup_func += "_" + setup_suffix
|
||
|
|
else:
|
||
|
|
print("%s:" % function, end="")
|
||
|
|
|
||
|
|
def wrapper(function, setup_func):
|
||
|
|
function = globals()[function]
|
||
|
|
setup_func = globals()[setup_func]
|
||
|
|
|
||
|
|
def wrapped():
|
||
|
|
return function(*setup_func())
|
||
|
|
|
||
|
|
return wrapped
|
||
|
|
|
||
|
|
results = timeit.repeat(wrapper(function, setup_func), repeat=repeat, number=number)
|
||
|
|
print("\t%5.1fus" % (min(results) * 1000000.0 / number))
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""Benchmark the cu2qu algorithm performance."""
|
||
|
|
run_benchmark("cu2qu", "curve_to_quadratic")
|
||
|
|
run_benchmark("cu2qu", "curves_to_quadratic")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
random.seed(1)
|
||
|
|
main()
|