TST: Test all dtypes for sort.

hameerabbasi · hameerabbasi · commit 7f586789e4b4 · 2019-05-11T15:12:41.000-07:00
Testing:
- np.int8/np.uint8
- np.int16/np.uint16
- np.int32/np.uint32
- np.int64/np.uint64
- np.float32/np.float64
- np.float16/np.longdouble
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py
@@ -1616,7 +1616,8 @@ def test_sort(self):
         # of sorted items must be greater than ~50 to check the actual
         # algorithm because quick and merge sort fall over to insertion
         # sort for small arrays.
-        for dtype in [np.int32, np.uint32, np.float32]:
+        # Test unsigned dtypes and nonnegative numbers
+        for dtype in [np.uint8, np.uint16, np.uint32, np.uint64, np.float16, np.float32, np.float64, np.longdouble]:
             a = np.arange(101, dtype=dtype)
             b = a[::-1].copy()
             for kind in self.sort_kinds:
@@ -1628,6 +1629,19 @@ def test_sort(self):
                 c.sort(kind=kind)
                 assert_equal(c, a, msg)
 
+        # Test signed dtypes and negative numbers as well
+        for dtype in [np.int8, np.int16, np.int32, np.int64, np.float16, np.float32, np.float64, np.longdouble]:
+            a = np.arange(-50, 51, dtype=dtype)
+            b = a[::-1].copy()
+            for kind in self.sort_kinds:
+                msg = "scalar sort, kind=%s, dtype=%s" % (kind, dtype)
+                c = a.copy()
+                c.sort(kind=kind)
+                assert_equal(c, a, msg)
+                c = b.copy()
+                c.sort(kind=kind)
+                assert_equal(c, a, msg)
+
         # test complex sorts. These use the same code as the scalars
         # but the compare function differs.
         ai = a*1j + 1
