Fix bug in test code.

Since the string literal "id" is never equal to an integer, these tests were definitely not working as intended before.
lsst · Jan 9, 2025 · fde205d · fde205d
1 parent e836c3b
commit fde205d
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/test_astrometryFail.py b/tests/test_astrometryFail.py
@@ -113,10 +113,11 @@ def testVisitCoordsAreNanForFailedAstrom(self):
         self.assertTrue(np.all(np.isfinite(visitTable["ra"])))
 
         visitSummary = self.butler.get("visitSummary", self.calexpMinimalDataId)
-        self.assertTrue(np.isfinite(visitSummary["id" == self.detector]["astromOffsetMean"]))
-        self.assertTrue(np.isfinite(visitSummary["id" == self.detector]["astromOffsetStd"]))
-        self.assertTrue(np.all(np.isnan(visitSummary["id" == 1]["raCorners"])))
-        self.assertTrue(np.all(np.isnan(visitSummary["id" == 1]["decCorners"])))
+        row = visitSummary.find(self.detector)
+        self.assertTrue(np.isfinite(row["astromOffsetMean"]))
+        self.assertTrue(np.isfinite(row["astromOffsetStd"]))
+        self.assertTrue(np.all(np.isnan(row["raCorners"])))
+        self.assertTrue(np.all(np.isnan(row["decCorners"])))
 
     def testMetadataForFailedAstrom(self):
         """Test that the metadata for a failed astrometic fit is set properly.