import timeit
import numpy as np

setup_quantize_slow = """
from test_quantize import quantize_slow
"""
print("Pure python:", np.mean(np.array(timeit.Timer("import numpy as np; a = np.random.rand(100000).astype(float); n = 3; quantize_slow(a, n)", setup=setup_quantize_slow).repeat(7,10))))


setup_quantize_fast = """
from quantize import quantize
"""

print("C++ implementation:", np.mean(np.array(timeit.Timer("import numpy as np; a = np.random.rand(100000).astype(float); n = 3; quantize(a, n)", setup=setup_quantize_fast).repeat(7,10))))