1 from __future__
import absolute_import, division, print_function
2 from future.utils
import with_metaclass
29 __all__ = [
"ButlerGetTests"]
34 Tests of obs_* Butler get() functionality. 36 In the subclasses's setUp(): 37 * Call setUp_butler_get() to fill in required parameters. 41 ccdExposureId_bits=None,
47 detector_serials=None,
51 good_detectorIds=None,
56 Set up the necessary variables for butlerGet tests. 58 All "exposure name" entries below should correspond to an entry in 64 ccdExposureId_bits : `int` 65 expected value of ccdExposureId_bits 67 dict of exposure name : ccdExposureId (the number as returned by the butler) 69 dict of exposure name : filter name 71 dict of exposure name : exposure time 72 detector_names : `dict` 73 dict of exposure name : detector name 75 dict of exposure name : detectorId 76 detector_serials : `dict` 77 dict of exposure name : detector serial 79 dict of exposure name : dimensions (as a geom.Extent2I) 80 sky_origin : `tuple` of `float` 81 Longitude, Latitude of 'raw' exposure 82 raw_subsets : `tuple` of (kwargs, `int`) 83 keyword args and expected number of subsets for butler.subset('raw', **kwargs) 84 good_detectorIds : `list` of `int` 85 list of valid ccd numbers 86 bad_detectorIds : `list` of `int` 87 list of invalid ccd numbers 88 linearizer_type : `dict` 89 dict of detectorId (usually `int`): LinearizerType 90 (e.g. lsst.ip.isr.LinearizeLookupTable.LinearityType), 91 or unittest.SkipTest to skip all linearizer tests. 94 fields = [
'ccdExposureId_bits',
108 ButlerGet = collections.namedtuple(
"ButlerGetData", fields)
111 exposureIds=exposureIds,
114 detectorIds=detectorIds,
115 detector_names=detector_names,
116 detector_serials=detector_serials,
117 dimensions=dimensions,
118 sky_origin=sky_origin,
119 raw_subsets=raw_subsets,
120 good_detectorIds=good_detectorIds,
121 bad_detectorIds=bad_detectorIds,
122 linearizer_type=linearizer_type
126 bits = self.butler.get(
'ccdExposureId_bits')
129 def _test_exposure(self, name):
130 if self.dataIds[name]
is unittest.SkipTest:
131 self.skipTest(
'Skipping %s as requested' % (inspect.currentframe().f_code.co_name))
133 exp = self.butler.get(name, self.dataIds[name])
134 self.assertEqual(exp.getDimensions(), self.
butler_get_data.dimensions[name])
135 self.assertEqual(exp.getDetector().getId(), self.
butler_get_data.detectorIds[name])
136 self.assertEqual(exp.getDetector().getName(), self.
butler_get_data.detector_names[name])
137 self.assertEqual(exp.getDetector().getSerial(), self.
butler_get_data.detector_serials[name])
138 self.assertEqual(exp.getFilter().getName(), self.
butler_get_data.filters[name])
139 exposureId = self.butler.get(
'ccdExposureId', dataId=self.dataIds[name])
141 self.assertEqual(exp.getInfo().getVisitInfo().getExposureTime(), self.
butler_get_data.exptimes[name])
149 self.assertEqual(exp.hasWcs(),
True)
150 origin = exp.getWcs().getSkyOrigin()
151 self.assertAlmostEqual(origin.getLongitude().asDegrees(), self.
butler_get_data.sky_origin[0])
152 self.assertAlmostEqual(origin.getLatitude().asDegrees(), self.
butler_get_data.sky_origin[1])
163 @unittest.skip(
'Cannot test this, as there is a bug in the butler! DM-8097')
165 exp = self.butler.get(
'raw', self.dataIds[
'raw'], immediate=
True)
168 sub = self.butler.get(
"raw_sub", self.dataIds[
'raw'], bbox=bbox, immediate=
True)
169 self.assertEqual(sub.getImage().getBBox(), bbox)
170 self.assertImagesEqual(sub, exp.Factory(exp, bbox))
174 subset = self.butler.subset(
"raw", **kwargs)
175 self.assertEqual(len(subset), expect, msg=
"Failed for kwargs: {}".format(kwargs))
178 """Test that we can get a linearizer for good detectorIds.""" 180 self.skipTest(
'Skipping %s as requested' % (inspect.currentframe().f_code.co_name))
182 camera = self.butler.get(
"camera")
184 detector = camera[detectorId]
185 linearizer = self.butler.get(
"linearizer", dataId=dict(ccd=detectorId), immediate=
True)
186 self.assertEqual(linearizer.LinearityType, self.
butler_get_data.linearizer_type[detectorId])
187 linearizer.checkDetector(detector)
190 """Do bad detectorIds raise?""" 192 self.skipTest(
'Skipping %s as requested' % (inspect.currentframe().f_code.co_name))
195 with self.assertRaises(RuntimeError):
196 self.butler.get(
"linearizer", dataId=dict(ccd=badccd), immediate=
True)
def test_exposureId_bits(self)
def test_raw_sub_bbox(self)
def _test_exposure(self, name)
def test_subset_raw(self)
def test_get_linearizer_bad_detectorIds(self)
def setUp_butler_get(self, ccdExposureId_bits=None, exposureIds=None, filters=None, exptimes=None, detectorIds=None, detector_names=None, detector_serials=None, dimensions=None, sky_origin=None, raw_subsets=None, good_detectorIds=None, bad_detectorIds=None, linearizer_type=None)
def test_get_linearizer(self)