summaryrefslogtreecommitdiff
path: root/numpy/lib/tests
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-07-13 10:57:09 -0600
commit8cd6f4ca00b6e0da3833fc267d50067b2ddbc069 (patch)
tree56f599a1c12b28de18a7ad3b77c4bd187b58aec8 /numpy/lib/tests
parent37846563156b3ed8287fc94d6a7d107a316c5ecf (diff)
downloadnumpy-8cd6f4ca00b6e0da3833fc267d50067b2ddbc069.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/tests')
-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))