MCPcopy Index your code
hub / github.com/numpy/numpy / test_mask_rowcols

Method test_mask_rowcols

numpy/ma/tests/test_extras.py:708–738  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

706 assert_equal(compress_rowcols(x, 1).size, 0)
707
708 def test_mask_rowcols(self):
709 # Tests mask_rowcols.
710 x = array(np.arange(9).reshape(3, 3),
711 mask=[[1, 0, 0], [0, 0, 0], [0, 0, 0]])
712 assert_equal(mask_rowcols(x).mask,
713 [[1, 1, 1], [1, 0, 0], [1, 0, 0]])
714 assert_equal(mask_rowcols(x, 0).mask,
715 [[1, 1, 1], [0, 0, 0], [0, 0, 0]])
716 assert_equal(mask_rowcols(x, 1).mask,
717 [[1, 0, 0], [1, 0, 0], [1, 0, 0]])
718 x = array(x._data, mask=[[0, 0, 0], [0, 1, 0], [0, 0, 0]])
719 assert_equal(mask_rowcols(x).mask,
720 [[0, 1, 0], [1, 1, 1], [0, 1, 0]])
721 assert_equal(mask_rowcols(x, 0).mask,
722 [[0, 0, 0], [1, 1, 1], [0, 0, 0]])
723 assert_equal(mask_rowcols(x, 1).mask,
724 [[0, 1, 0], [0, 1, 0], [0, 1, 0]])
725 x = array(x._data, mask=[[1, 0, 0], [0, 1, 0], [0, 0, 0]])
726 assert_equal(mask_rowcols(x).mask,
727 [[1, 1, 1], [1, 1, 1], [1, 1, 0]])
728 assert_equal(mask_rowcols(x, 0).mask,
729 [[1, 1, 1], [1, 1, 1], [0, 0, 0]])
730 assert_equal(mask_rowcols(x, 1,).mask,
731 [[1, 1, 0], [1, 1, 0], [1, 1, 0]])
732 x = array(x._data, mask=[[1, 0, 0], [0, 1, 0], [0, 0, 1]])
733 assert_(mask_rowcols(x).all() is masked)
734 assert_(mask_rowcols(x, 0).all() is masked)
735 assert_(mask_rowcols(x, 1).all() is masked)
736 assert_(mask_rowcols(x).mask.all())
737 assert_(mask_rowcols(x, 0).mask.all())
738 assert_(mask_rowcols(x, 1).mask.all())
739
740 @pytest.mark.parametrize("axis", [None, 0, 1])
741 @pytest.mark.parametrize(["func", "rowcols_axis"],

Callers

nothing calls this directly

Calls 6

arrayFunction · 0.90
assert_equalFunction · 0.90
mask_rowcolsFunction · 0.90
assert_Function · 0.85
reshapeMethod · 0.80
allMethod · 0.45

Tested by

no test coverage detected