From 154f56c31f1362442ff01be323d6a8321cbaf302 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Wed, 1 Apr 2020 15:56:05 +0200
Subject: [PATCH] More tests
---
tests/test_offline.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/tests/test_offline.py b/tests/test_offline.py
index dba5b1f..dfd8e7f 100644
--- a/tests/test_offline.py
+++ b/tests/test_offline.py
@@ -250,6 +250,7 @@ class TestOfflineHits(unittest.TestCase):
class TestOfflineTracks(unittest.TestCase):
def setUp(self):
+ self.f = OFFLINE_FILE
self.tracks = OFFLINE_FILE.events.tracks
self.tracks_numucc = OFFLINE_NUMUCC
self.n_events = 10
@@ -291,6 +292,20 @@ class TestOfflineTracks(unittest.TestCase):
self.assertListEqual(list(tracks.E[:, 0][_slice]),
list(tracks[_slice].E[:, 0]))
+ def test_nested_indexing(self):
+ self.assertAlmostEqual(
+ self.f.events.tracks.fitinf[3:5][1][9][2],
+ self.f.events[3:5].tracks[1].fitinf[9][2])
+ self.assertAlmostEqual(
+ self.f.events.tracks.fitinf[3:5][1][9][2],
+ self.f.events[3:5][1][9][2].tracks.fitinf)
+ self.assertAlmostEqual(
+ self.f.events.tracks.fitinf[3:5][1][9][2],
+ self.f.events[3:5][1].tracks[9][2].fitinf)
+ self.assertAlmostEqual(
+ self.f.events.tracks.fitinf[3:5][1][9][2],
+ self.f.events[3:5][1].tracks[9].fitinf[2])
+
class TestBranchIndexingMagic(unittest.TestCase):
--
GitLab