Coverage for tests/test_ringsSkyMap.py: 21%

117 statements  

« prev     ^ index     » next       coverage.py v7.2.1, created at 2023-03-12 01:49 -0800

1import unittest 

2import math 

3import numpy as np 

4 

5import lsst.utils.tests 

6import lsst.geom 

7 

8from lsst.skymap.ringsSkyMap import RingsSkyMap 

9from helper import skyMapTestCase 

10 

11 

12class RingsTestCase(skyMapTestCase.SkyMapTestCase): 

13 

14 def setUp(self): 

15 config = RingsSkyMap.ConfigClass() 

16 config.numRings = 3 

17 self.setAttributes( 

18 SkyMapClass=RingsSkyMap, 

19 name="rings", 

20 config=config, 

21 numTracts=26, 

22 neighborAngularSeparation=None, # no uniform tract separation 

23 numNeighbors=None, # ignored because neighborAngularSeparation=None 

24 ) 

25 

26 def testPoles(self): 

27 """Test that findAllTracts behaves at the poles 

28 

29 Testing fix to DM-10686. 

30 """ 

31 skymap = self.getSkyMap() 

32 for ra in (0, 123, 321, 359.9): 

33 tracts = skymap.findAllTracts(lsst.geom.SpherePoint(ra, 90, lsst.geom.degrees)) 

34 self.assertListEqual(tracts, [skymap[len(skymap) - 1]]) 

35 tracts = skymap.findAllTracts(lsst.geom.SpherePoint(ra, -90, lsst.geom.degrees)) 

36 self.assertListEqual(tracts, [skymap[0]]) 

37 

38 def testSha1Compare(self): 

39 """Test that RingsSkyMap's extra state is included in its hash.""" 

40 defaultSkyMap = self.getSkyMap() 

41 for numRings in (4, 5): 

42 config = self.getConfig() 

43 config.numRings = numRings 

44 skyMap = self.getSkyMap(config=config) 

45 self.assertNotEqual(skyMap, defaultSkyMap) 

46 for raStart in (60.0, 75.0): 

47 config = self.getConfig() 

48 config.raStart = raStart 

49 skyMap = self.getSkyMap(config=config) 

50 self.assertNotEqual(skyMap, defaultSkyMap) 

51 

52 def testCorners(self): 

53 """Test that corners of a tract can be found in the tract""" 

54 skymap = self.getSkyMap() 

55 for tract in skymap: 

56 vertices = tract.getVertexList() 

57 for coord in vertices: 

58 self.assertIn(tract.getId(), [tt.getId() for tt in skymap.findAllTracts(coord)]) 

59 

60 

61class NonzeroRaStartRingsTestCase(RingsTestCase): 

62 """Test that setting raStart != 0 works""" 

63 def getConfig(self): 

64 config = super().getConfig() 

65 config.raStart = 234 

66 return config 

67 

68 

69class HscRingsTestCase(lsst.utils.tests.TestCase): 

70 def getConfig(self): 

71 """Return a configuration matching that used for the HSC SSP""" 

72 config = RingsSkyMap.ConfigClass() 

73 config.numRings = 120 

74 config.projection = "TAN" 

75 config.tractOverlap = 1.0/60 # Overlap between tracts (degrees) 

76 config.pixelScale = 0.168 

77 return config 

78 

79 def setUp(self): 

80 self.skymap = RingsSkyMap(self.getConfig()) 

81 

82 def tearDown(self): 

83 del self.skymap 

84 

85 def testDm7770(self): 

86 """Test that DM-7770 has been fixed 

87 

88 These operations previously caused: 

89 lsst::pex::exceptions::RuntimeError: 'Error: wcslib 

90 returned a status code of 9 at sky 30.18, -3.8 deg: 

91 One or more of the world coordinates were invalid' 

92 

93 We are only testing function, and not the actual results. 

94 """ 

95 coordList = [lsst.geom.SpherePoint(ra, dec, lsst.geom.degrees) for 

96 ra, dec in [(30.18, -3.8), (31.3, -3.8), (31.3, -2.7), (30.18, -2.7)]] 

97 for coord in coordList: 

98 self.skymap.findAllTracts(coord) 

99 self.skymap.findTractPatchList(coordList) 

100 

101 def testDm14809(self): 

102 """Test that DM-14809 has been fixed""" 

103 skyMapTestCase.checkDm14809(self, self.skymap) 

104 

105 # Check that the first tract in the last ring exists 

106 coord = self.getFirstTractLastRingCoord() 

107 tract = self.skymap.findTract(coord) 

108 self.assertTrue(tract.contains(coord)) 

109 

110 tractId = self.skymap.findTractIdArray(coord.getLongitude().asRadians(), 

111 coord.getLatitude().asRadians(), 

112 degrees=False) 

113 self.assertEqual(tractId[0], tract.getId()) 

114 

115 def testWraparound(self): 

116 """Check wrapping at RA=0 

117 

118 How-to-reproduce of a bug identified by Sogo Mineo. 

119 """ 

120 tractId = 9712 

121 deviation = 10 / 3600.0 # 10 arcsec 

122 tract = self.skymap[tractId] 

123 center = tract.getCtrCoord() 

124 centerRa = center.getRa().asDegrees() 

125 centerDec = center.getDec().asDegrees() 

126 for devRa in [-deviation, deviation]: 

127 coord = lsst.geom.SpherePoint(centerRa + devRa, centerDec, lsst.geom.degrees) 

128 foundTractId = self.skymap.findTract(coord).getId() 

129 self.assertEqual(tractId, foundTractId) 

130 

131 foundTractArrayId = self.skymap.findTractIdArray([centerRa + devRa], 

132 [centerDec], 

133 degrees=True) 

134 self.assertEqual(tractId, foundTractArrayId[0]) 

135 

136 def testFindTractIdArray(self): 

137 """Test findTractIdArray. 

138 

139 Test an array of positions to ensure that findTract and findTractIdArray 

140 give the same answers. 

141 """ 

142 np.random.seed(12345) 

143 

144 ras = np.random.uniform(low=0.0, high=360.0, size=1000) 

145 decs = np.random.uniform(low=-90.0, high=90.0, size=1000) 

146 

147 coords = [lsst.geom.SpherePoint(ra*lsst.geom.degrees, dec*lsst.geom.degrees) 

148 for ra, dec in zip(ras, decs)] 

149 

150 indexes = [self.skymap.findTract(coord).getId() for coord in coords] 

151 indexes2 = self.skymap.findTractIdArray(ras, decs, degrees=True) 

152 

153 np.testing.assert_array_equal(indexes2, indexes) 

154 

155 def getFirstTractLastRingCoord(self): 

156 """Return the coordinates of the first tract in the last ring 

157 

158 This tract is missing in version=0, but this is fixed in version=1.in 

159 """ 

160 ringNum = self.skymap.config.numRings - 1 

161 ringSize = math.pi/(self.skymap.config.numRings + 1) 

162 firstRingStart = ringSize*0.5 - 0.5*math.pi 

163 dec = ringNum*ringSize + firstRingStart 

164 return lsst.geom.SpherePoint(self.skymap.config.raStart*lsst.geom.degrees, 

165 dec*lsst.geom.radians) 

166 

167 

168class Version0HscRingsTestCase(HscRingsTestCase): 

169 """Testing that the version=0 RingsSkyMap works in the expected way""" 

170 def setUp(self): 

171 self.skymap = RingsSkyMap(self.getConfig(), version=0) 

172 

173 def testDm14809(self): 

174 """Test that DM-14809 has been partially fixed 

175 

176 The observed behaviour was: 

177 

178 skyMap.findTract(skyMap[9712].getCtrCoord()).getId() != 9712 

179 

180 and 

181 

182 skyMap[1].getCtrCoord() == skyMap[11].getCtrCoord() 

183 

184 Specifically for version=0, we fixed the ``findTract`` behaviour but 

185 left the tract duplication (tract 11 duplicates tract 1) and the 

186 missing tract (the first tract in the last ring) so that the tract 

187 numbering would remain unchanged. 

188 """ 

189 # Check that the tract found for central coordinate of a tract is that tract 

190 expect = [tract.getId() for tract in self.skymap] 

191 expect[self.skymap._ringNums[0] + 1] = 1 # Due to the bug 

192 got = [self.skymap.findTract(tract.getCtrCoord()).getId() for tract in self.skymap] 

193 self.assertEqual(got, expect) 

194 

195 # Check that the tract central coordinates are unique 

196 # Round to integer arcminutes so differences are relatively immune to small numerical inaccuracies 

197 centers = set([(int(coord.getRa().asArcminutes()), int(coord.getDec().asArcminutes())) for 

198 coord in (tract.getCtrCoord() for tract in self.skymap)]) 

199 self.assertEqual(len(centers), len(self.skymap) - 1) # One tract is duplicated 

200 self.assertEqual(self.skymap[1].getCtrCoord(), 

201 self.skymap[self.skymap._ringNums[0] + 1].getCtrCoord()) # This is the duplicate 

202 

203 # Check that some particular tracts we know and love haven't moved 

204 degrees = lsst.geom.degrees 

205 # 9712 is at RA=0, and was identified as problematic in DM-14809 

206 self.assertEqual(self.skymap[9712].getCtrCoord(), 

207 lsst.geom.SpherePoint(0.0*degrees, 0.7438016528925696*degrees)) 

208 # The Cosmos field 

209 self.assertEqual(self.skymap[9813].getCtrCoord(), 

210 lsst.geom.SpherePoint(150.2479338842975*degrees, 2.2314049586776834*degrees)) 

211 

212 # Check that the first tract in the last ring does NOT exist (due to the bug) 

213 coord = self.getFirstTractLastRingCoord() 

214 tract = self.skymap.findTract(coord) 

215 self.assertFalse(tract.contains(coord)) 

216 

217 

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

219 pass 

220 

221 

222def setup_module(module): 

223 lsst.utils.tests.init() 

224 

225 

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

227 lsst.utils.tests.init() 

228 unittest.main()