(self)
| 598 | assert_equal(a, a2) |
| 599 | |
| 600 | def test_compress_rowcols(self): |
| 601 | # Tests compress_rowcols |
| 602 | x = array(np.arange(9).reshape(3, 3), |
| 603 | mask=[[1, 0, 0], [0, 0, 0], [0, 0, 0]]) |
| 604 | assert_equal(compress_rowcols(x), [[4, 5], [7, 8]]) |
| 605 | assert_equal(compress_rowcols(x, 0), [[3, 4, 5], [6, 7, 8]]) |
| 606 | assert_equal(compress_rowcols(x, 1), [[1, 2], [4, 5], [7, 8]]) |
| 607 | x = array(x._data, mask=[[0, 0, 0], [0, 1, 0], [0, 0, 0]]) |
| 608 | assert_equal(compress_rowcols(x), [[0, 2], [6, 8]]) |
| 609 | assert_equal(compress_rowcols(x, 0), [[0, 1, 2], [6, 7, 8]]) |
| 610 | assert_equal(compress_rowcols(x, 1), [[0, 2], [3, 5], [6, 8]]) |
| 611 | x = array(x._data, mask=[[1, 0, 0], [0, 1, 0], [0, 0, 0]]) |
| 612 | assert_equal(compress_rowcols(x), [[8]]) |
| 613 | assert_equal(compress_rowcols(x, 0), [[6, 7, 8]]) |
| 614 | assert_equal(compress_rowcols(x, 1,), [[2], [5], [8]]) |
| 615 | x = array(x._data, mask=[[1, 0, 0], [0, 1, 0], [0, 0, 1]]) |
| 616 | assert_equal(compress_rowcols(x).size, 0) |
| 617 | assert_equal(compress_rowcols(x, 0).size, 0) |
| 618 | assert_equal(compress_rowcols(x, 1).size, 0) |
| 619 | |
| 620 | def test_mask_rowcols(self): |
| 621 | # Tests mask_rowcols. |
nothing calls this directly
no test coverage detected