From 6a60bbf7b59b630f45f1753e28e6dcc7fa1f0a49 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Thu, 24 Mar 2022 12:40:37 +0000
Subject: [PATCH] Unhide counter from mc_trks
---
setup.cfg | 2 +-
src/km3io/offline.py | 1 +
tests/test_offline.py | 8 ++++++++
3 files changed, 10 insertions(+), 1 deletion(-)
diff --git a/setup.cfg b/setup.cfg
index e31081a..0ac7926 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -54,7 +54,7 @@ where = src
all =
dev =
black==21.6b0
- km3net-testdata>=0.2.26
+ km3net-testdata>=0.3.3
ipykernel
matplotlib
memory_profiler
diff --git a/src/km3io/offline.py b/src/km3io/offline.py
index a811ae4..0812799 100644
--- a/src/km3io/offline.py
+++ b/src/km3io/offline.py
@@ -83,6 +83,7 @@ class OfflineReader(EventReader):
"len": "mc_trks.len",
"status": "mc_trks.status",
"mother_id": "mc_trks.mother_id",
+ "counter": "mc_trks.counter",
"pdgid": "mc_trks.type",
"hit_ids": "mc_trks.hit_ids",
"usr": "mc_trks.usr", # TODO: trouble with uproot4
diff --git a/tests/test_offline.py b/tests/test_offline.py
index b4310b4..2cdf33c 100644
--- a/tests/test_offline.py
+++ b/tests/test_offline.py
@@ -491,6 +491,14 @@ class TestOfflineTracks(unittest.TestCase):
)
+class TestMisc(unittest.TestCase):
+ def test_mc_tracks_counter(self):
+ r = OfflineReader(data_path("gseagen/DAT000001.gSeaGen.1.aa.root"))
+ np.testing.assert_allclose(
+ [0, 0, 6, 7, 0, 0, 2, 53, 0, 0, 6, 57, 0], r.mc_tracks.counter[0][:13]
+ )
+
+
class TestBranchIndexingMagic(unittest.TestCase):
def setUp(self):
self.events = OFFLINE_FILE.events
--
GitLab