From 64ec26d793371e938d2c61330ab3eec6bdd75887 Mon Sep 17 00:00:00 2001
From: Jiahui Zhuo <jiahui.zhuo@cern.ch>
Date: Wed, 12 Oct 2022 14:07:47 +0200
Subject: [PATCH] Add new input support to MCHierarchy functorcollection

---
 .../tupling/option_davinci_tupling_from_collections.py         | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/DaVinciExamples/python/DaVinciExamples/tupling/option_davinci_tupling_from_collections.py b/DaVinciExamples/python/DaVinciExamples/tupling/option_davinci_tupling_from_collections.py
index d52546969..4f394f08c 100644
--- a/DaVinciExamples/python/DaVinciExamples/tupling/option_davinci_tupling_from_collections.py
+++ b/DaVinciExamples/python/DaVinciExamples/tupling/option_davinci_tupling_from_collections.py
@@ -72,6 +72,7 @@ def main(options: Options):
             mcrtible, extra_info=True),
         functorcollections.MCReconstructed_Collection(
             mcrted_all, extra_info=False),
+        functorcollections.MCHierarchy(),
     ]
 
     evt_collections = [
@@ -80,7 +81,7 @@ def main(options: Options):
     ]
 
     assert len(collections) + len(evt_collections) + len(
-        MC_collections) == len(
+        MC_collections) - 1 == len(  # -1 because MCHierarchy is tested twice.
             functorcollections.__all__
         ), "Oh no! Did you forget to add a new collection to this test?"
 
-- 
GitLab