(self)
| 618 | assert_equal(compress_rowcols(x, 1).size, 0) |
| 619 | |
| 620 | def test_mask_rowcols(self): |
| 621 | # Tests mask_rowcols. |
| 622 | x = array(np.arange(9).reshape(3, 3), |
| 623 | mask=[[1, 0, 0], [0, 0, 0], [0, 0, 0]]) |
| 624 | assert_equal(mask_rowcols(x).mask, |
| 625 | [[1, 1, 1], [1, 0, 0], [1, 0, 0]]) |
| 626 | assert_equal(mask_rowcols(x, 0).mask, |
| 627 | [[1, 1, 1], [0, 0, 0], [0, 0, 0]]) |
| 628 | assert_equal(mask_rowcols(x, 1).mask, |
| 629 | [[1, 0, 0], [1, 0, 0], [1, 0, 0]]) |
| 630 | x = array(x._data, mask=[[0, 0, 0], [0, 1, 0], [0, 0, 0]]) |
| 631 | assert_equal(mask_rowcols(x).mask, |
| 632 | [[0, 1, 0], [1, 1, 1], [0, 1, 0]]) |
| 633 | assert_equal(mask_rowcols(x, 0).mask, |
| 634 | [[0, 0, 0], [1, 1, 1], [0, 0, 0]]) |
| 635 | assert_equal(mask_rowcols(x, 1).mask, |
| 636 | [[0, 1, 0], [0, 1, 0], [0, 1, 0]]) |
| 637 | x = array(x._data, mask=[[1, 0, 0], [0, 1, 0], [0, 0, 0]]) |
| 638 | assert_equal(mask_rowcols(x).mask, |
| 639 | [[1, 1, 1], [1, 1, 1], [1, 1, 0]]) |
| 640 | assert_equal(mask_rowcols(x, 0).mask, |
| 641 | [[1, 1, 1], [1, 1, 1], [0, 0, 0]]) |
| 642 | assert_equal(mask_rowcols(x, 1,).mask, |
| 643 | [[1, 1, 0], [1, 1, 0], [1, 1, 0]]) |
| 644 | x = array(x._data, mask=[[1, 0, 0], [0, 1, 0], [0, 0, 1]]) |
| 645 | assert_(mask_rowcols(x).all() is masked) |
| 646 | assert_(mask_rowcols(x, 0).all() is masked) |
| 647 | assert_(mask_rowcols(x, 1).all() is masked) |
| 648 | assert_(mask_rowcols(x).mask.all()) |
| 649 | assert_(mask_rowcols(x, 0).mask.all()) |
| 650 | assert_(mask_rowcols(x, 1).mask.all()) |
| 651 | |
| 652 | @pytest.mark.parametrize("axis", [None, 0, 1]) |
| 653 | @pytest.mark.parametrize(["func", "rowcols_axis"], |
nothing calls this directly
no test coverage detected