Test for DecayTreeFitterAlg

Test for DecayTreeFitterAlg and subsequent functors. Based on Phys!970 (merged) and Rec!2504 (merged)

DaVinciExamples/python/DaVinciExamples/tupling/example-tupling-DTF-run-mc.py

@@ -42,7 +42,8 @@ DTF = DecayTreeFitterAlg(
 # Let's print them
 #
 pdtDM = PrintDecayTree(name="PrintDiMuon", Input=dimuons)
-pdtDTF = PrintDecayTree(name="PrintDTF", Input=DTF)
+psis_refitted, jpsi_to_refitted_map = PrintDecayTree(
+    name="PrintDTF", Input=DTF)
 
 #FunTuple: Jpsi info
 branches = {}
@@ -56,7 +57,8 @@ print('Variables in ALL branch', variables_all)
 #associate FunctorCollection to branch name
 variables = {}
 variables['ALL'] = variables_all  #adds variables to all branches
-# variables['Jpsi'] = { 'DTF_M' : F.DTF_M(DTF.Particles) }
+PT_refit = F.MAP_INPUT(functor=F.PT, mapping=jpsi_to_refitted_map, default=-1),
+variables['Jpsi'] = {'DTF_PT': PT_refit}
 
 #FunTuple: define list of preambles for loki
 loki_preamble = ['TRACK_MAX_PT = MAXTREE(ISBASIC & HASTRACK, PT, -1)']