add loading pythonizations from podio (#290)

* added loading pythonizations from podio * Remove stray statement from test creation Copy-paste error that was not caught in the large PR that introduced the tests --------- Co-authored-by: tmadlener <thomas.madlener@desy.de>
key4hep · Sep 16, 2024 · 109fc7b · 109fc7b
1 parent 189f162
commit 109fc7b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/edm4hep/__init__.py b/python/edm4hep/__init__.py
@@ -24,6 +24,9 @@
     raise RuntimeError('Failed to load Constants.h')
 from ROOT import edm4hep
 
+from podio.pythonizations import load_pythonizations
+load_pythonizations('edm4hep')
+
 # Make TAB completion work for utils
 setattr(edm4hep, 'utils', edm4hep.utils)
 

diff --git a/test/test_EDM4hepFile.py b/test/test_EDM4hepFile.py
@@ -250,7 +250,6 @@ def test_ClusterCollection(event):
     calo_hit = event.get("CalorimeterHitCollection")[0]
     assert len(cluster.getHits()) == 1
     assert cluster.getHits()[0] == calo_hit
-    cluster.addToHits(calo_hit)
 
 
 def test_TrackerHit3DCollection(event):