diff options
| author | ImenRajhi <imen.rajhi.ir@gmail.com> | 2021-06-14 14:19:14 +0000 |
|---|---|---|
| committer | ImenRajhi <imen.rajhi.ir@gmail.com> | 2021-06-14 14:19:14 +0000 |
| commit | 8ca435cff334671152bed11c817ec04ca5a702de (patch) | |
| tree | a05f34d2cc9e1c8e517babf23ce18cb71f0e6b7c /numpy | |
| parent | 2ecb55d48e8d54b059304d4898b403d0f9031afa (diff) | |
| download | numpy-8ca435cff334671152bed11c817ec04ca5a702de.tar.gz | |
added tests for 18951 fix
Diffstat (limited to 'numpy')
| -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): |