diff --git a/km3io/gseagen.py b/km3io/gseagen.py index 8ee92f89671b2ed289edded435ac1aa98bc1102a..f7c570b79713f15110933a478b8c4a415d075f8c 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 8cb0073ebc97bc001775a0a4a62cdb0b30a936e8..85a41a2baa0c73fe2bddc66ce7bcf738bec77469 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):