Fix test. Use expected result computed by numpy.

Deezy33-creator · Mar 14, 2023 · ae07d34 · ae07d34
1 parent f6a4580
commit ae07d34
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/numba/tests/test_np_functions.py b/numba/tests/test_np_functions.py
@@ -2218,11 +2218,12 @@ def test_argpartition_basic(self):
         cfunc = jit(nopython=True)(pyfunc)
 
         d = np.array([], dtype=np.int64)
+        expected = pyfunc(d, 0)
         got = cfunc(d, 0)
-        self.assertPreciseEqual(d, got)
+        self.assertPreciseEqual(expected, got)
 
         d = np.ones(1, dtype=np.int64)
-        expected = np.zeros(1, dtype=np.int64)
+        expected = pyfunc(d, 0)
         got = cfunc(d, 0)
         self.assertPreciseEqual(expected, got)