def test_even(self):
"""Test roi with even side length"""
actual = cm.circle_mask((4, 4), 2)
expected = np.array([[False, True, True, False],
[True, True, True, True],
[True, True, True, True],
[False, True, True, False]])
self.assertIsNone(np.testing.assert_array_equal(actual, expected))
def test_offcenter(self):
"""Test specifying off center location"""
actual = cm.circle_mask((5, 5), 2, center=(2, 3))
expected = np.array([[False, False, False, True, False],
[False, False, True, True, True],
[False, True, True, True, True],
[False, False, True, True, True],
[False, False, False, True, False]])
self.assertIsNone(np.testing.assert_array_equal(actual, expected))
def test_mask(self):
shape, rad = self.roi.roi.shape, self.roi.rad
mask = circle_mask(shape, rad)
masked_roi = self.roi.mask(mask)
self.assertTrue(np.isnan(masked_roi[shape[0]//2, shape[1]//2]))
self.assertTrue(np.isnan(self.roi.roi[shape[0]//2, shape[1]//2]))
def test_radius1(self):
"""Test radius 1"""
actual = cm.circle_mask((3, 3), 1)
expected = np.array([[False, True, False], [True, True, True],
[False, True, False]])
self.assertIsNone(np.testing.assert_array_equal(actual, expected))