94 """Check that a mapping's reverse transform is the opposite of forward
96 amap is the mapping to test
97 poslist is a list of input position for a forward transform;
98 a numpy array with shape [nin, num points]
99 or collection that can be cast to same
100 rtol is the relative tolerance for numpy.testing.assert_allclose
101 atol is the absolute tolerance for numpy.testing.assert_allclose
103 poslist = np.array(poslist, dtype=float)
104 if len(poslist.shape) == 1:
106 poslist.shape = (1, len(poslist))
108 to_poslist = amap.applyForward(poslist)
109 rt_poslist = amap.applyInverse(to_poslist)
110 assert_allclose(poslist, rt_poslist, rtol=rtol, atol=atol)
113 amapinv = amap.inverted()
114 rt2_poslist = amapinv.applyForward(to_poslist)
115 assert_allclose(poslist, rt2_poslist, rtol=rtol, atol=atol)
118 acmp = amap.then(amapinv)
119 assert_allclose(poslist, acmp.applyForward(poslist), rtol=rtol, atol=atol)
122 posvec = list(poslist.flat)
123 to_posvec = amap.applyForward(posvec)
126 assert_allclose(to_posvec, list(to_poslist.flat), rtol=rtol, atol=atol)
128 rt_posvec = amap.applyInverse(to_posvec)
129 assert_allclose(posvec, rt_posvec, rtol=rtol, atol=atol)
132 """Check basic simplfication for a reversible mapping
135 - A compound mapping of a amap and its inverse simplifies to
137 - A compound mapping of a amap and a unit amap simplifies to
140 amapinv = amap.inverted()
141 cmp1 = amap.then(amapinv)
142 unit1 = cmp1.simplified()
143 self.assertEqual(unit1.className,
"UnitMap")
144 self.assertEqual(amap.nIn, cmp1.nIn)
145 self.assertEqual(amap.nIn, cmp1.nOut)
146 self.assertEqual(cmp1.nIn, unit1.nIn)
147 self.assertEqual(cmp1.nOut, unit1.nOut)
149 cmp2 = amapinv.then(amap)
150 unit2 = cmp2.simplified()
151 self.assertEqual(unit2.className,
"UnitMap")
152 self.assertEqual(amapinv.nIn, cmp2.nIn)
153 self.assertEqual(amapinv.nIn, cmp2.nOut)
154 self.assertEqual(cmp2.nIn, unit2.nIn)
155 self.assertEqual(cmp2.nOut, unit2.nOut)
157 for ma, mb, desmap3
in (
160 (unit2, amapinv, amapinv),
161 (amapinv, unit1, amapinv),
164 cmp3simp = cmp3.simplified()
165 self.assertEqual(cmp3simp.className, amap.simplified().className)
166 self.assertEqual(ma.nIn, cmp3.nIn)
167 self.assertEqual(mb.nOut, cmp3.nOut)
168 self.assertEqual(cmp3.nIn, cmp3simp.nIn)
169 self.assertEqual(cmp3.nOut, cmp3simp.nOut)
172 """Check that a mapping gives identical answers to unpersisted copy
174 poslist is a list of input position for a forward transform
175 (if it exists), or the inverse transform (if not).
176 A numpy array with shape [nAxes, num points]
177 or collection that can be cast to same
179 Checks each direction, if present. However, for generality,
180 does not check that the two directions are inverses of each other;
181 call checkRoundTrip for that.
183 Does everything checkPersistence does, so no need to call both.
185 for channelType, options
in (
187 (FitsChan,
"Encoding=Native"),
194 amap_copy = chan.read()
195 self.assertEqual(amap.className, amap_copy.className)
196 self.assertEqual(amap.show(), amap_copy.show())
197 self.assertEqual(str(amap), str(amap_copy))
198 self.assertEqual(repr(amap), repr(amap_copy))
201 outPoslist = amap.applyForward(poslist)
202 assert_array_equal(outPoslist, amap_copy.applyForward(poslist))
205 assert_array_equal(amap.applyInverse(outPoslist),
206 amap_copy.applyInverse(outPoslist))
208 elif amap.hasInverse:
209 assert_array_equal(amap.applyInverse(poslist),
210 amap_copy.applyInverse(poslist))
213 raise RuntimeError(
"mapping has neither forward nor inverse transform")
216 """Check the memory usage for a compoundObject
218 obj1: first object in compound object
219 obj2: second object in compound object
220 cmpObj: compound object (SeriesMap, ParallelMap, CmpMap or CmpFrame)
221 isSeries: is compound object in series? None to not test
226 deltaObj = 2
if type(obj1) == type(obj2)
else 1
228 initialNumObj1 = obj1.getNObject()
229 initialNumObj2 = obj2.getNObject()
230 initialNumCmpObj = cmpObj.getNObject()
231 initialRefCountObj1 = obj1.getRefCount()
232 initialRefCountObj2 = obj2.getRefCount()
233 initialRefCountCmpObj = cmpObj.getRefCount()
234 self.assertEqual(obj1.getNObject(), initialNumObj1)
235 self.assertEqual(obj2.getNObject(), initialNumObj2)
236 if isSeries
is not None:
238 self.assertTrue(cmpObj.series)
239 elif isSeries
is False:
240 self.assertFalse(cmpObj.series)
245 self.assertEqual(cmpObj.getRefCount(), initialRefCountCmpObj)
246 self.assertEqual(cmpObj.getNObject(), initialNumCmpObj + 1)
247 self.assertEqual(obj1.getRefCount(), initialRefCountObj1)
248 self.assertEqual(obj2.getRefCount(), initialRefCountObj2)
249 self.assertEqual(obj1.getNObject(), initialNumObj1 + deltaObj)
250 self.assertEqual(obj2.getNObject(), initialNumObj2 + deltaObj)
254 self.assertEqual(cmpObj.getRefCount(), initialRefCountCmpObj)
255 self.assertEqual(cmpObj.getNObject(), initialNumCmpObj)
256 self.assertEqual(obj1.getRefCount(), initialRefCountObj1)
257 self.assertEqual(obj1.getNObject(), initialNumObj1)
258 self.assertEqual(obj2.getRefCount(), initialRefCountObj2)
259 self.assertEqual(obj2.getNObject(), initialNumObj2)