summaryrefslogtreecommitdiff
path: root/numpy/lib
diff options
context:
space:
mode:
authorGreg Lucas <greg.lucas@lasp.colorado.edu>2020-04-15 10:28:02 -0600
committerGreg Lucas <greg.m.lucas@gmail.com>2022-06-29 19:17:15 -0600
commit6d77c591c59b5678f14ae5af2127eebb7d2415bc (patch)
tree0c406f4d9519a563a8e30fec3a71d89e0f847c4a /numpy/lib
parentf9bed20bffd88bce06dbc8be200179edfe7580a4 (diff)
downloadnumpy-6d77c591c59b5678f14ae5af2127eebb7d2415bc.tar.gz
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.
Diffstat (limited to 'numpy/lib')
-rw-r--r--numpy/lib/tests/test_function_base.py2
1 files changed, 1 insertions, 1 deletions
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))