Remove try statement. Add test.

cmarmo · cmarmo · commit 4213779a4a11 · 2022-09-07T08:25:28.000-10:00
diff --git a/numpy/ma/core.py b/numpy/ma/core.py
@@ -2357,10 +2357,7 @@ def masked_invalid(a, copy=True):
 
     """
 
-    try:
-        return masked_where(~(np.isfinite(getdata(a))), a, copy=copy)
-    except TypeError:
-        raise
+    return masked_where(~(np.isfinite(getdata(a))), a, copy=copy)
 
 ###############################################################################
 #                            Printing options                                 #
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
@@ -5317,6 +5317,10 @@ def test_masked_array_no_copy():
     a = np.ma.array([1, 2, 3, 4], mask=[1, 0, 0, 0])
     _ = np.ma.masked_where(a == 3, a, copy=False)
     assert_array_equal(a.mask, [True, False, True, False])
+    # check masked array with masked_invalid is updated in place
+    a = np.ma.array([np.inf, 1, 2, 3, 4])
+    _ = np.ma.masked_invalid(a, copy=False)
+    assert_array_equal(a.mask, [True, False, False, False, False])
 
 def test_append_masked_array():
     a = np.ma.masked_equal([1,2,3], value=2)
