TST: Verify that sorting subarrays is fixed

Tests for gh-8069

1 files changed, 14 insertions, 0 deletions

diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
index a44f765e3..3a401caed 100644
--- a/numpy/ma/tests/test_core.py
+++ b/numpy/ma/tests/test_core.py
@@ -3128,6 +3128,20 @@ class TestMaskedArrayMethods(TestCase):
         assert_equal(test, mask_first)
         assert_equal(test.mask, mask_first.mask)
 
+        # Test sort on dtype with subarray (gh-8069)
+        dt = np.dtype([('v', int, 2)])
+        a = a.view(dt)
+        mask_last = mask_last.view(dt)
+        mask_first = mask_first.view(dt)
+
+        test = sort(a)
+        assert_equal(test, mask_last)
+        assert_equal(test.mask, mask_last.mask)
+
+        test = sort(a, endwith=False)
+        assert_equal(test, mask_first)
+        assert_equal(test.mask, mask_first.mask)
+
     def test_argsort(self):
         # Test argsort
         a = array([1, 5, 2, 4, 3], mask=[1, 0, 0, 1, 0])