Skip to content
Snippets Groups Projects
Commit 154f56c3 authored by Tamas Gal's avatar Tamas Gal :speech_balloon:
Browse files

More tests

parent 204258e6
No related branches found
No related tags found
1 merge request!27Refactor offline I/O
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment