diff options
author | Greg Lucas <greg.lucas@lasp.colorado.edu> | 2020-04-15 10:28:02 -0600 |
---|---|---|
committer | Greg Lucas <greg.m.lucas@gmail.com> | 2022-06-29 19:17:15 -0600 |
commit | 6d77c591c59b5678f14ae5af2127eebb7d2415bc (patch) | |
tree | 0c406f4d9519a563a8e30fec3a71d89e0f847c4a /numpy/lib | |
parent | f9bed20bffd88bce06dbc8be200179edfe7580a4 (diff) | |
download | numpy-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.py | 2 |
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)) |