From 6d77c591c59b5678f14ae5af2127eebb7d2415bc Mon Sep 17 00:00:00 2001 From: Greg Lucas Date: Wed, 15 Apr 2020 10:28:02 -0600 Subject: ENH: Adding __array_ufunc__ capability to MaskedArrays. This enables any ufunc numpy operations that are called on a MaskedArray to use the masked version of that function automatically without needing to resort to np.ma.func() calls. --- numpy/lib/tests/test_function_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/lib/tests/test_function_base.py') diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 8457551ca..1c3c32bdd 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -780,7 +780,7 @@ class TestDiff: mask=[[False, False], [True, False], [False, True], [True, True], [False, False]]) out = diff(x) - assert_array_equal(out.data, [[1], [1], [1], [1], [1]]) + assert_array_equal(out.data, [[1], [4], [6], [8], [1]]) assert_array_equal(out.mask, [[False], [True], [True], [True], [False]]) assert_(type(out) is type(x)) -- cgit v1.2.1