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_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_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)
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)
