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

# 

# LSST Data Management System 

# Copyright 2008-2017 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 unittest 

 

import lsst.afw.geom as afwGeom 

import lsst.utils.tests 

import lsst.ip.isr.straylight as straylight 

import lsst.ip.isr.vignette as vignette 

import lsst.ip.isr.masking as masking 

import lsst.ip.isr.linearize as linearize 

 

import lsst.ip.isr.isrMock as isrMock 

 

 

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

 

def setUp(self): 

self.inputExp = isrMock.TrimmedRawMock().run() 

self.mi = self.inputExp.getMaskedImage() 

 

def test_straylight(self): 

"""Assert that the straylight task does not error when given an exposure. 

""" 

task = straylight.StrayLightTask() 

with self.assertRaises(NotImplementedError): 

task.run(self.inputExp, None) 

 

def test_vignette_noWrite(self): 

"""Assert that the vignette task does not error when given an exposure 

""" 

config = vignette.VignetteConfig() 

config.radius = 125.0 

config.xCenter = 100.0 

config.yCenter = 100.0 

 

config.doWriteVignettePolygon = False 

task = vignette.VignetteTask(config=config) 

result = task.run(self.inputExp) 

 

# DM-19707: ip_isr functionality not fully tested by unit tests 

self.assertIsNone(result) 

 

def test_vignette_doWrite(self): 

"""Assert that the vignette task does not error when given an exposure 

""" 

config = vignette.VignetteConfig() 

config.radius = 125.0 

config.xCenter = 100.0 

config.yCenter = 100.0 

 

config.doWriteVignettePolygon = True 

task = vignette.VignetteTask(config=config) 

result = task.run(self.inputExp) 

 

self.assertIsInstance(result, afwGeom.Polygon) 

 

def test_masking(self): 

"""Assert that the camera-specific masking task does not error when 

given an exposure. 

""" 

task = masking.MaskingTask() 

result = task.run(self.inputExp) 

 

# DM-19707: ip_isr functionality not fully tested by unit tests 

self.assertIsNone(result) 

 

def test_linearize(self): 

"""Assert that the linearize task does not error when a linearity is requested. 

""" 

for linearityTypeName in ('LookupTable', 'Squared', 'Unknown'): 

result = linearize.getLinearityTypeByName(linearityTypeName) 

# These return the actual class to use, so use Equal instead of IsInstance. 

if linearityTypeName == 'LookupTable': 

self.assertEqual(result, linearize.LinearizeLookupTable, msg=f"{linearityTypeName}") 

elif linearityTypeName == 'Squared': 

self.assertEqual(result, linearize.LinearizeSquared, msg=f"{linearityTypeName}") 

else: 

# DM-19707: ip_isr functionality not fully tested by unit tests 

self.assertIsNone(result) 

 

 

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

pass 

 

 

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

lsst.utils.tests.init() 

unittest.main()