Coverage for tests/test_phosim.py: 35%
63 statements
« prev ^ index » next coverage.py v6.4.1, created at 2022-06-24 02:50 -0700
« prev ^ index » next coverage.py v6.4.1, created at 2022-06-24 02:50 -0700
1# This file is part of obs_lsst.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (http://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 <http://www.gnu.org/licenses/>.
21#
22import os
23import sys
24import unittest
25import unittest.mock
27import lsst.utils.tests
28from lsst.geom import arcseconds, Extent2I
29import lsst.afw.image
31from lsst.obs.lsst.testHelper import ObsLsstButlerTests, ObsLsstObsBaseOverrides
32from lsst.obs.lsst import LsstCamPhoSim
34TESTDIR = os.path.abspath(os.path.dirname(__file__))
37def _clean_metadata_provenance(hdr):
38 """Remove metadata fix up provenance."""
39 for k in hdr:
40 if k.startswith("HIERARCH ASTRO METADATA"):
41 del hdr[k]
44class TestPhosim(ObsLsstObsBaseOverrides, ObsLsstButlerTests):
45 instrumentDir = "phosim"
47 @classmethod
48 def getInstrument(cls):
49 return LsstCamPhoSim()
51 def setUp(self):
52 dataIds = {'raw': {'exposure': 204595, 'name_in_raft': 'S20', 'raft': 'R11'},
53 'bias': unittest.SkipTest,
54 'flat': unittest.SkipTest,
55 'dark': unittest.SkipTest
56 }
57 self.setUp_tests(self._butler, None, dataIds)
59 ccdExposureId_bits = 34
60 exposureIds = {'raw': 204595042}
61 filters = {'raw': 'i'}
62 exptimes = {'raw': 30.0}
63 detectorIds = {'raw': 42}
64 detector_names = {'raw': 'R11_S20'}
65 detector_serials = {'raw': 'ITL-3800C-102-Dev'}
66 dimensions = {'raw': Extent2I(4176, 4020)}
67 sky_origin = (55.67759886, -30.44239357)
68 raw_subsets = (({}, 1),
69 ({'physical_filter': 'i'}, 1),
70 ({'physical_filter': 'foo'}, 0),
71 ({'exposure': 204595}, 1),
72 )
73 linearizer_type = unittest.SkipTest
74 self.setUp_butler_get(ccdExposureId_bits=ccdExposureId_bits,
75 exposureIds=exposureIds,
76 filters=filters,
77 exptimes=exptimes,
78 detectorIds=detectorIds,
79 detector_names=detector_names,
80 detector_serials=detector_serials,
81 dimensions=dimensions,
82 sky_origin=sky_origin,
83 raw_subsets=raw_subsets,
84 linearizer_type=linearizer_type
85 )
87 self.raw_filename = '00204595-R11-S20-det042.fits'
89 self.setUp_camera(camera_name='LSSTCam-PhoSim',
90 n_detectors=201,
91 first_detector_name='R01_S00',
92 plate_scale=20.0 * arcseconds,
93 )
95 super().setUp()
97 def testMetadata(self):
98 """Check that we can read headers properly"""
99 dataId = {'exposure': 204595, 'name_in_raft': 'S20', 'raft': 'R11'}
100 md = self.butler.get("raw.metadata", dataId)
102 # This header comes from amp header
103 self.assertEqual(md["PRESS"], 520.0)
105 # This header is in HDU0
106 self.assertEqual(md["TESTTYPE"], "PHOSIM")
108 visitInfo = self.butler.get("raw.visitInfo", dataId)
109 weather = visitInfo.getWeather()
110 self.assertEqual(weather.getAirTemperature(), 20.0)
112 # Check that corrections are being applied
113 with unittest.mock.patch.dict(os.environ,
114 {"METADATA_CORRECTIONS_PATH": os.path.join(TESTDIR, "data")}):
115 # Check that corrections are applied during simple md get
116 md_md = self.butler.get("raw.metadata", dataId)
117 self.assertEqual(md_md["NEWHDR"], "corrected")
119 # Check that corrections are applied if we do assembly
120 raw = self.butler.get("raw", dataId)
121 raw_md = raw.getMetadata()
123 _clean_metadata_provenance(raw_md)
124 _clean_metadata_provenance(md_md)
126 self.assertEqual(raw_md, md_md)
128 # And finally ensure that visitInfo gets corrections
129 visitInfo = self.butler.get("raw.visitInfo", dataId)
130 weather = visitInfo.getWeather()
131 self.assertEqual(weather.getAirTemperature(), 10.0)
134class MemoryTester(lsst.utils.tests.MemoryTestCase):
135 pass
138def setup_module(module):
139 lsst.utils.tests.init()
142if __name__ == '__main__': 142 ↛ 143line 142 didn't jump to line 143, because the condition on line 142 was never true
143 setup_module(sys.modules[__name__])
144 unittest.main()