Test all Exposure components

lsst · Feb 15, 2018 · 6ec10d6 · 6ec10d6
1 parent 571c1e9
commit 6ec10d6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/test_posixDatastoreFits.py b/tests/test_posixDatastoreFits.py
@@ -127,7 +127,13 @@ def testExposurePutGet(self):
         exposureOut = datastore.get(uri, storageClass=storageClass, parameters=None)
         self.assertEqual(type(exposure), type(exposureOut))
 
-        # Get a component
+        # Get some components
+        for c in ("wcs", "image", "mask", "coaddInputs", "psf"):
+            self.assertIn(c, comps)
+            component = datastore.get(comps[c], storageClass=storageClass)
+            self.assertIsNotNone(component)
+
+        # Get a component to check it
         self.assertIn("wcs", comps)
         wcs = datastore.get(comps["wcs"], storageClass=storageClass)
 
@@ -147,7 +153,7 @@ def testExposureCompositePutGet(self):
                                    typeName=None)
 
         # Get a component
-        for c in ("wcs", ):
+        for c in ("wcs", "image", "mask", "coaddInputs", "psf"):
             self.assertIn(c, comps)
             component = datastore.get(comps[c], storageClass=storageClass.components[c])
             self.assertIsNotNone(component)