def test_overscanCorrection_isNotInt(self):
"""Expect smaller median/smaller std after.
Expect exception if overscan fit type isn't known.
"""
inputExp = isrMock.RawMock().run()
amp = inputExp.getDetector()[0]
ampI = inputExp.maskedImage[amp.getRawDataBBox()]
overscanI = inputExp.maskedImage[amp.getRawHorizontalOverscanBBox()]
for fitType in ('MEAN', 'MEDIAN', 'MEANCLIP', 'POLY', 'CHEB',
'NATURAL_SPLINE', 'CUBIC_SPLINE', 'UNKNOWN'):
if fitType in ('NATURAL_SPLINE', 'CUBIC_SPLINE'):
order = 3
else:
order = 1
if fitType == 'UNKNOWN':
with self.assertRaises(
pexExcept.Exception,
msg=
f"overscanCorrection overscanIsNotInt fitType: {fitType}"
):
ipIsr.overscanCorrection(ampI,
overscanI,
fitType=fitType,
order=order,
collapseRej=3.0,
statControl=None,
overscanIsInt=False)
else:
response = ipIsr.overscanCorrection(ampI,
overscanI,
fitType=fitType,
order=order,
collapseRej=3.0,
statControl=None,
overscanIsInt=False)
self.assertIsInstance(
response,
pipeBase.Struct,
msg=
f"overscanCorrection overscanIsNotInt Bad response: {fitType}")
self.assertIsNotNone(
response.imageFit,
msg=
f"overscanCorrection overscanIsNotInt Bad imageFit: {fitType}")
self.assertIsNotNone(
response.overscanFit,
msg=
f"overscanCorrection overscanIsNotInt Bad overscanFit: {fitType}"
)
self.assertIsInstance(
response.overscanImage,
afwImage.MaskedImageF,
msg=
f"overscanCorrection overscanIsNotInt Bad overscanImage: {fitType}"
)
def test_untrimmedSimple(self):
"""Confirm untrimmed data classes are generated consistently.
"""
exposure = isrMock.RawMock().run()
fringe = isrMock.UntrimmedFringeMock().run()
initialStd = np.std(exposure.getMaskedImage().getImage().getArray()[:])
diff = (exposure.getMaskedImage().getImage().getArray()[:] -
fringe.getMaskedImage().getImage().getArray()[:])
newStd = np.std(diff[:])
self.assertLess(newStd, initialStd)
def setUp(self):
self.config = IsrTaskConfig()
self.config.qa = IsrQaConfig()
self.task = IsrTask(config=self.config)
self.mockConfig = isrMock.IsrMockConfig()
self.mockConfig.isTrimmed = False
self.doGenerateImage = True
self.dataRef = isrMock.DataRefMock(config=self.mockConfig)
self.camera = isrMock.IsrMock(config=self.mockConfig).getCamera()
self.inputExp = isrMock.RawMock(config=self.mockConfig).run()
self.amp = self.inputExp.getDetector()[0]
self.mi = self.inputExp.getMaskedImage()
def testAssembleCcdTask_exp_doTrim(self):
self.assertEqual(self.runTest(inputExp=isrMock.RawMock().run(), doTrim=True).getBBox(),
self.outputExp.getBBox())
def setUp(self):
self.outputExp = isrMock.TrimmedRawMock().run()
self.outputUntrimmedExp = isrMock.RawMock().run()
