diff options
| -rw-r--r-- | numpy/core/tests/test_multiarray.py | 44 |
1 files changed, 22 insertions, 22 deletions
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py index fba7a2981..99861748c 100644 --- a/numpy/core/tests/test_multiarray.py +++ b/numpy/core/tests/test_multiarray.py @@ -491,7 +491,7 @@ class TestMethods(TestCase): # 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. - a = np.arange(100) + a = np.arange(101) b = a[::-1].copy() for kind in ['q','m','h'] : msg = "scalar sort, kind=%s" % kind @@ -527,7 +527,7 @@ class TestMethods(TestCase): # test string sorts. s = 'aaaaaaaa' - a = np.array([s + chr(i) for i in range(100)]) + a = np.array([s + chr(i) for i in range(101)]) b = a[::-1].copy() for kind in ['q', 'm', 'h'] : msg = "string sort, kind=%s" % kind @@ -540,7 +540,7 @@ class TestMethods(TestCase): # test unicode sorts. s = 'aaaaaaaa' - a = np.array([s + chr(i) for i in range(100)], dtype=np.unicode) + a = np.array([s + chr(i) for i in range(101)], dtype=np.unicode) b = a[::-1].copy() for kind in ['q', 'm', 'h'] : msg = "unicode sort, kind=%s" % kind @@ -552,8 +552,8 @@ class TestMethods(TestCase): assert_equal(c, a, msg) # test object array sorts. - a = np.empty((100,), dtype=np.object) - a[:] = range(100) + a = np.empty((101,), dtype=np.object) + a[:] = range(101) b = a[::-1] for kind in ['q', 'h', 'm'] : msg = "object sort, kind=%s" % kind @@ -566,7 +566,7 @@ class TestMethods(TestCase): # test record array sorts. dt = np.dtype([('f',float),('i',int)]) - a = array([(i,i) for i in range(100)], dtype = dt) + a = array([(i,i) for i in range(101)], dtype = dt) b = a[::-1] for kind in ['q', 'h', 'm'] : msg = "object sort, kind=%s" % kind @@ -578,7 +578,7 @@ class TestMethods(TestCase): assert_equal(c, a, msg) # test datetime64 sorts. - a = np.arange(0, 100, dtype='datetime64[D]') + a = np.arange(0, 101, dtype='datetime64[D]') b = a[::-1] for kind in ['q', 'h', 'm'] : msg = "datetime64 sort, kind=%s" % kind @@ -590,7 +590,7 @@ class TestMethods(TestCase): assert_equal(c, a, msg) # test timedelta64 sorts. - a = np.arange(0, 100, dtype='timedelta64[D]') + a = np.arange(0, 101, dtype='timedelta64[D]') b = a[::-1] for kind in ['q', 'h', 'm'] : msg = "timedelta64 sort, kind=%s" % kind @@ -657,7 +657,7 @@ class TestMethods(TestCase): # 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. - a = np.arange(100) + a = np.arange(101) b = a[::-1].copy() for kind in ['q','m','h'] : msg = "scalar argsort, kind=%s" % kind @@ -681,9 +681,9 @@ class TestMethods(TestCase): # test string argsorts. s = 'aaaaaaaa' - a = np.array([s + chr(i) for i in range(100)]) + a = np.array([s + chr(i) for i in range(101)]) b = a[::-1].copy() - r = np.arange(100) + r = np.arange(101) rr = r[::-1] for kind in ['q', 'm', 'h'] : msg = "string argsort, kind=%s" % kind @@ -692,9 +692,9 @@ class TestMethods(TestCase): # test unicode argsorts. s = 'aaaaaaaa' - a = np.array([s + chr(i) for i in range(100)], dtype=np.unicode) + a = np.array([s + chr(i) for i in range(101)], dtype=np.unicode) b = a[::-1] - r = np.arange(100) + r = np.arange(101) rr = r[::-1] for kind in ['q', 'm', 'h'] : msg = "unicode argsort, kind=%s" % kind @@ -702,10 +702,10 @@ class TestMethods(TestCase): assert_equal(b.copy().argsort(kind=kind), rr, msg) # test object array argsorts. - a = np.empty((100,), dtype=np.object) - a[:] = range(100) + a = np.empty((101,), dtype=np.object) + a[:] = range(101) b = a[::-1] - r = np.arange(100) + r = np.arange(101) rr = r[::-1] for kind in ['q', 'm', 'h'] : msg = "object argsort, kind=%s" % kind @@ -714,9 +714,9 @@ class TestMethods(TestCase): # test structured array argsorts. dt = np.dtype([('f',float),('i',int)]) - a = array([(i,i) for i in range(100)], dtype = dt) + a = array([(i,i) for i in range(101)], dtype = dt) b = a[::-1] - r = np.arange(100) + r = np.arange(101) rr = r[::-1] for kind in ['q', 'm', 'h'] : msg = "structured array argsort, kind=%s" % kind @@ -724,9 +724,9 @@ class TestMethods(TestCase): assert_equal(b.copy().argsort(kind=kind), rr, msg) # test datetime64 argsorts. - a = np.arange(0, 100, dtype='datetime64[D]') + a = np.arange(0, 101, dtype='datetime64[D]') b = a[::-1] - r = np.arange(100) + r = np.arange(101) rr = r[::-1] for kind in ['q', 'h', 'm'] : msg = "datetime64 argsort, kind=%s" % kind @@ -734,9 +734,9 @@ class TestMethods(TestCase): assert_equal(b.copy().argsort(kind=kind), rr, msg) # test timedelta64 argsorts. - a = np.arange(0, 100, dtype='timedelta64[D]') + a = np.arange(0, 101, dtype='timedelta64[D]') b = a[::-1] - r = np.arange(100) + r = np.arange(101) rr = r[::-1] for kind in ['q', 'h', 'm'] : msg = "timedelta64 argsort, kind=%s" % kind |