From 283bb98740a42e950c3f39a9c1e220a9f07a20e7 Mon Sep 17 00:00:00 2001 From: Stuart Archibald Date: Tue, 16 Nov 2021 12:06:11 +0000 Subject: [PATCH] Skip NaT input on np.median test for np=1.21. As title. See: https://github.com/numpy/numpy/issues/20376 --- numba/tests/test_array_reductions.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/numba/tests/test_array_reductions.py b/numba/tests/test_array_reductions.py index de0d445fe66..06996b56211 100644 --- a/numba/tests/test_array_reductions.py +++ b/numba/tests/test_array_reductions.py @@ -686,8 +686,12 @@ def _do_check_nptimedelta(self, pyfunc, arr): np.random.shuffle(arr) self.assertPreciseEqual(cfunc(arr), pyfunc(arr)) # Test with a NaT - arr[arr.size // 2] = 'NaT' - self.assertPreciseEqual(cfunc(arr), pyfunc(arr)) + if numpy_version == (1, 21): + # There's problems with NaT handling on at least NumPy 1.21.{3, 4} + # See https://github.com/numpy/numpy/issues/20376 + if 'median' not in pyfunc.__name__: + arr[arr.size // 2] = 'NaT' + self.assertPreciseEqual(cfunc(arr), pyfunc(arr)) if 'median' not in pyfunc.__name__: # Test with (val, NaT)^N (and with the random NaT from above) # use a loop, there's some weird thing/bug with arr[1::2] = 'NaT'