Hide keyboard shortcuts

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 unittest 

24 

25import lsst.utils.tests 

26from lsst.obs.base.gen2to3 import convertTests 

27 

28testDataPackage = "obs_lsst" 

29try: 

30 testDataDirectory = lsst.utils.getPackageDir(testDataPackage) 

31except LookupError: 

32 testDataDirectory = None 

33 

34 

35class ConvertGen2To3TestCase(convertTests.ConvertGen2To3TestCase): 

36 detectorKey = "detector" 

37 exposureKey = "expId" 

38 config = os.path.join(os.path.abspath(os.path.dirname(__file__)), 

39 "config", "convert2to3Config.py") 

40 

41 def setUp(self): 

42 rawpath = os.path.join(testDataDirectory, f"data/input/{self.instrumentDir}") 

43 calibpath = os.path.join(testDataDirectory, f"{rawpath}/CALIB") 

44 

45 self.gen2root = rawpath 

46 if os.path.exists(calibpath): 

47 self.gen2calib = calibpath 

48 

49 super().setUp() 

50 

51 if self.gen2calib: 

52 self.collections.add(f"calib/{self.instrumentName}") 

53 

54 

55class LatissGen2To3TestCase(ConvertGen2To3TestCase, lsst.utils.tests.TestCase): 

56 instrumentName = "LATISS" 

57 instrumentDir = "latiss" 

58 instrumentClass = "lsst.obs.lsst.Latiss" 

59 biases = [{"detector": 0, "calibration_label": "gen2/bias_2018-09-21_000", 

60 "instrument": instrumentName}] 

61 

62 

63class TS8Gen2To3TestCase(ConvertGen2To3TestCase, lsst.utils.tests.TestCase): 

64 

65 instrumentName = "LSST-TS8" 

66 instrumentDir = "ts8" 

67 instrumentClass = "lsst.obs.lsst.LsstTS8" 

68 biases = [{"detector": 67, "calibration_label": "gen2/bias_2018-07-24_067", 

69 "instrument": instrumentName}] 

70 

71 

72class TS3Gen2To3TestCase(ConvertGen2To3TestCase, lsst.utils.tests.TestCase): 

73 

74 instrumentName = "LSST-TS3" 

75 instrumentDir = "ts3" 

76 instrumentClass = "lsst.obs.lsst.LsstTS3" 

77 

78 

79class UCDCamGen2To3TestCase(ConvertGen2To3TestCase, lsst.utils.tests.TestCase): 

80 

81 instrumentName = "LSST-UCDCam" 

82 instrumentDir = "ucd" 

83 instrumentClass = "lsst.obs.lsst.LsstUCDCam" 

84 

85 

86class PhoSimGen2To3TestCase(ConvertGen2To3TestCase, lsst.utils.tests.TestCase): 

87 

88 instrumentName = "LSST-PhoSim" 

89 instrumentDir = "phosim" 

90 instrumentClass = "lsst.obs.lsst.LsstPhoSim" 

91 

92 

93class ImSimGen2To3TestCase(ConvertGen2To3TestCase, lsst.utils.tests.TestCase): 

94 

95 instrumentName = "LSST-ImSim" 

96 instrumentDir = "imsim" 

97 instrumentClass = "lsst.obs.lsst.LsstImSim" 

98 biases = [{"detector": 42, "calibration_label": "gen2/bias_2022-01-01_042", 

99 "instrument": instrumentName}] 

100 darks = [{"detector": 42, "calibration_label": "gen2/dark_2022-01-01_042", 

101 "instrument": instrumentName}] 

102 flats = [{"detector": 42, "calibration_label": "gen2/flat_2022-08-06_042_i", 

103 "instrument": instrumentName, "physical_filter": "i"}] 

104 

105 

106def setup_module(module): 

107 lsst.utils.tests.init() 

108 

109 

110if __name__ == "__main__": 110 ↛ 111line 110 didn't jump to line 111, because the condition on line 110 was never true

111 lsst.utils.tests.init() 

112 unittest.main()