(self)
| 3626 | assert_equal(xh._data, [3, 4, 2, 4, 5]) |
| 3627 | |
| 3628 | def test_putmask(self): |
| 3629 | x = arange(6) + 1 |
| 3630 | mx = array(x, mask=[0, 0, 0, 1, 1, 1]) |
| 3631 | mask = [0, 0, 1, 0, 0, 1] |
| 3632 | # w/o mask, w/o masked values |
| 3633 | xx = x.copy() |
| 3634 | putmask(xx, mask, 99) |
| 3635 | assert_equal(xx, [1, 2, 99, 4, 5, 99]) |
| 3636 | # w/ mask, w/o masked values |
| 3637 | mxx = mx.copy() |
| 3638 | putmask(mxx, mask, 99) |
| 3639 | assert_equal(mxx._data, [1, 2, 99, 4, 5, 99]) |
| 3640 | assert_equal(mxx._mask, [0, 0, 0, 1, 1, 0]) |
| 3641 | # w/o mask, w/ masked values |
| 3642 | values = array([10, 20, 30, 40, 50, 60], mask=[1, 1, 1, 0, 0, 0]) |
| 3643 | xx = x.copy() |
| 3644 | putmask(xx, mask, values) |
| 3645 | assert_equal(xx._data, [1, 2, 30, 4, 5, 60]) |
| 3646 | assert_equal(xx._mask, [0, 0, 1, 0, 0, 0]) |
| 3647 | # w/ mask, w/ masked values |
| 3648 | mxx = mx.copy() |
| 3649 | putmask(mxx, mask, values) |
| 3650 | assert_equal(mxx._data, [1, 2, 30, 4, 5, 60]) |
| 3651 | assert_equal(mxx._mask, [0, 0, 1, 1, 1, 0]) |
| 3652 | # w/ mask, w/ masked values + hardmask |
| 3653 | mxx = mx.copy() |
| 3654 | mxx.harden_mask() |
| 3655 | putmask(mxx, mask, values) |
| 3656 | assert_equal(mxx, [1, 2, 30, 4, 5, 60]) |
| 3657 | |
| 3658 | def test_ravel(self): |
| 3659 | # Tests ravel |
nothing calls this directly
no test coverage detected