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