| """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(): | |
| run_benchmark("cu2qu", "curve_to_quadratic") | |
| run_benchmark("cu2qu", "curves_to_quadratic") | |
| if __name__ == "__main__": | |
| random.seed(1) | |
| main() | |