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

# 

# LSST Data Management System 

# Copyright 2008-2016 LSST Corporation. 

# 

# 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 <http://www.lsstcorp.org/LegalNotices/>. 

# 

import unittest 

 

import lsst.utils.tests 

import lsst.afw.image as afwImage 

import lsst.geom as geom 

import lsst.ip.diffim as ipDiffim 

import lsst.log.utils as logUtils 

 

verbosity = 0 

logUtils.traceSetAt("ip.diffim", verbosity) 

 

 

class DiffimTestCases(unittest.TestCase): 

 

def testNoMask(self): 

mask = afwImage.Mask(geom.Extent2I(20, 20)) 

mask.set(0) 

fsb = ipDiffim.FindSetBitsU() 

 

bbox = geom.Box2I(geom.Point2I(0, 10), 

geom.Point2I(9, 12)) 

fsb.apply(afwImage.Mask(mask, bbox, afwImage.LOCAL)) 

 

self.assertEqual(fsb.getBits(), 0) 

 

def testOneMask(self): 

mask = afwImage.Mask(geom.Extent2I(20, 20)) 

mask.set(0) 

bitmaskBad = mask.getPlaneBitMask('BAD') 

fsb = ipDiffim.FindSetBitsU() 

 

bbox = geom.Box2I(geom.Point2I(9, 10), 

geom.Point2I(11, 12)) 

submask = afwImage.Mask(mask, bbox, afwImage.LOCAL) 

submask |= bitmaskBad 

 

bbox2 = geom.Box2I(geom.Point2I(8, 8), 

geom.Point2I(19, 19)) 

fsb.apply(afwImage.Mask(mask, bbox2, afwImage.LOCAL)) 

 

self.assertEqual(fsb.getBits(), bitmaskBad) 

 

def testManyMask(self): 

mask = afwImage.Mask(geom.Extent2I(20, 20)) 

mask.set(0) 

bitmaskBad = mask.getPlaneBitMask('BAD') 

bitmaskSat = mask.getPlaneBitMask('SAT') 

fsb = ipDiffim.FindSetBitsU() 

 

bbox = geom.Box2I(geom.Point2I(9, 10), 

geom.Point2I(11, 12)) 

submask = afwImage.Mask(mask, bbox, afwImage.LOCAL) 

submask |= bitmaskBad 

 

bbox2 = geom.Box2I(geom.Point2I(8, 8), 

geom.Point2I(19, 19)) 

submask2 = afwImage.Mask(mask, bbox2, afwImage.LOCAL) 

submask2 |= bitmaskSat 

 

bbox3 = geom.Box2I(geom.Point2I(0, 0), 

geom.Point2I(19, 19)) 

fsb.apply(afwImage.Mask(mask, bbox3, afwImage.LOCAL)) 

 

self.assertEqual(fsb.getBits(), bitmaskBad | bitmaskSat) 

 

##### 

 

 

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

pass 

 

 

def setup_module(module): 

lsst.utils.tests.init() 

 

 

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

lsst.utils.tests.init() 

unittest.main()