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

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

# 

# LSST Data Management System 

# Copyright 2016 AURA/LSST. 

# 

# This product includes software developed by the 

# LSST Project (http://www.lsst.org/). 

# 

# This program is free software: you can redistribute it and/or modify 

# it under the terms of the GNU General Public License as published by 

# the Free Software Foundation, either version 3 of the License, or 

# (at your option) any later version. 

# 

# This program is distributed in the hope that it will be useful, 

# but WITHOUT ANY WARRANTY; without even the implied warranty of 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

# GNU General Public License for more details. 

# 

# You should have received a copy of the LSST License Statement and 

# the GNU General Public License along with this program. If not, 

# see <https://www.lsstcorp.org/LegalNotices/>. 

# 

 

import os 

import shutil 

import tempfile 

import unittest 

import warnings 

 

import lsst.utils.tests 

import lsst.afw.image as afwImage 

import lsst.geom as geom 

import lsst.pex.exceptions as pexExcept 

from lsst.pipe.tasks.processCcd import ProcessCcdTask 

from lsst.utils import getPackageDir 

from lsst.base import disableImplicitThreading 

 

OutputName = None # Specify a name (as a string) to save the output repository 

 

 

class ProcessCcdTestCase(lsst.utils.tests.TestCase): 

"""Tests to run processCcd or tests with processed data""" 

@classmethod 

def setUpClass(cls): 

"""Runs ProcessCcdTask so the test* methods can inspect the results.""" 

try: 

cls.datadir = getPackageDir("testdata_decam") 

except pexExcept.NotFoundError: 

message = "testdata_decam not setup. Skipping." 

warnings.warn(message) 

raise unittest.SkipTest(message) 

 

cls.outPath = tempfile.mkdtemp() if OutputName is None else OutputName 

cls.dataId = {'visit': 229388, 'ccdnum': 1} 

configPath = os.path.join(getPackageDir("obs_decam"), "config") 

argsList = [os.path.join(cls.datadir, "rawData"), "--output", cls.outPath, "--id"] 

argsList += ["%s=%s" % (key, val) for key, val in cls.dataId.items()] 

argsList += ["--calib", os.path.join(cls.datadir, "rawData/cpCalib")] 

argsList += ["--config", "calibrate.doPhotoCal=False", "calibrate.doAstrometry=False", 

# This test uses CP-MasterCal calibration products 

"-C", "%s/processCcdCpIsr.py" % configPath] 

argsList.append('--doraise') 

disableImplicitThreading() # avoid contention with other processes 

fullResult = ProcessCcdTask.parseAndRun(args=argsList, doReturnResults=True) 

cls.butler = fullResult.parsedCmd.butler 

cls.config = fullResult.parsedCmd.config 

 

@classmethod 

def tearDownClass(cls): 

del cls.butler 

if OutputName is None: 

shutil.rmtree(cls.outPath) 

else: 

print("testProcessCcd.py's output data saved to %r" % (OutputName,)) 

 

def testProcessRaw(self): 

"""Sanity check of running processCcd with raw data""" 

exp = self.butler.get("calexp", self.dataId, immediate=True) 

self.assertIsInstance(exp, afwImage.ExposureF) 

self.assertEqual(exp.getWidth(), 2048) 

self.assertEqual(exp.getHeight(), 4096) 

 

def testCcdKey(self): 

"""Test to retrieve calexp using ccd as the ccd key""" 

exp = self.butler.get("calexp", visit=229388, ccd=1, immediate=True) 

self.assertIsInstance(exp, afwImage.ExposureF) 

 

def testWcsPostIsr(self): 

"""Test the wcs of postISRCCD products 

 

The postISRCCD wcs should be the same as the raw wcs 

after adding camera distortion and 

adjustment of overscan/prescan trimming. Test DM-4859. 

""" 

if not self.config.isr.doWrite or not self.config.isr.assembleCcd.doTrim: 

return 

expRaw = self.butler.get("raw", self.dataId, immediate=True) 

expPost = self.butler.get("postISRCCD", self.dataId, immediate=True) 

self.assertIsInstance(expPost, afwImage.ExposureF) 

wcsRaw = expRaw.getWcs() 

wcsPost = expPost.getWcs() 

# Shift WCS for trimming the prescan and overscan region 

# ccdnum 1 is S29, with overscan in the bottom 

wcsRaw = wcsRaw.copyAtShiftedPixelOrigin(geom.Extent2D(-56, -50)) 

self.assertWcsAlmostEqualOverBBox(wcsRaw, wcsPost, expPost.getBBox()) 

 

 

class MemoryTester(lsst.utils.tests.MemoryTestCase): 

pass 

 

 

def setup_module(module): 

lsst.utils.tests.init() 

 

 

115 ↛ 116line 115 didn't jump to line 116, because the condition on line 115 was never trueif __name__ == "__main__": 

lsst.utils.tests.init() 

unittest.main()