50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
"""Benchmark the cu2qu algorithm performance."""
|
|
|
|
from .cu2qu import *
|
|
import random
|
|
import timeit
|
|
|
|
MAX_ERR = 5
|
|
|
|
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(
|
|
benchmark_module, 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. / number))
|
|
|
|
def main():
|
|
"""Benchmark the cu2qu algorithm performance."""
|
|
run_benchmark('cu2qu.benchmark', 'cu2qu', 'curve_to_quadratic')
|
|
run_benchmark('cu2qu.benchmark', 'cu2qu', 'curves_to_quadratic')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
random.seed(1)
|
|
main()
|