Allow Astropy or Pandas for objectTable in tests.

lsst · Jan 9, 2025 · 569abf9 · 569abf9
1 parent 4b833f8
commit 569abf9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_coadd_outputs.py b/tests/test_coadd_outputs.py
@@ -143,8 +143,8 @@ def test_coadd_transmission_curves(self):
 
             to_check = ctx.choice(len(cat), size=n_object_test, replace=False)
             for index in to_check:
-                coadd_coord = geom.SpherePoint(cat["coord_ra"].values[index]*geom.degrees,
-                                               cat["coord_dec"].values[index]*geom.degrees)
+                coadd_coord = geom.SpherePoint(np.asarray(cat["coord_ra"][index])*geom.degrees,
+                                               np.asarray(cat["coord_dec"][index])*geom.degrees)
                 summed_throughput = np.zeros(wavelengths.shape, dtype=np.float64)
                 weight_sum = 0.0
                 for rec in coadd_inputs.subsetContaining(coadd_coord, includeValidPolygon=True):
@@ -330,7 +330,7 @@ def test_coadd_psf(self):
             to_check = ctx.choice(len(star_cat), size=n_object_test, replace=False)
             n_good = 0
             for index in to_check:
-                position = geom.Point2D(star_cat["x"].values[index], star_cat["y"].values[index])
+                position = geom.Point2D(np.asarray(star_cat["x"][index]), np.asarray(star_cat["y"][index]))
                 psf_image = coadd_psf.computeImage(position)
                 psf_image_bbox = psf_image.getBBox()
                 star_image = lsst.afw.image.ImageF(