def test_masking_multiple_3D(self): im = np.ones((9, 9, 9)) radius = 1 # N pix is 7 sli = mask_image([[4, 4, 4], [4, 4, 4]], im, radius) assert_equal(sli.sum(), 7) assert_equal(sli.shape, im.shape) sli = mask_image([[4, 4, 6], [4, 4, 2]], im, radius) assert_equal(sli.sum(), 14) sli = mask_image([[4, 4, 0], [4, 4, 4]], im, radius) assert_equal(sli.sum(), 13)
def test_masking_multiple_2D(self): im = np.ones((9, 9)) radius = 1 # N pix is 5 sli = mask_image([[4, 2], [4, 6]], im, radius) assert_equal(sli.sum(), 10) sli = mask_image([[4, 4], [4, 4]], im, radius) assert_equal(sli.sum(), 5) sli = mask_image([[0, 4], [4, 4]], im, radius) assert_equal(sli.sum(), 9) sli = mask_image([[-1, 4], [4, 4]], im, radius) assert_equal(sli.sum(), 6) sli = mask_image([[-20, 40], [4, 4]], im, radius) assert_equal(sli.sum(), 5)
def test_masking_single_3D(self): im = np.ones((9, 9, 9)) radius = 1 # N pix is 7 sli = mask_image([4, 4, 4], im, radius) assert_equal(sli.sum(), 7) assert_equal(sli.shape, im.shape) sli = mask_image([0, 4, 4], im, radius) assert_equal(sli.sum(), 6) sli = mask_image([4, 0, 0], im, radius) assert_equal(sli.sum(), 5) sli = mask_image([0, 0, 0], im, radius) assert_equal(sli.sum(), 4) sli = mask_image([-1, 4, 4], im, radius) assert_equal(sli.sum(), 1) sli = mask_image([-1, -1, -1], im, radius) assert_equal(sli.sum(), 0)