def makeTestCandidateCellSet(self):
"""Populate a SpatialCellSet"""
if False: # Print the bboxes for the cells
print()
for i in range(len(self.cellSet.getCellList())):
cell = self.cellSet.getCellList()[i]
print(
i, "%3d,%3d -- %3d,%3d" %
(cell.getBBox().getMinX(), cell.getBBox().getMinY(),
cell.getBBox().getMaxX(), cell.getBBox().getMaxY()),
cell.getLabel())
self.assertEqual(len(self.cellSet.getCellList()), 6)
self.NTestCandidates = 0 # number of candidates
for x, y in ([5, 0], [1, 1], [2, 2], [0, 0], [4,
4], [3,
4]): # all in cell0
self.cellSet.insertCandidate(testLib.TestCandidate(x, y, -x))
self.NTestCandidates += 1
self.cellSet.insertCandidate(testLib.TestCandidate(305, 0,
100)) # in cell1
self.NTestCandidates += 1
self.cellSet.insertCandidate(testLib.TestCandidate(
500, 500, 100)) # the top right corner of cell5
self.NTestCandidates += 1
def testBuildCandidateListByInsertion(self):
"""Build a candidate list by inserting candidates"""
self.cell = afwMath.SpatialCell("Test", afwGeom.Box2I())
for x, y in ([5, 0], [1, 1], [2, 2], [0, 0], [4, 4], [3, 4]):
self.cell.insertCandidate(testLib.TestCandidate(x, y, getFlux(x)))
self.assertEqual(self.cell[0].getXCenter(), 0)
def setUp(self):
candidateList = afwMath.SpatialCellCandidateList()
self.nCandidate = 5
for i in (0, 1, 4, 3, 2): # must be all numbers in range(self.nCandidate)
x, y = i, 5*i
candidateList.append(testLib.TestCandidate(x, y, getFlux(x)))
self.cell = afwMath.SpatialCell("Test", afwGeom.Box2I(), candidateList)
self.assertEqual(self.cell.getLabel(), "Test")
def tst():
self.cellSet.insertCandidate(testLib.TestCandidate(
501, 501, 100)) # Doesn't fit
