Coverage for tests/test_fitsRawFormatter.py : 44%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1# This file is part of obs_base.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (https://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This program is free software: you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation, either version 3 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program. If not, see <https://www.gnu.org/licenses/>.
22import unittest
24from astropy.coordinates import Angle
25import astropy.units as u
27import lsst.utils.tests
29from astro_metadata_translator import FitsTranslator, StubTranslator
30from astro_metadata_translator.translators.helpers import tracking_from_degree_headers
31from lsst.afw.cameraGeom.testUtils import CameraWrapper, DetectorWrapper
32import lsst.afw.geom
33import lsst.daf.base
34import lsst.daf.butler
35import lsst.geom
36from lsst.obs.base import FitsRawFormatterBase, MakeRawVisitInfoViaObsInfo, FilterDefinitionCollection
37from lsst.obs.base.utils import createInitialSkyWcs, InitialSkyWcsError
40class SimpleTestingTranslator(FitsTranslator, StubTranslator):
41 _const_map = {"boresight_rotation_angle": Angle(90*u.deg),
42 "boresight_rotation_coord": "sky",
43 "detector_exposure_id": 12345,
44 # The following are defined to prevent warnings about
45 # undefined translators
46 "dark_time": 0.0*u.s,
47 "exposure_time": 0.0*u.s,
48 "physical_filter": "u",
49 "detector_num": 0,
50 "detector_name": "0",
51 "detector_group": "",
52 "detector_unique_name": "0",
53 "detector_serial": "",
54 "observation_id": "--",
55 "science_program": "unknown",
56 "object": "unknown",
57 "exposure_id": 0,
58 "visit_id": 0,
59 "relative_humidity": 30.0,
60 "pressure": 0.0*u.MPa,
61 "temperature": 273*u.K,
62 "altaz_begin": None,
63 }
64 _trivial_map = {"boresight_airmass": "AIRMASS",
65 "observation_type": "OBSTYPE"}
67 def to_tracking_radec(self):
68 radecsys = ("RADESYS", )
69 radecpairs = (("RA", "DEC"),)
70 return tracking_from_degree_headers(self, radecsys, radecpairs, unit=(u.deg, u.deg))
73class MakeTestingRawVisitInfo(MakeRawVisitInfoViaObsInfo):
74 metadataTranslator = SimpleTestingTranslator
77class SimpleFitsRawFormatter(FitsRawFormatterBase):
78 filterDefinitions = FilterDefinitionCollection()
80 @property
81 def translatorClass(self):
82 return SimpleTestingTranslator
84 def getDetector(self, id):
85 """Use CameraWrapper to create a fake detector that can map from
86 PIXELS to FIELD_ANGLE.
88 Always return Detector #10, so all the tests are self-consistent.
89 """
90 return CameraWrapper().camera.get(10)
93class FitsRawFormatterTestCase(lsst.utils.tests.TestCase):
94 def setUp(self):
95 # reset the filters before we test anything
96 FilterDefinitionCollection.reset()
98 # The FITS WCS and VisitInfo coordinates in this header are intentionally
99 # different, to make comparisons between them more obvious.
100 self.boresight = lsst.geom.SpherePoint(10., 20., lsst.geom.degrees)
101 self.header = {
102 "TELESCOP": "TEST",
103 "INSTRUME": "UNKNOWN",
104 "AIRMASS": 1.2,
105 "RADESYS": "ICRS",
106 "OBSTYPE": "science",
107 "EQUINOX": 2000,
108 "OBSGEO-X": "-5464588.84421314",
109 "OBSGEO-Y": "-2493000.19137644",
110 "OBSGEO-Z": "2150653.35350771",
111 "RA": self.boresight.getLatitude().asDegrees(),
112 "DEC": self.boresight.getLongitude().asDegrees(),
113 "CTYPE1": "RA---SIN",
114 "CTYPE2": "DEC--SIN",
115 "CRPIX1": 5,
116 "CRPIX2": 6,
117 "CRVAL1": self.boresight.getLatitude().asDegrees() + 1,
118 "CRVAL2": self.boresight.getLongitude().asDegrees() + 1,
119 "CD1_1": 1e-5,
120 "CD1_2": 0,
121 "CD2_2": 1e-5,
122 "CD2_1": 0
123 }
124 # make a property list of the above, for use by the formatter.
125 self.metadata = lsst.daf.base.PropertyList()
126 self.metadata.update(self.header)
128 maker = MakeTestingRawVisitInfo()
129 self.visitInfo = maker(self.header)
131 self.metadataSkyWcs = lsst.afw.geom.makeSkyWcs(self.metadata, strip=False)
132 self.boresightSkyWcs = createInitialSkyWcs(self.visitInfo, CameraWrapper().camera.get(10))
134 # set these to `contextlib.nullcontext()` to print the log warnings
135 self.warnContext = self.assertLogs(level="WARNING")
136 self.logContext = lsst.log.UsePythonLogging()
138 # We have no file in these tests, so make an empty descriptor.
139 fileDescriptor = lsst.daf.butler.FileDescriptor(None, None)
140 self.formatter = SimpleFitsRawFormatter(fileDescriptor)
141 # Force the formatter's metadata to be what we've created above.
142 self.formatter._metadata = self.metadata
144 def test_makeWcs(self):
145 detector = self.formatter.getDetector(1)
146 wcs = self.formatter.makeWcs(self.visitInfo, detector)
147 self.assertNotEqual(wcs, self.metadataSkyWcs)
148 self.assertEqual(wcs, self.boresightSkyWcs)
150 def test_makeWcs_warn_if_metadata_is_bad(self):
151 """If the metadata is bad, log a warning and use the VisitInfo WCS.
152 """
153 detector = self.formatter.getDetector(1)
154 self.metadata.remove("CTYPE1")
155 with self.warnContext, self.logContext:
156 wcs = self.formatter.makeWcs(self.visitInfo, detector)
157 self.assertNotEqual(wcs, self.metadataSkyWcs)
158 self.assertEqual(wcs, self.boresightSkyWcs)
160 def test_makeWcs_warn_if_visitInfo_is_None(self):
161 """If VisitInfo is None, log a warning and use the metadata WCS.
162 """
163 detector = self.formatter.getDetector(1)
164 with self.warnContext, self.logContext:
165 wcs = self.formatter.makeWcs(None, detector)
166 self.assertEqual(wcs, self.metadataSkyWcs)
167 self.assertNotEqual(wcs, self.boresightSkyWcs)
169 def test_makeWcs_fail_if_visitInfo_is_None(self):
170 """If VisitInfo is None and metadata failed, raise an exception.
171 """
172 detector = self.formatter.getDetector(1)
173 self.metadata.remove("CTYPE1")
174 with self.warnContext, self.logContext, self.assertRaises(InitialSkyWcsError):
175 self.formatter.makeWcs(None, detector)
177 def test_makeWcs_fail_if_detector_is_bad(self):
178 """If Detector is broken, raise an exception.
179 """
180 # This detector doesn't know about FIELD_ANGLE, so can't be used to
181 # make a SkyWcs.
182 detector = DetectorWrapper().detector
183 with self.assertRaises(InitialSkyWcsError):
184 self.formatter.makeWcs(self.visitInfo, detector)
187class MemoryTester(lsst.utils.tests.MemoryTestCase):
188 pass
191def setup_module(module):
192 lsst.utils.tests.init()
195if __name__ == '__main__': 195 ↛ 196line 195 didn't jump to line 196, because the condition on line 195 was never true
196 lsst.utils.tests.init()
197 unittest.main()