LSST Applications g013ef56533+d2224463a4,g199a45376c+0ba108daf9,g19c4beb06c+9f335b2115,g1fd858c14a+2459ca3e43,g210f2d0738+2d3d333a78,g262e1987ae+abbb004f04,g2825c19fe3+eedc38578d,g29ae962dfc+0cb55f06ef,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+19c3a54948,g47891489e3+501a489530,g4cdb532a89+a047e97985,g511e8cfd20+ce1f47b6d6,g53246c7159+8c5ae1fdc5,g54cd7ddccb+890c8e1e5d,g5fd55ab2c7+951cc3f256,g64539dfbff+2d3d333a78,g67b6fd64d1+501a489530,g67fd3c3899+2d3d333a78,g74acd417e5+0ea5dee12c,g786e29fd12+668abc6043,g87389fa792+8856018cbb,g89139ef638+501a489530,g8d7436a09f+5ea4c44d25,g8ea07a8fe4+81eaaadc04,g90f42f885a+34c0557caf,g9486f8a5af+165c016931,g97be763408+d5e351dcc8,gbf99507273+8c5ae1fdc5,gc2a301910b+2d3d333a78,gca7fc764a6+501a489530,gce8aa8abaa+8c5ae1fdc5,gd7ef33dd92+501a489530,gdab6d2f7ff+0ea5dee12c,ge410e46f29+501a489530,geaed405ab2+e3b4b2a692,gf9a733ac38+8c5ae1fdc5,w.2025.41
LSST Data Management Base Package
Loading...
Searching...
No Matches
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 Sun Oct 12 2025 07:02:56 for LSST Applications by doxygen 1.13.2