TST: Verify np.ma.take works on scalars

eric-wieser · eric-wieser · commit 6e5aa5e6b798 · 2016-04-28T20:58:07.000-04:00
See #7585
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
@@ -2998,6 +2998,10 @@ def test_take(self):
         assert_equal(x.take([[0, 1], [0, 1]]),
                      masked_array([[10, 20], [10, 20]], [[0, 1], [0, 1]]))
 
+        # assert_equal crashes when passed np.ma.mask
+        self.assertIs(x[1], np.ma.masked)
+        self.assertIs(x.take(1), np.ma.masked)
+
         x = array([[10, 20, 30], [40, 50, 60]], mask=[[0, 0, 1], [1, 0, 0, ]])
         assert_equal(x.take([0, 2], axis=1),
                      array([[10, 30], [40, 60]], mask=[[0, 1], [1, 0]]))
