"""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():
    run_benchmark('cu2qu.benchmark', 'cu2qu', 'curve_to_quadratic')
    run_benchmark('cu2qu.benchmark', 'cu2qu', 'curves_to_quadratic')


if __name__ == '__main__':
    random.seed(1)
    main()

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

Document main()