diff --git a/tests/test_offline.py b/tests/test_offline.py index 4f47a4917e3cd77104dfaed1f4f2ebc058702cc1..dba5b1f17de29376c1aff09ccfab595abdbee367 100644 --- a/tests/test_offline.py +++ b/tests/test_offline.py @@ -156,6 +156,10 @@ class TestOfflineEvents(unittest.TestCase): assert np.allclose(self.events[3:5].n_hits, self.events.n_hits[3:5]) assert np.allclose(self.events[3:5][0].n_hits, self.events.n_hits[3:5][0]) + assert np.allclose(self.events[3:5].hits[1].dom_id[4], + self.events.hits[3:5][1][4].dom_id) + assert np.allclose(self.events.hits[3:5][1][4].dom_id, + self.events[3:5][1][4].hits.dom_id) def test_str(self): assert str(self.n_events) in str(self.events) @@ -273,20 +277,20 @@ class TestOfflineTracks(unittest.TestCase): tracks = self.tracks self.assertEqual(10, len(tracks)) self.assertEqual(1, len(tracks[0])) - # track_selection = tracks[2:7] - # assert 5 == len(track_selection) - # track_selection_2 = tracks[1:3] - # assert 2 == len(track_selection_2) - # for _slice in [ - # slice(0, 0), - # slice(0, 1), - # slice(0, 2), - # slice(1, 5), - # slice(3, -2) - # ]: - # self.assertListEqual(list(tracks.E[:, 0][_slice]), - # list(tracks[_slice].E[:, 0])) - # + track_selection = tracks[2:7] + assert 5 == len(track_selection) + track_selection_2 = tracks[1:3] + assert 2 == len(track_selection_2) + for _slice in [ + slice(0, 0), + slice(0, 1), + slice(0, 2), + slice(1, 5), + slice(3, -2) + ]: + self.assertListEqual(list(tracks.E[:, 0][_slice]), + list(tracks[_slice].E[:, 0])) + class TestBranchIndexingMagic(unittest.TestCase): @@ -300,10 +304,6 @@ class TestBranchIndexingMagic(unittest.TestCase): assert np.allclose(self.events[3:6].tracks.pos_y[:, 0], self.events.tracks.pos_y[3:6, 0]) - # # test slicing with a tuple - # assert np.allclose(self.events[0].hits[1].dom_id[0:10], - # self.events.hits[(0, 1)].dom_id[0:10]) - # test selecting with a list self.assertEqual(3, len(self.events[[0, 2, 3]]))