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