Merge branch 'issue-25607-take-32bit-bug' of https://github.com/JuliaPoo/numpy into issue-25607-take-32bit-bug

Author: JuliaPoo

numpy/_core/tests/test_numeric.py

@@ -324,37 +324,15 @@ def test_take(self):
         out = np.take(a, indices)
         assert_equal(out, tgt)
 
-        # take 32 64
-        x32 = np.array([1, 2, 3, 4, 5], dtype=np.int32)
-        x64 = np.array([0, 2, 2, 3], dtype=np.int64)
-        tgt = np.array([1, 3, 3, 4], dtype=np.int32)
-        out = np.take(x32, x64)
-        assert_equal(out, tgt)
-        assert_equal(out.dtype, tgt.dtype)
-
-        # take 64 32
-        x64 = np.array([1, 2, 3, 4, 5], dtype=np.int64)
-        x32 = np.array([0, 2, 2, 3], dtype=np.int32)
-        tgt = np.array([1, 3, 3, 4], dtype=np.int64)
-        out = np.take(x64, x32)
-        assert_equal(out, tgt)
-        assert_equal(out.dtype, tgt.dtype)
+        pairs = [(np.int32, np.int32), (np.int32, np.int64), (np.int64, np.int32), (np.int64, np.int64)]
+        for array_type, indices_type in pairs:
+                x = np.array([1, 2, 3, 4, 5], dtype=array_type)
+                ind = np.array([0, 2, 2, 3], dtype=indices_type )
+                tgt = np.array([1, 3, 3, 4], dtype=array_type)
+                out = np.take(x, ind)
+                assert_equal(out, tgt)
+                assert_equal(out.dtype, tgt.dtype)    
 
-        # take 32 32
-        x32_0 = np.array([1, 2, 3, 4, 5], dtype=np.int32)
-        x32_1 = np.array([0, 2, 2, 3], dtype=np.int32)
-        tgt = np.array([1, 3, 3, 4], dtype=np.int32)
-        out = np.take(x32_0, x32_1)
-        assert_equal(out, tgt)
-        assert_equal(out.dtype, tgt.dtype)
-
-        # take 64 64
-        x64_0 = np.array([1, 2, 3, 4, 5], dtype=np.int64)
-        x64_1 = np.array([0, 2, 2, 3], dtype=np.int64)
-        tgt = np.array([1, 3, 3, 4], dtype=np.int64)
-        out = np.take(x64_0, x64_1)
-        assert_equal(out, tgt)
-        assert_equal(out.dtype, tgt.dtype)
 
     def test_trace(self):
         c = [[1, 2], [3, 4], [5, 6]]