Coverage for tests/test_imsim.py : 23%

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_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 TestImsim(ObsLsstObsBaseOverrides, ObsLsstButlerTests):
34 instrumentDir = "imsim"
36 def setUp(self):
37 dataIds = {'raw': {'expId': 204595, 'detectorName': 'S20', 'raftName': 'R11'},
38 'bias': {'expId': 204595, 'detectorName': 'S20', 'raftName': 'R11'},
39 'flat': {'expId': 204595, 'detectorName': 'S20', 'raftName': 'R11', 'filter': 'i'},
40 'dark': {'expId': 204595, 'detectorName': 'S20', 'raftName': 'R11'}
41 }
42 self.setUp_tests(self._butler, self._mapper, dataIds)
44 ccdExposureId_bits = 38
45 exposureIds = {'raw': 204595042,
46 'bias': 204595042,
47 'dark': 204595042,
48 'flat': 204595042
49 }
50 filters = {'raw': 'i',
51 'bias': '_unknown_',
52 'dark': '_unknown_',
53 'flat': 'i'
54 }
55 exptimes = {'raw': 30.0,
56 'bias': 0.0,
57 'dark': 1.0,
58 'flat': 1.0
59 }
60 detectorIds = {'raw': 42,
61 'bias': 42,
62 'dark': 42,
63 'flat': 42
64 }
65 detector_names = {'raw': 'R11_S20',
66 'bias': 'R11_S20',
67 'dark': 'R11_S20',
68 'flat': 'R11_S20'
69 }
70 detector_serials = {'raw': 'ITL-3800C-102-Dev',
71 'bias': 'ITL-3800C-102-Dev',
72 'dark': 'ITL-3800C-102-Dev',
73 'flat': 'ITL-3800C-102-Dev'
74 }
75 dimensions = {'raw': Extent2I(4256, 4040),
76 'bias': Extent2I(4072, 4000),
77 'dark': Extent2I(4072, 4000),
78 'flat': Extent2I(4072, 4000),
79 }
80 sky_origin = (55.67759886, -30.44239357)
81 raw_subsets = (({'level': 'sensor'}, 1),
82 ({'level': 'sensor', 'filter': 'i'}, 1),
83 ({'level': 'sensor', 'filter': 'foo'}, 0),
84 ({'level': 'sensor', 'expId': 204595}, 1),
85 ({'level': 'filter', 'expId': 204595}, 1),
86 ({'level': 'filter'}, 1),
87 ({'level': 'expId'}, 1),
88 ({'level': 'expId', 'filter': 'i'}, 1),
89 ({'level': 'expId', 'filter': 'foo'}, 0)
90 )
91 linearizer_type = unittest.SkipTest
92 self.setUp_butler_get(ccdExposureId_bits=ccdExposureId_bits,
93 exposureIds=exposureIds,
94 filters=filters,
95 exptimes=exptimes,
96 detectorIds=detectorIds,
97 detector_names=detector_names,
98 detector_serials=detector_serials,
99 dimensions=dimensions,
100 sky_origin=sky_origin,
101 raw_subsets=raw_subsets,
102 linearizer_type=linearizer_type
103 )
105 path_to_raw = os.path.join(self.data_dir, "raw", "204595", "R11",
106 "00204595-R11-S20-det042.fits")
107 keys = set(('filter', 'patch', 'tract', 'visit', 'channel', 'amp', 'style', 'detector', 'dstype',
108 'snap', 'run', 'calibDate', 'half', 'detectorName', 'raftName', 'label',
109 'numSubfilters', 'fgcmcycle', 'name', 'pixel_id', 'description', 'subfilter', 'expId',
110 'dayObs', 'seqNum',))
111 query_format = ["visit", "filter"]
112 queryMetadata = (({'visit': 204595}, [(204595, 'i')]),
113 ({'filter': 'i'}, [(204595, 'i')]),
114 )
115 map_python_type = lsst.afw.image.DecoratedImageF
116 map_python_std_type = lsst.afw.image.ExposureF
117 map_cpp_type = 'DecoratedImageF'
118 map_storage_name = 'FitsStorage'
119 metadata_output_path = os.path.join('processCcd_metadata', '00204595-i', 'R11',
120 'processCcdMetadata_00204595'
121 '-i-R11-S20-det042.yaml')
122 raw_filename = '00204595-R11-S20-det042.fits'
123 default_level = 'sensor'
124 raw_levels = (('sensor', set(['expId', 'detector', 'run', 'detectorName', 'raftName'])),
125 ('skyTile', set(['expId', 'run'])),
126 ('filter', set(['expId'])),
127 ('visit', set(['expId']))
128 )
129 self.setUp_mapper(output=self.data_dir,
130 path_to_raw=path_to_raw,
131 keys=keys,
132 query_format=query_format,
133 queryMetadata=queryMetadata,
134 metadata_output_path=metadata_output_path,
135 map_python_type=map_python_type,
136 map_python_std_type=map_python_std_type,
137 map_cpp_type=map_cpp_type,
138 map_storage_name=map_storage_name,
139 raw_filename=raw_filename,
140 default_level=default_level,
141 raw_levels=raw_levels,
142 test_config_metadata=True
143 )
145 self.setUp_camera(camera_name='LSST-ImSim',
146 n_detectors=189,
147 first_detector_name='R01_S00',
148 plate_scale=20.0 * arcseconds,
149 )
151 super().setUp()
153 def testCcdExposureId(self):
154 with self.assertRaises(KeyError):
155 self.butler.get('ccdExposureId', dataId={})
157 exposureId = self.butler.get('ccdExposureId', dataId={"visit": 1, "detector": 1})
158 self.assertEqual(exposureId, 1001)
160 exposureId = self.butler.get('ccdExposureId', dataId={"visit": 1, "raftName": "R01",
161 "detectorName": "S01"})
162 self.assertEqual(exposureId, 1001)
164 with self.assertRaises(ValueError):
165 self.butler.get('ccdExposureId', dataId={"visit": 1, "detector": 2000})
167 with self.assertRaises(KeyError):
168 self.butler.get('ccdExposureId', dataId={"visit": 1})
170 with self.assertRaises(ValueError):
171 self.butler.get('ccdExposureId', dataId={"visit": 1, "raftName": "R99",
172 "detectorName": "S01"})
174 def testDetectorName(self):
175 name = self.mapper._extractDetectorName({"raftName": "R00", "detectorName": "S00"})
176 self.assertEqual(name, "R00_S00")
178 name = self.mapper._extractDetectorName({'visit': 204595, 'detectorName': 'S20'})
179 self.assertEqual(name, "R11_S20")
181 name = self.mapper._extractDetectorName({'visit': 204595, 'detector': 42})
182 self.assertEqual(name, "R11_S20")
184 name = self.mapper._extractDetectorName({'detector': 42})
185 self.assertEqual(name, "R11_S20")
187 name = self.mapper._extractDetectorName({'visit': 204595})
188 self.assertEqual(name, "R11_S20")
190 with self.assertRaises(RuntimeError):
191 self.mapper._extractDetectorName({'visit': 1})
194class MemoryTester(lsst.utils.tests.MemoryTestCase):
195 pass
198def setup_module(module):
199 lsst.utils.tests.init()
202if __name__ == '__main__': 202 ↛ 203line 202 didn't jump to line 203, because the condition on line 202 was never true
203 setup_module(sys.modules[__name__])
204 unittest.main()