Fix to test

lsst · Nov 21, 2023 · 15f73e3 · 15f73e3
1 parent 78e7d78
commit 15f73e3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/test_gbdesAstrometricFit.py b/tests/test_gbdesAstrometricFit.py
@@ -32,6 +32,7 @@
 import lsst.afw.table as afwTable
 import lsst.afw.geom as afwgeom
 from lsst.drp.tasks.gbdesAstrometricFit import (GbdesAstrometricFitConfig, GbdesAstrometricFitTask,
+                                                GbdesGlobalAstrometricFitConfig,
                                                 GbdesGlobalAstrometricFitTask)
 from lsst.daf.base import PropertyList
 from lsst.meas.algorithms import ReferenceObjectLoader
@@ -432,8 +433,9 @@ def test_loading_and_association(self):
         # Running `_load_catalogs_and_associate` changes the input WCSs, so 
         # recalculate them here so that the variables shared among tests are
         # not affected.
+        instrument = wcsfit.Instrument(self.instrumentName)
         _, exposuresHelper, extensionInfo = self.task._get_exposure_info(
-            self.inputVisitSummary, self.instrument, refEpoch=self.refEpoch)
+            self.inputVisitSummary, instrument, refEpoch=self.refEpoch)
 
         tmpAssociations = wcsfit.FoFClass(self.fields, [self.instrument], exposuresHelper,
                                           [self.fieldRadius.asDegrees()],
@@ -445,8 +447,8 @@ def test_loading_and_association(self):
         matchIds = []
         correctMatches = []
         for (s, e, o) in zip(tmpAssociations.sequence, tmpAssociations.extn, tmpAssociations.obj):
-            objVisitInd = self.extensionInfo.visitIndex[e]
-            objDet = self.extensionInfo.detector[e]
+            objVisitInd = extensionInfo.visitIndex[e]
+            objDet = extensionInfo.detector[e]
             ExtnInds = self.inputCatalogRefs[objVisitInd].get()['detector'] == objDet
             objInfo = self.inputCatalogRefs[objVisitInd].get()[ExtnInds].iloc[o]
             if s == 0:
@@ -549,7 +551,7 @@ def setUpClass(cls):
                                                                     f'visitSummary_{testVisit}.fits'))
             cls.inputVisitSummary.append(visSum)
 
-        cls.config = GbdesAstrometricFitConfig()
+        cls.config = GbdesGlobalAstrometricFitConfig()
         cls.config.systematicError = 0
         cls.config.devicePolyOrder = 4
         cls.config.exposurePolyOrder = 6