MCPcopy Create free account
hub / github.com/numpy/numpy / test_putmask

Method test_putmask

numpy/ma/tests/test_core.py:3414–3442  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

3412 assert_equal(xh._data, [3, 4, 2, 4, 5])
3413
3414 def test_putmask(self):
3415 x = arange(6) + 1
3416 mx = array(x, mask=[0, 0, 0, 1, 1, 1])
3417 mask = [0, 0, 1, 0, 0, 1]
3418 # w/o mask, w/o masked values
3419 xx = x.copy()
3420 putmask(xx, mask, 99)
3421 assert_equal(xx, [1, 2, 99, 4, 5, 99])
3422 # w/ mask, w/o masked values
3423 mxx = mx.copy()
3424 putmask(mxx, mask, 99)
3425 assert_equal(mxx._data, [1, 2, 99, 4, 5, 99])
3426 assert_equal(mxx._mask, [0, 0, 0, 1, 1, 0])
3427 # w/o mask, w/ masked values
3428 values = array([10, 20, 30, 40, 50, 60], mask=[1, 1, 1, 0, 0, 0])
3429 xx = x.copy()
3430 putmask(xx, mask, values)
3431 assert_equal(xx._data, [1, 2, 30, 4, 5, 60])
3432 assert_equal(xx._mask, [0, 0, 1, 0, 0, 0])
3433 # w/ mask, w/ masked values
3434 mxx = mx.copy()
3435 putmask(mxx, mask, values)
3436 assert_equal(mxx._data, [1, 2, 30, 4, 5, 60])
3437 assert_equal(mxx._mask, [0, 0, 1, 1, 1, 0])
3438 # w/ mask, w/ masked values + hardmask
3439 mxx = mx.copy()
3440 mxx.harden_mask()
3441 putmask(mxx, mask, values)
3442 assert_equal(mxx, [1, 2, 30, 4, 5, 60])
3443
3444 def test_ravel(self):
3445 # Tests ravel

Callers

nothing calls this directly

Calls 6

arrayFunction · 0.90
putmaskFunction · 0.90
assert_equalFunction · 0.90
arangeFunction · 0.85
copyMethod · 0.45
harden_maskMethod · 0.45

Tested by

no test coverage detected