diff --git a/test/test_algorithm.py b/test/test_algorithm.py index 42084e36ba6af6a48c3d83393eb8fa67c54bc324..8770d1a7cfc37e62eb3f752dd07ca92f8d6e3360 100644 --- a/test/test_algorithm.py +++ b/test/test_algorithm.py @@ -875,11 +875,18 @@ def test_sort(ctx_factory, scan_kernel): a_sorted = np.sort(a) numpy_end = time() + assert (a_dev_sorted.get() == a_sorted).all() + numpy_elapsed = numpy_end-dev_end dev_elapsed = dev_end-dev_start - print(" dev: {:.2f} MKeys/s numpy: {:.2f} MKeys/s ratio: {:.2f}x".format( - 1e-6*n/dev_elapsed, 1e-6*n/numpy_elapsed, numpy_elapsed/dev_elapsed)) - assert (a_dev_sorted.get() == a_sorted).all() + + # windows clock has really low resolution (16 milliseconds) and the + # difference in time will end up at zero for smaller array sizes. + if numpy_elapsed != 0 and dev_elapsed != 0: + print( + " dev: {:.2f} MKeys/s numpy: {:.2f} MKeys/s ratio: {:.2f}x".format( + 1e-6*n/dev_elapsed, 1e-6*n/numpy_elapsed, + numpy_elapsed/dev_elapsed)) def test_list_builder(ctx_factory):