Debug unittest.

lsst · Sep 24, 2020 · 8f00a89 · 8f00a89
1 parent 8ad582c
commit 8f00a89
Showing 1 changed file with 8 additions and 24 deletions.
diff --git a/tests/test_diaForcedSource.py b/tests/test_diaForcedSource.py
@@ -61,23 +61,6 @@ def create_test_dia_objects(n_points, wcs, startPos=100):
         src.setCoord(wcs.pixelToSky(startPos + src_idx,
                                     startPos + src_idx))
 
-    # Add points outside of the CCD.
-    # Fully outside.
-    src = objects.addNew()
-    src['id'] = 10000000
-    src.setCoord(wcs.pixelToSky(-100000,
-                                -100000))
-    # y outside
-    src = objects.addNew()
-    src['id'] = 10000001
-    src.setCoord(wcs.pixelToSky(100,
-                                -100000))
-    # x outside
-    src = objects.addNew()
-    src['id'] = 10000002
-    src.setCoord(wcs.pixelToSky(-100000,
-                                100))
-
     return objects
 
 
@@ -181,21 +164,21 @@ def setUp(self):
         # xy outside
         src = self.testDiaObjects.addNew()
         src['id'] = 10000000
-        src.setCoord(wcs.pixelToSky(-100000,
-                                    -100000))
+        src.setCoord(self.wcs.pixelToSky(-100000,
+                                         -100000))
         # y outside
         src = self.testDiaObjects.addNew()
         src['id'] = 10000001
-        src.setCoord(wcs.pixelToSky(100,
-                                    -100000))
+        src.setCoord(self.wcs.pixelToSky(100,
+                                         -100000))
         # x outside
         src = self.testDiaObjects.addNew()
         src['id'] = 10000002
-        src.setCoord(wcs.pixelToSky(-100000,
-                                    100))
+        src.setCoord(self.wcs.pixelToSky(-100000,
+                                         100))
         # Ids of objects that were "updated" during "ap_association"
         # processing.
-        self.updatedTestIds = [1, 2, 3, 4, 10000000]
+        self.updatedTestIds = np.array([1, 2, 3, 4, 10000001], dtype=np.uint64)
         # Expecdted number of sources is the number of updated ids plus
         # plus any that are within the CCD footprint but are not in the
         # above list of ids.
@@ -210,6 +193,7 @@ def testRun(self):
         test_objects = self._convert_to_pandas(self.testDiaObjects)
         test_objects.rename(columns={"id": "diaObjectId"},
                             inplace=True)
+        test_objects.set_index("diaObjectId", inplace=True, drop=False)
         dfs = DiaForcedSourceTask()
         dia_forced_sources = dfs.run(
             test_objects, self.updatedTestIds, self.expIdBits, self.exposure, self.diffim)