| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- """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()
|
