def bench_length():
repeat = 10
nb_streamlines = DATA['nb_streamlines']
streamlines = DATA["streamlines"] # Streamlines as a list of ndarrays.
print("Timing length() with {0:,} streamlines.".format(nb_streamlines))
python_time = measure("[length_python(s) for s in streamlines]", repeat)
print("Python time: {0:.2}sec".format(python_time))
cython_time = measure("length(streamlines)", repeat)
print("Cython time: {0:.3}sec".format(cython_time))
print("Speed up of {0:.2f}x".format(python_time / cython_time))
# Make sure it produces the same results.
assert_array_equal([length_python(s) for s in DATA["streamlines"]],
length(DATA["streamlines"]))
streamlines = DATA['streamlines_arrseq']
cython_time_arrseq = measure("length(streamlines)", repeat)
print("Cython time (ArrSeq): {0:.3}sec".format(cython_time_arrseq))
print("Speed up of {0:.2f}x".format(python_time / cython_time_arrseq))
# Make sure it produces the same results.
assert_array_equal(length(DATA["streamlines"]),
length(DATA["streamlines_arrseq"]))
def bench_length():
repeat = 10
nb_streamlines = DATA['nb_streamlines']
msg = "Timing length() with {0:,} streamlines."
print(msg.format(nb_streamlines * repeat))
python_time = measure("[length_python(s) for s in streamlines]", repeat)
print("Python time: {0:.2f} sec".format(python_time))
cython_time = measure("length(streamlines)", repeat)
print("Cython time: {0:.3f} sec".format(cython_time))
print("Speed up of {0:.2f}x".format(python_time/cython_time))
# Make sure it produces the same results.
assert_array_almost_equal([length_python(s) for s in DATA["streamlines"]],
length(DATA["streamlines"]))
cython_time_arrseq = measure("length(streamlines)", repeat)
print("Cython time (ArrSeq): {0:.3f} sec".format(cython_time_arrseq))
print("Speed up of {0:.2f}x".format(python_time/cython_time_arrseq))
# Make sure it produces the same results.
assert_array_equal(length(DATA["streamlines"]),
length(DATA["streamlines_arrseq"]))