Remove try statement. Add test.

numpy · Sep 7, 2022 · 4213779 · 4213779
1 parent 2d73e10
commit 4213779
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
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)