test: update test_2799 for Numba 0.59.0

scikit-hep · Feb 1, 2024 · 4e3d2cc · 4e3d2cc
1 parent 9c9c90f
commit 4e3d2cc
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_2799_numba_ufunc_resolution.py b/tests/test_2799_numba_ufunc_resolution.py
@@ -22,8 +22,6 @@
 
 @pytest.mark.skipif(not NUMBA_HAS_NEP_50, reason="Numba does not have NEP-50 support")
 def test_numba_ufunc_nep_50():
-    raise NotImplementedError
-
     @numba.vectorize(nopython=True)
     def add(x, y):
         return x + y
@@ -35,7 +33,7 @@ def add(x, y):
         result = add(array, np.int16(np.iinfo(np.int8).max + 1))
 
     flattened = ak.to_numpy(ak.flatten(result))
-    assert flattened.dtype == np.dtype(np.int8)
+    assert flattened.dtype == np.dtype(np.int64)
 
 
 @pytest.mark.skipif(NUMBA_HAS_NEP_50, reason="Numba has NEP-50 support")