Loading [MathJax]/extensions/tex2jax.js
LSST Applications g04a91732dc+9666464c73,g0fba68d861+d6c6f70ffa,g1fd858c14a+94f68680cf,g208c678f98+1ca806343c,g271391ec13+ac98094cfc,g2c84ff76c0+120a924478,g2c9e612ef2+a92a2e6025,g35bb328faa+fcb1d3bbc8,g4d2262a081+7c332456db,g4e0f332c67+c58e4b632d,g53246c7159+fcb1d3bbc8,g60b5630c4e+a92a2e6025,g67b6fd64d1+9d1b2ab50a,g78460c75b0+2f9a1b4bcd,g786e29fd12+cf7ec2a62a,g7b71ed6315+fcb1d3bbc8,g8852436030+506db7da85,g89139ef638+9d1b2ab50a,g8d6b6b353c+a92a2e6025,g9125e01d80+fcb1d3bbc8,g989de1cb63+9d1b2ab50a,g9f33ca652e+d1749da127,ga2b97cdc51+a92a2e6025,gabe3b4be73+1e0a283bba,gb1101e3267+6ecbd0580e,gb58c049af0+f03b321e39,gb89ab40317+9d1b2ab50a,gb90eeb9370+384e1fc23b,gcf25f946ba+506db7da85,gd315a588df+382ef11c06,gd6cbbdb0b4+75aa4b1db4,gd9a9a58781+fcb1d3bbc8,gde0f65d7ad+9984fb7d9e,ge278dab8ac+c61fbefdff,ge410e46f29+9d1b2ab50a,ge82c20c137+e12a08b75a,gf67bdafdda+9d1b2ab50a,gfd5510ef7b+864ead200a,v29.0.0.rc2
LSST Data Management Base Package
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
Public Member Functions | Static Public Member Functions | List of all members
lsst.shapelet.tests.ShapeletTestCase Class Reference
Inheritance diagram for lsst.shapelet.tests.ShapeletTestCase:

Public Member Functions

 compareShapeletFunctions (self, a, b, rtolEllipse=1E-13, rtolCoeff=1E-13, atolEllipse=1E-14, atolCoeff=1E-14)
 
 simplifyMultiShapeletFunction (self, msf)
 
 compareMultiShapeletFunctions (self, a, b, simplify=True, rtolEllipse=1E-13, rtolCoeff=1E-13, atolEllipse=1E-14, atolCoeff=1E-14)
 
 checkMoments (self, function, x, y, z)
 
 checkConvolution (self, f1, f2)
 

Static Public Member Functions

 makeUnitVector (i, n)
 
 makeImage (function, x, y)
 
 makeRandomShapeletFunction (order=2, zeroCenter=False, ellipse=None, scale=1.0)
 
 makeRandomMultiShapeletFunction (nComponents=3, ellipse=None)
 

Detailed Description

Definition at line 38 of file tests.py.

Member Function Documentation

◆ checkConvolution()

lsst.shapelet.tests.ShapeletTestCase.checkConvolution ( self,
f1,
f2 )

Definition at line 140 of file tests.py.

140 def checkConvolution(self, f1, f2):
141 bbox = lsst.geom.Box2I(lsst.geom.Point2I(-50, -50), lsst.geom.Point2I(50, 50))
142 i1 = lsst.afw.image.ImageD(bbox)
143 f1.evaluate().addToImage(i1)
144 self.assertFloatsAlmostEqual(i1.getArray().sum(), f1.evaluate().integrate(), rtol=1E-3)
145 i2 = lsst.afw.image.ImageD(bbox)
146 f2.evaluate().addToImage(i2)
147 self.assertFloatsAlmostEqual(i2.getArray().sum(), f2.evaluate().integrate(), rtol=1E-3)
148 fc1 = f1.convolve(f2)
149 fc2 = f2.convolve(f1)
150 ic1 = lsst.afw.image.ImageD(bbox)
151 fc1.evaluate().addToImage(ic1)
152 ic2 = lsst.afw.image.ImageD(bbox)
153 fc2.evaluate().addToImage(ic2)
154 self.assertFloatsAlmostEqual(ic1.getArray(), ic2.getArray())
155 out = lsst.afw.image.ImageD(bbox)
156 if scipy is None:
157 print("Skipping convolution test; scipy could not be imported.")
158 return
159 # I'm using scipy.ndimage to convolve test images, because I can't figure
160 # out how to make afw do it (afw can convolve images with kernels, but two similarly-sized
161 # are apparently another matter; if I try to make a FixedKernel from one of the images,
162 # I can't even make the operation commutative, let alone correct.
163 scipy.ndimage.convolve(i1.getArray(), i2.getArray(), output=out.getArray(),
164 mode="constant", cval=0.0)
165 self.assertFloatsAlmostEqual(out.getArray(), ic1.getArray(), rtol=1E-4, atol=1E-5)
166 self.assertFloatsAlmostEqual(out.getArray(), ic2.getArray(), rtol=1E-4, atol=1E-5)
167 return fc1, fc2
lsst::geom::Box2I
An integer coordinate rectangle.
Definition Box.h:55

◆ checkMoments()

lsst.shapelet.tests.ShapeletTestCase.checkMoments ( self,
function,
x,
y,
z )

Definition at line 114 of file tests.py.

114 def checkMoments(self, function, x, y, z):
115 gx, gy = numpy.meshgrid(x, y)
116 m = z.sum()
117 dipole = lsst.geom.Point2D((gx * z).sum() / m, (gy * z).sum() / m)
118 gx -= dipole.getX()
119 gy -= dipole.getY()
120 quadrupole = lsst.afw.geom.ellipses.Quadrupole(
121 (gx**2 * z).sum() / m,
122 (gy**2 * z).sum() / m,
123 (gx * gy * z).sum() / m
124 )
125 imageMoments = lsst.afw.geom.ellipses.Ellipse(quadrupole, dipole)
126 shapeletMoments = function.evaluate().computeMoments()
127 self.assertFloatsAlmostEqual(imageMoments.getCenter().getX(),
128 shapeletMoments.getCenter().getX(), rtol=1E-3)
129 self.assertFloatsAlmostEqual(imageMoments.getCenter().getY(),
130 shapeletMoments.getCenter().getY(), rtol=1E-3)
131 self.assertFloatsAlmostEqual(imageMoments.getCore().getIxx(),
132 shapeletMoments.getCore().getIxx(), rtol=1E-3)
133 self.assertFloatsAlmostEqual(imageMoments.getCore().getIyy(),
134 shapeletMoments.getCore().getIyy(), rtol=1E-3)
135 self.assertFloatsAlmostEqual(imageMoments.getCore().getIxy(),
136 shapeletMoments.getCore().getIxy(), rtol=1E-3)
137 integral = numpy.trapz(numpy.trapz(z, gx, axis=1), y, axis=0)
138 self.assertFloatsAlmostEqual(integral, function.evaluate().integrate(), rtol=1E-3)
139
lsst::afw::geom::ellipses::Ellipse
An ellipse defined by an arbitrary BaseCore and a center point.
Definition Ellipse.h:51
lsst::afw::geom::ellipses::Quadrupole
An ellipse core with quadrupole moments as parameters.
Definition Quadrupole.h:47

◆ compareMultiShapeletFunctions()

lsst.shapelet.tests.ShapeletTestCase.compareMultiShapeletFunctions ( self,
a,
b,
simplify = True,
rtolEllipse = 1E-13,
rtolCoeff = 1E-13,
atolEllipse = 1E-14,
atolCoeff = 1E-14 )

Definition at line 104 of file tests.py.

105 atolEllipse=1E-14, atolCoeff=1E-14):
106 if simplify:
107 a = self.simplifyMultiShapeletFunction(a)
108 b = self.simplifyMultiShapeletFunction(b)
109 self.assertEqual(len(a.getComponents()), len(b.getComponents()))
110 for sa, sb in zip(a.getComponents(), b.getComponents()):
111 self.compareShapeletFunctions(sa, sb, rtolEllipse=rtolEllipse, rtolCoeff=rtolCoeff,
112 atolEllipse=atolEllipse, atolCoeff=atolCoeff)
113

◆ compareShapeletFunctions()

lsst.shapelet.tests.ShapeletTestCase.compareShapeletFunctions ( self,
a,
b,
rtolEllipse = 1E-13,
rtolCoeff = 1E-13,
atolEllipse = 1E-14,
atolCoeff = 1E-14 )

Definition at line 82 of file tests.py.

83 atolEllipse=1E-14, atolCoeff=1E-14):
84 self.assertEqual(a.getOrder(), b.getOrder())
85 self.assertEqual(a.getBasisType(), b.getBasisType())
86 self.assertFloatsAlmostEqual(a.getEllipse().getParameterVector(),
87 b.getEllipse().getParameterVector(),
88 rtol=rtolEllipse, atol=atolEllipse)
89 self.assertFloatsAlmostEqual(a.getCoefficients(), b.getCoefficients(),
90 rtol=rtolCoeff, atol=atolCoeff)
91

◆ makeImage()

lsst.shapelet.tests.ShapeletTestCase.makeImage ( function,
x,
y )
static

Definition at line 47 of file tests.py.

47 def makeImage(function, x, y):
48 z = numpy.zeros((y.size, x.size), dtype=float)
49 e = function.evaluate()
50 for i, py in enumerate(y):
51 for j, px in enumerate(x):
52 z[i, j] = e(float(px), float(py))
53 return z
54

◆ makeRandomMultiShapeletFunction()

lsst.shapelet.tests.ShapeletTestCase.makeRandomMultiShapeletFunction ( nComponents = 3,
ellipse = None )
static

Definition at line 76 of file tests.py.

76 def makeRandomMultiShapeletFunction(nComponents=3, ellipse=None):
77 components = []
78 for n in range(nComponents):
79 components.append(ShapeletTestCase.makeRandomShapeletFunction(ellipse=ellipse))
80 return lsst.shapelet.MultiShapeletFunction(components)
81
lsst::shapelet::MultiShapeletFunction
A multi-scale shapelet function.
Definition MultiShapeletFunction.h:38

◆ makeRandomShapeletFunction()

lsst.shapelet.tests.ShapeletTestCase.makeRandomShapeletFunction ( order = 2,
zeroCenter = False,
ellipse = None,
scale = 1.0 )
static

Definition at line 56 of file tests.py.

56 def makeRandomShapeletFunction(order=2, zeroCenter=False, ellipse=None, scale=1.0):
57 center = lsst.geom.Point2D()
58 if not zeroCenter:
59 center = lsst.geom.Point2D(numpy.random.randn(), numpy.random.randn())
60 if ellipse is None:
61 ellipse = lsst.afw.geom.ellipses.Ellipse(
62 lsst.afw.geom.ellipses.Axes(
63 float(numpy.random.uniform(low=1, high=2)),
64 float(numpy.random.uniform(low=1, high=2)),
65 float(numpy.random.uniform(low=0, high=numpy.pi))
66 ),
67 center
68 )
69 coefficients = numpy.random.randn(lsst.shapelet.computeSize(order))
70 result = lsst.shapelet.ShapeletFunction(order, lsst.shapelet.HERMITE, coefficients)
71 result.setEllipse(ellipse)
72 result.getEllipse().scale(scale)
73 return result
74
lsst::afw::geom::ellipses::Axes
An ellipse core for the semimajor/semiminor axis and position angle parametrization (a,...
Definition Axes.h:47
lsst::shapelet::ShapeletFunction
A 2-d function defined by an expansion onto a Gauss-Laguerre or Gauss-Hermite basis.
Definition ShapeletFunction.h:65
lsst::shapelet::computeSize
int computeSize(int order)
Return the size of the coefficient vector for the given order.
Definition constants.h:97

◆ makeUnitVector()

lsst.shapelet.tests.ShapeletTestCase.makeUnitVector ( i,
n )
static

Definition at line 41 of file tests.py.

41 def makeUnitVector(i, n):
42 v = numpy.zeros(n, dtype=float)
43 v[i] = 1.0
44 return v
45

◆ simplifyMultiShapeletFunction()

lsst.shapelet.tests.ShapeletTestCase.simplifyMultiShapeletFunction ( self,
msf )

Definition at line 92 of file tests.py.

92 def simplifyMultiShapeletFunction(self, msf):
93 keep = []
94 for s in msf.getComponents():
95 if not numpy.allclose(s.getCoefficients(), 0.0):
96 params = tuple(s.getEllipse().getParameterVector()) + tuple(s.getCoefficients())
97 keep.append((params, s))
98 msf = lsst.shapelet.MultiShapeletFunction()
99 keep.sort(key=lambda t: numpy.sum(t[0]))
100 for params, s in keep:
101 msf.addComponent(s)
102 return msf
103
The documentation for this class was generated from the following file:
Generated on Tue Mar 25 2025 07:53:37 for LSST Applications by doxygen 1.13.2