Coverage for tests/test_lsstCam.py: 22%
80 statements
« prev ^ index » next coverage.py v7.1.0, created at 2023-02-05 18:52 -0800
« prev ^ index » next coverage.py v7.1.0, created at 2023-02-05 18:52 -0800
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
26import lsst.utils.tests
27from lsst.geom import arcseconds, Extent2I
28import lsst.afw.image
30from lsst.obs.lsst.testHelper import ObsLsstButlerTests, ObsLsstObsBaseOverrides
33class TestLsstCam(ObsLsstObsBaseOverrides, ObsLsstButlerTests):
34 instrumentDir = "lsstCam"
36 def setUp(self):
37 dataIds = {'raw': {'expId': 3019031900001, 'detectorName': 'S02', 'raftName': 'R10'},
38 'bias': unittest.SkipTest,
39 'flat': unittest.SkipTest,
40 'dark': unittest.SkipTest,
41 }
42 self.setUp_tests(self._butler, self._mapper, dataIds)
44 ccdExposureId_bits = 52
45 exposureIds = {'raw': 3019031900001029,
46 }
47 filters = {'raw': 'unknown',
48 }
49 exptimes = {'raw': 0.0,
50 }
51 detectorIds = {'raw': 29,
52 }
53 detector_names = {'raw': 'R10_S02',
54 }
55 # This name comes from the camera and not from the butler
56 detector_serials = {'raw': 'ITL-3800C-167',
57 }
58 dimensions = {'raw': Extent2I(4608, 4096),
59 }
60 sky_origin = unittest.SkipTest
61 raw_subsets = (({'level': 'sensor'}, 3),
62 ({'level': 'sensor', 'filter': 'unknown'}, 2),
63 ({'level': 'sensor', 'filter': 'foo'}, 0),
64 ({'level': 'sensor', 'expId': 3019031900001}, 2),
65 ({'level': 'sensor', 'expId': 3019032200002}, 1),
66 ({'level': 'sensor', 'expId': 9999999999999}, 0),
67 ({'level': 'filter'}, 2),
68 ({'level': 'filter', 'expId': 3019031900001}, 1),
69 ({'level': 'filter', 'expId': 3019032200002}, 1),
70 ({'level': 'expId'}, 2),
71 ({'level': 'expId', 'filter': 'unknown'}, 1),
72 ({'level': 'expId', 'filter': 'SDSSi~ND_OD0.5'}, 1),
73 ({'level': 'expId', 'filter': 'foo'}, 0)
74 )
75 linearizer_type = unittest.SkipTest
76 self.setUp_butler_get(ccdExposureId_bits=ccdExposureId_bits,
77 exposureIds=exposureIds,
78 filters=filters,
79 exptimes=exptimes,
80 detectorIds=detectorIds,
81 detector_names=detector_names,
82 detector_serials=detector_serials,
83 dimensions=dimensions,
84 sky_origin=sky_origin,
85 raw_subsets=raw_subsets,
86 linearizer_type=linearizer_type
87 )
89 path_to_raw = os.path.join(self.data_dir, "raw", "30190319", "R10",
90 "3019031900001-R10-S02-det029.fits")
91 keys = set(('filter', 'patch', 'tract', 'visit', 'channel', 'amp', 'style', 'detector', 'dstype',
92 'snap', 'run', 'calibDate', 'half', 'detectorName', 'raftName', 'label', 'expId',
93 'numSubfilters', 'fgcmcycle', 'name', 'pixel_id', 'description', 'subfilter',
94 'dayObs', 'seqNum', 'subdir',))
95 query_format = ["expId", "filter"]
96 queryMetadata = (({'expId': 3019031900001}, [(3019031900001, 'unknown')]),
97 ({'filter': 'unknown'}, [(3019031900001, 'unknown')]),
98 )
99 map_python_type = lsst.afw.image.DecoratedImageF
100 map_python_std_type = lsst.afw.image.ExposureF
101 map_cpp_type = 'DecoratedImageF'
102 map_storage_name = 'FitsStorage'
103 metadata_output_path = os.path.join("processCcd_metadata/3019031900001-unknown/R10",
104 "processCcdMetadata_3019031900001-unknown-R10-S02-det029.yaml")
105 raw_filename = '3019031900001-R10-S02-det029.fits'
106 default_level = 'sensor'
107 raw_levels = (('sensor', set(['expId', 'detector', 'dayObs', 'detectorName', 'raftName'])),
108 ('skyTile', set(['expId', 'dayObs'])),
109 ('filter', set(['expId'])),
110 ('expId', set(['expId', 'dayObs']))
111 )
112 self.setUp_mapper(output=self.data_dir,
113 path_to_raw=path_to_raw,
114 keys=keys,
115 query_format=query_format,
116 queryMetadata=queryMetadata,
117 metadata_output_path=metadata_output_path,
118 map_python_type=map_python_type,
119 map_python_std_type=map_python_std_type,
120 map_cpp_type=map_cpp_type,
121 map_storage_name=map_storage_name,
122 raw_filename=raw_filename,
123 default_level=default_level,
124 raw_levels=raw_levels,
125 test_config_metadata=True
126 )
128 self.setUp_camera(camera_name='LSSTCam',
129 n_detectors=205,
130 first_detector_name='R01_S00',
131 plate_scale=20.0 * arcseconds,
132 )
134 super().setUp()
136 def testObsid(self):
137 """Check that we can retrieve data using the obsid."""
138 dataId = {'obsid': "MC_C_20190319_000001", 'detectorName': 'S02',
139 'raftName': 'R10'}
140 raw = self.butler.get('raw', dataId)
141 self.assertIsNotNone(raw)
143 # And that we can get just the header
144 md = self.butler.get('raw_md', dataId)
145 self.assertEqual(md["TELESCOP"], "LSST")
147 def testCcdExposureId(self):
148 with self.assertRaises(KeyError):
149 self.butler.get('ccdExposureId', dataId={})
151 exposureId = self.butler.get('ccdExposureId', dataId={"visit": 1, "detector": 1})
152 self.assertEqual(exposureId, 1001)
154 exposureId = self.butler.get('ccdExposureId', dataId={"visit": 1, "raftName": "R01",
155 "detectorName": "S01"})
156 self.assertEqual(exposureId, 1001)
158 with self.assertRaises(ValueError):
159 self.butler.get('ccdExposureId', dataId={"visit": 1, "detector": 2000})
161 with self.assertRaises(KeyError):
162 self.butler.get('ccdExposureId', dataId={"visit": 1})
164 with self.assertRaises(ValueError):
165 self.butler.get('ccdExposureId', dataId={"visit": 1, "raftName": "R99",
166 "detectorName": "S01"})
168 def testDetectorName(self):
169 name = self.mapper._extractDetectorName({"raftName": "R00", "detectorName": "S00"})
170 self.assertEqual(name, "R00_S00")
172 name = self.mapper._extractDetectorName({'visit': 3019031900001, 'detectorName': 'S02'})
173 self.assertEqual(name, "R10_S02")
175 name = self.mapper._extractDetectorName({'visit': 3019031900001, 'detector': 29})
176 self.assertEqual(name, "R10_S02")
178 name = self.mapper._extractDetectorName({'expId': 3019031900001, 'detector': 29})
179 self.assertEqual(name, "R10_S02")
181 name = self.mapper._extractDetectorName({'detector': 29})
182 self.assertEqual(name, "R10_S02")
184 name = self.mapper._extractDetectorName({'obsid': "MC_C_20190319_000001",
185 'lsstSerial': "ITL-3800C-041"})
186 self.assertEqual(name, "R10_S02")
188 with self.assertRaises(RuntimeError):
189 self.mapper._extractDetectorName({'visit': 1})
192class MemoryTester(lsst.utils.tests.MemoryTestCase):
193 pass
196def setup_module(module):
197 lsst.utils.tests.init()
200if __name__ == '__main__': 200 ↛ 201line 200 didn't jump to line 201, because the condition on line 200 was never true
201 setup_module(sys.modules[__name__])
202 unittest.main()