def testCircle(self):
"""Test circle_fill with different input parameters"""
square3x3 = numpy.array(
((-1, -1, -1, 0, 0, 0, 1, 1, 1), (-1, 0, 1, -1, 0, 1, -1, 0, 1)))
tests = [
#crow, ccol, radius, ref_coords = (ref_rows, ref_cols)
(0, 0, 1, ((0, ), (0, ))),
(10, 15, 1, ((10, ), (15, ))),
(0, 0, 1.5, square3x3),
(5, 10, 2, (5 + square3x3[0], 10 + square3x3[1])),
(10, 20, 3.5, (10 + numpy.array(
(-3, -3, -3, -2, -2, -2, -2, -2, -1, -1, -1, -1, -1, -1, -1, 0,
0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3,
3)), 20 + numpy.array(
(-1, 0, 1, -2, -1, 0, 1, 2, -3, -2, -1, 0, 1, 2, 3, -3,
-2, -1, 0, 1, 2, 3, -3, -2, -1, 0, 1, 2, 3, -2, -1, 0, 1,
2, -1, 0, 1)))),
]
for crow, ccol, radius, ref_coords in tests:
with self.subTest(crow=crow, ccol=ccol, radius=radius):
coords = shapes.circle_fill(crow, ccol, radius)
is_equal = numpy.all(numpy.equal(coords, ref_coords))
if not is_equal:
_logger.debug('result:\n%s', str(coords))
_logger.debug('ref:\n%s', str(ref_coords))
self.assertTrue(is_equal)
def updateDisk(self, level, crow, ccol, radius, mask=True):
"""Mask/Unmask a disk of the given mask level.
:param int level: Mask level to update.
:param int crow: Disk center row.
:param int ccol: Disk center column.
:param float radius: Radius of the disk in mask array unit
:param bool mask: True to mask (default), False to unmask.
"""
rows, cols = shapes.circle_fill(crow, ccol, radius)
self.updatePoints(level, rows, cols, mask)
def updateDisk(self, level, crow, ccol, radius, mask=True):
"""Mask/Unmask a disk of the given mask level.
:param int level: Mask level to update.
:param int crow: Disk center row.
:param int ccol: Disk center column.
:param float radius: Radius of the disk in mask array unit
:param bool mask: True to mask (default), False to unmask.
"""
rows, cols = shapes.circle_fill(crow, ccol, radius)
self.updatePoints(level, rows, cols, mask)
def testCircle(self):
"""Test circle_fill with different input parameters"""
square3x3 = numpy.array(((-1, -1, -1, 0, 0, 0, 1, 1, 1),
(-1, 0, 1, -1, 0, 1, -1, 0, 1)))
tests = [
# crow, ccol, radius, ref_coords = (ref_rows, ref_cols)
(0, 0, 1, ((0,), (0,))),
(10, 15, 1, ((10,), (15,))),
(0, 0, 1.5, square3x3),
(5, 10, 2, (5 + square3x3[0], 10 + square3x3[1])),
(10, 20, 3.5, (
10 + numpy.array((-3, -3, -3,
-2, -2, -2, -2, -2,
-1, -1, -1, -1, -1, -1, -1,
0, 0, 0, 0, 0, 0, 0,
1, 1, 1, 1, 1, 1, 1,
2, 2, 2, 2, 2,
3, 3, 3)),
20 + numpy.array((-1, 0, 1,
-2, -1, 0, 1, 2,
-3, -2, -1, 0, 1, 2, 3,
-3, -2, -1, 0, 1, 2, 3,
-3, -2, -1, 0, 1, 2, 3,
-2, -1, 0, 1, 2,
-1, 0, 1)))),
]
for crow, ccol, radius, ref_coords in tests:
with self.subTest(crow=crow, ccol=ccol, radius=radius):
coords = shapes.circle_fill(crow, ccol, radius)
is_equal = numpy.all(numpy.equal(coords, ref_coords))
if not is_equal:
_logger.debug('result:\n%s', str(coords))
_logger.debug('ref:\n%s', str(ref_coords))
self.assertTrue(is_equal)
