MAINT: improve coverage of slow path by using a user-defined dtype

mattip · mattip · commit 217e2a7086a3 · 2023-01-03T11:41:30.000+02:00
diff --git a/numpy/core/tests/test_ufunc.py b/numpy/core/tests/test_ufunc.py
@@ -1923,12 +1923,19 @@ def __rmul__(self, other):
         assert_(MyThing.rmul_count == 1, MyThing.rmul_count)
         assert_(MyThing.getitem_count <= 2, MyThing.getitem_count)
 
-    def test_inplace_fancy_indexing(self):
+    @pytest.mark.parametrize("a", (
+                             np.arange(10, dtype=int),
+                             np.arange(10, dtype=_rational_tests.rational),
 
-        a = np.arange(10)
+    ))
+    def test_inplace_fancy_indexing(self, a):
         np.add.at(a, [2, 5, 2], 1)
         assert_equal(a, [0, 1, 4, 3, 4, 6, 6, 7, 8, 9])
 
+        np.negative.at(a, [2, 5, 3])
+        assert_equal(a, [0, 1, -4, -3, 4, -6, 6, 7, 8, 9])
+
+        # reset a
         a = np.arange(10)
         b = np.array([100, 100, 100])
         np.add.at(a, [2, 5, 2], b)
