diff options
| -rw-r--r-- | numpy/ma/tests/test_old_ma.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/numpy/ma/tests/test_old_ma.py b/numpy/ma/tests/test_old_ma.py index ab003b94e..7ac4c328f 100644 --- a/numpy/ma/tests/test_old_ma.py +++ b/numpy/ma/tests/test_old_ma.py @@ -697,6 +697,13 @@ class TestMa: assert_equal(b[0].shape, ()) assert_equal(b[1].shape, ()) + def test_assignment_by_condition(self): + # Test for gh-18951 + a = array([1, 2, 3, 4], mask=[1, 0, 1, 0]) + c = a>=3 + a[c] = 5 + assert_(a[2] is masked) + class TestUfuncs: def setup(self): |