def testNoInputImages(self):
"""Test that providing no images returns nothing.
"""
self.localSetUp(mockFWHMs=[])
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList)
self.assertEqual(len(result.exposureInfoList), 0)
def testSelectDefaultMakeDataRefListIsFalse(self):
"""Test appropriate behavior for makeDataRefList = False.
"""
self.localSetUp(
mockFWHMs=[self.avgFwhm for i in range(self.config.nImagesMax)])
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList,
makeDataRefList=False)
self.assertIsNone(result.dataRefList)
def testNImagesLessThanNMax(self):
"""Test case of input selectDataList shorter than nImagesMax.
"""
self.config.nImagesMax = 5
self.localSetUp(mockFWHMs=[self.avgFwhm for i in range(3)])
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList)
self.assertEqual(len(result.exposureInfoList), 3)
for dataRef in result.dataRefList:
self.assertEqual(dataRef.fwhm, self.avgFwhm)
def testSelectDefault(self):
"""Test that the default configuration is self-consistent.
"""
self.localSetUp(
mockFWHMs=[self.avgFwhm for i in range(self.config.nImagesMax)])
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList)
self.assertEqual(len(result.exposureInfoList), self.config.nImagesMax)
for dataRef in result.dataRefList:
self.assertEqual(dataRef.fwhm, self.avgFwhm)
def testSelectOne(self):
"""Test that configs requesting one image return one image.
"""
self.config.nImagesMax = 1
self.localSetUp(
mockFWHMs=[self.avgFwhm for i in range(self.config.nImagesMax)])
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList)
self.assertEqual(len(result.exposureInfoList), 1)
self.assertEqual(result.exposureInfoList[0].dataId,
self.makeDataId(self.avgFwhm))
self.assertEqual(result.dataRefList[0].fwhm, self.avgFwhm)
def testBestSeeingFwhmSelection(self):
"""Test that the selection picks the best seeing images.
"""
self.config.minPsfFwhm = 1.0
self.config.maxPsfFwhm = 1.5
self.config.nImagesMax = 3
mockFWHMs = [0.5, 1.0, 1.1, 1.2, 1.25, 1.3, 1.4, 1.5, 2.0]
self.localSetUp(mockFWHMs=mockFWHMs)
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList)
self.assertEqual(len(result.exposureInfoList), self.config.nImagesMax)
for i, fwhm in enumerate([1.0, 1.1, 1.2]):
self.assertEqual(result.exposureInfoList[i].dataId,
self.makeDataId(fwhm))
self.assertEqual(result.dataRefList[i].fwhm, fwhm)
def testFwhmSelection(self):
"""Test that minimum and maximum FWHM selections work.
"""
self.config.minPsfFwhm = 1.0
self.config.maxPsfFwhm = 1.5
self.config.nImagesMax = 10
mockFWHMs = [0.5, 1.0, 1.25, 1.5, 2.0]
self.localSetUp(mockFWHMs=mockFWHMs)
task = selectImages.BestSeeingWcsSelectImagesTask(config=self.config)
result = task.runDataRef(self.dataRef,
self.coordList,
selectDataList=self.selectList)
self.assertEqual(len(result.exposureInfoList), 3)
for i, fwhm in enumerate([1.0, 1.25, 1.5]):
self.assertEqual(result.exposureInfoList[i].dataId,
self.makeDataId(fwhm))
self.assertEqual(result.dataRefList[i].fwhm, fwhm)
