From 09b877617a846e350338f159727efe828b1f5ff6 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Fri, 11 Dec 2020 13:42:06 +0100
Subject: [PATCH] Fix gSeaGen reader
---
km3io/gseagen.py | 8 ++++----
tests/test_gseagen.py | 6 ------
2 files changed, 4 insertions(+), 10 deletions(-)
diff --git a/km3io/gseagen.py b/km3io/gseagen.py
index 8ee92f8..f7c570b 100644
--- a/km3io/gseagen.py
+++ b/km3io/gseagen.py
@@ -11,15 +11,15 @@ from .tools import cached_property
class GSGReader(EventReader):
"""reader for gSeaGen ROOT files"""
+ header_key = "Header"
event_path = "Events"
- skip_keys = ["Header"]
+ skip_keys = [header_key]
@cached_property
def header(self):
- header_key = "Header"
- if header_key in self._fobj:
+ if self.header_key in self._fobj:
header = {}
- for k, v in self._fobj[header_key].items():
+ for k, v in self._fobj[self.header_key].items():
v = v.array()[0]
if isinstance(v, bytes):
try:
diff --git a/tests/test_gseagen.py b/tests/test_gseagen.py
index 8cb0073..85a41a2 100644
--- a/tests/test_gseagen.py
+++ b/tests/test_gseagen.py
@@ -61,12 +61,6 @@ class TestGSGHeader(unittest.TestCase):
assert self.header["NNu"] == 2
self.assertListEqual(self.header["NuList"].tolist(), [-14, 14])
- @unittest.skip
- def test_unsupported_header(self):
- f = GSGReader(data_path("online/km3net_online.root"))
- with self.assertWarns(UserWarning):
- f.header
-
class TestGSGEvents(unittest.TestCase):
def setUp(self):
--
GitLab