Merge pull request #171 from lsst/tickets/DM-20566-daf_butler

DM-20566: Use lsst.geom not lsst.afw.geom
lsst · Jul 15, 2019 · 43e9eed · 43e9eed
2 parents 730ec69 + ce8deb4
commit 43e9eed
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_butlerFits.py b/tests/test_butlerFits.py
@@ -34,7 +34,7 @@
 try:
     import lsst.afw.image
     from lsst.afw.image import LOCAL
-    from lsst.geom import Box2I, Point2I
+    from lsst.geom import Box2I, Point2I, Extent2I
 except ImportError:
     lsst.afw = None
 
@@ -114,8 +114,7 @@ def runExposureCompositePutGetTest(self, storageClass, datasetTypeName):
             #                       compRef.datasetType.storageClass.pytype)
             compsRead[compName] = component
         # Simple check of WCS
-        bbox = lsst.afw.geom.Box2I(lsst.afw.geom.Point2I(0, 0),
-                                   lsst.afw.geom.Extent2I(9, 9))
+        bbox = Box2I(Point2I(0, 0), Extent2I(9, 9))
         self.assertWcsAlmostEqualOverBBox(compsRead["wcs"], exposure.getWcs(), bbox)
 
         # With parameters