From bfa6743756db4db48d61fb136bd9112f518c5a46 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Tue, 10 Mar 2020 14:28:07 +0100
Subject: [PATCH] Add more tests
---
tests/test_offline.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/tests/test_offline.py b/tests/test_offline.py
index 86f6390..693da0b 100644
--- a/tests/test_offline.py
+++ b/tests/test_offline.py
@@ -210,6 +210,16 @@ class TestOfflineTracks(unittest.TestCase):
# ]:
# self.assertListEqual(list(tracks.E[:, 0][_slice]),
# list(tracks[_slice].E[:, 0]))
+ #
+
+class TestBranchIndexingMagic(unittest.TestCase):
+ def setUp(self):
+ self.events = OFFLINE_FILE.events
+
+ def test_foo(self):
+ assert 318 == self.events[2:4].n_hits[0]
+ assert np.allclose(self.events[3].tracks.dir_z[10], self.events.tracks.dir_z[3, 10])
+ assert np.allclose(self.events[3:6].tracks.pos_y[:, 0], self.events.tracks.pos_y[3:6, 0])
class TestUsr(unittest.TestCase):
--
GitLab