Fix test for sorting with other-endian field.

barrogar · Jan 19, 2010 · a9905df · a9905df
1 parent d61aaf7
commit a9905df
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/numpy/core/tests/test_multiarray.py b/numpy/core/tests/test_multiarray.py
@@ -414,11 +414,13 @@ def test_sort_order(self):
             strtype = '>i2'
         else:
             strtype = '<i2'
+        mydtype = [('name', 'S5'),('col2',strtype)]
         r = np.array([('a', 1),('b', 255), ('c', 3), ('d', 258)],
-                     dtype=[('name', 'S5'),('col2',strtype)])
+                     dtype= mydtype)
         r.sort(order='col2')
         assert_equal(r['col2'], [1, 3, 255, 258])
-        assert_equal(r, [('a', 1), ('c', 3), ('b', 255), ('d', 258)])
+        assert_equal(r, np.array([('a', 1), ('c', 3), ('b', 255), ('d', 258)],
+                                 dtype=mydtype))
 
     def test_argsort(self):
         # all c scalar argsorts use the same code with different types