Supply current time for new object radecMjdTai

Author: isullivan

python/lsst/ap/association/testApdb.py

@@ -272,18 +272,19 @@ def run(self, visit, detector,
             nDiaSources = diaObjects.nDiaSources
             self.log.info(f"diaObjects contain {np.min(nDiaSources)} to {np.max(nDiaSources)} diaSources")
 
+        dateTime = DateTime.now().toAstropy()
         # Associate DiaSources with DiaObjects
         associatedDiaSources, newDiaObjects, associatedDiaObjects = self.associateDiaSources(diaSources,
                                                                                              diaObjects,
                                                                                              diaSourcesReal,
                                                                                              diaSourcesBogus,
+                                                                                             dateTime,
                                                                                              )
         # Merge new and preloaded diaObjects
         mergedDiaObjects = self.mergeAssociatedCatalogs(associatedDiaObjects, newDiaObjects)
 
         nObj = len(mergedDiaObjects)
         nSrc = len(associatedDiaSources)
-        dateTime = DateTime.now().toAstropy()
         ind = 0
         t_write0 = time.time()
         # Note that nObj must always be equal to or greater than nSrc
@@ -433,7 +434,7 @@ def simpleMatch(self, diaSourceTable, diaObjects, updateNdiaSources=True):
                                matchedDiaObjects=diaObjects.loc[matchedDiaObjectInds],
                                )
 
-    def associateDiaSources(self, diaSourceTable, diaObjects, diaSourcesReal, diaSourcesBogus):
+    def associateDiaSources(self, diaSourceTable, diaObjects, diaSourcesReal, diaSourcesBogus, dateTime):
         """Associate DiaSources with DiaObjects.
 
         Parameters
@@ -458,7 +459,7 @@ def associateDiaSources(self, diaSourceTable, diaObjects, diaSourcesReal, diaSou
         toAssociate = []
 
         # Create new DiaObjects from unassociated diaSources.
-        createResults = self.createNewDiaObjects(assocResults.unAssocDiaSources)
+        createResults = self.createNewDiaObjects(assocResults.unAssocDiaSources, dateTime)
         if len(assocResults.matchedDiaSources) > 0:
             toAssociate.append(assocResults.matchedDiaSources)
         toAssociate.append(createResults.diaSources)
@@ -483,7 +484,7 @@ def associateDiaSources(self, diaSourceTable, diaObjects, diaSourcesReal, diaSou
                                        inplace=True)
         return (associatedDiaSources, createResults.newDiaObjects, assocResults.matchedDiaObjects)
 
-    def createNewDiaObjects(self, unAssocDiaSources):
+    def createNewDiaObjects(self, unAssocDiaSources, dateTime):
         """Loop through the set of DiaSources and create new DiaObjects
         for unassociated DiaSources.
 
@@ -511,7 +512,8 @@ def createNewDiaObjects(self, unAssocDiaSources):
             # For this simulation we are using custom diaObjectIds,
             #  and need to preserve them.
             # unAssocDiaSources["diaObjectId"] = unAssocDiaSources["diaSourceId"]
-            preserveColumns = ["diaObjectId", ]
+            preserveColumns = ["diaObjectId", "radecMjdTai"]
+            unAssocDiaSources["radecMjdTai"] = dateTime
 
             # Fill the forced sources for each diaObject with random data
             newDiaObjects = fillRandomTable(self.schema, unAssocDiaSources,