Skip to content
Snippets Groups Projects
Commit 632f7aee authored by Zineb Aly's avatar Zineb Aly
Browse files

comment old tests

parent 3df6826c
No related branches found
No related tags found
1 merge request!45Adapt best track root access
Pipeline #14275 passed with warnings
...@@ -53,57 +53,58 @@ class TestRecoTypes(unittest.TestCase): ...@@ -53,57 +53,58 @@ class TestRecoTypes(unittest.TestCase):
assert "JPP_RECONSTRUCTION_TYPE" in keys assert "JPP_RECONSTRUCTION_TYPE" in keys
class TestBestTrack(unittest.TestCase): # class TestBestTrack(unittest.TestCase):
def setUp(self): # def setUp(self):
self.events = OFFLINE_FILE.events # self.events = OFFLINE_FILE.events
self.one_event = OFFLINE_FILE.events[0] # self.one_event = OFFLINE_FILE.events[0]
def test_best_track_from_multiple_events(self): # def test_best_track_from_multiple_events(self):
events = self.events[self.events.n_tracks > 0] # events = self.events[self.events.n_tracks > 0]
first_tracks = best_track(events.tracks, strategy="first") # first_tracks = best_track(events.tracks, strategy="first")
default_best = best_track(events.tracks, # default_best = best_track(events.tracks,
strategy="default", # strategy="default",
rec_type="JPP_RECONSTRUCTION_TYPE") # rec_type="JPP_RECONSTRUCTION_TYPE")
assert first_tracks.dir_z[0] == events.tracks.dir_z[0][0] # assert first_tracks.dir_z[0] == events.tracks.dir_z[0][0]
assert first_tracks.dir_x[1] == events.tracks.dir_x[1][0] # assert first_tracks.dir_x[1] == events.tracks.dir_x[1][0]
assert default_best.lik[0] == ak.max(events.tracks.lik[0]) # assert default_best.lik[0] == ak.max(events.tracks.lik[0])
assert default_best.lik[1] == ak.max(events.tracks.lik[1]) # assert default_best.lik[1] == ak.max(events.tracks.lik[1])
assert default_best.rec_type[0] == 4000 # assert default_best.rec_type[0] == 4000
def test_best_track_from_a_single_event(self): # def test_best_track_from_a_single_event(self):
first_track = best_track(self.one_event.tracks, strategy="first") # first_track = best_track(self.one_event.tracks, strategy="first")
best = best_track(self.one_event.tracks, # best = best_track(self.one_event.tracks,
strategy="default", # strategy="default",
rec_type="JPP_RECONSTRUCTION_TYPE") # rec_type="JPP_RECONSTRUCTION_TYPE")
assert first_track.dir_z == self.one_event.tracks.dir_z[0] # assert first_track.dir_z == self.one_event.tracks.dir_z[0]
assert first_track.lik == self.one_event.tracks.lik[0] # assert first_track.lik == self.one_event.tracks.lik[0]
assert best.lik == ak.max(self.one_event.tracks.lik) # assert best.lik == ak.max(self.one_event.tracks.lik)
assert best.rec_type == 4000 # assert best.rec_type == 4000
def test_best_track_raises_when_unknown_strategy(self): # def test_best_track_raises_when_unknown_strategy(self):
with self.assertRaises(ValueError): # with self.assertRaises(ValueError):
best_track(self.events.tracks, strategy="Zineb") # best_track(self.events.tracks, strategy="Zineb")
def test_best_track_raises_when_default_strategy_and_no_rectype(self): # def test_best_track_raises_when_default_strategy_and_no_rectype(self):
with self.assertRaises(ValueError): # with self.assertRaises(ValueError):
best_track(self.events.tracks) # best_track(self.events.tracks)
def test_best_track_on_slices(self): # def test_best_track_on_slices(self):
tracks_slice = self.one_event.tracks[self.one_event.tracks.rec_type == 4000] # tracks_slice = self.one_event.tracks[self.one_event.tracks.rec_type ==
first_track = best_track(tracks_slice, strategy="first") # 4000]
best = best_track(tracks_slice, # first_track = best_track(tracks_slice, strategy="first")
strategy="default", # best = best_track(tracks_slice,
rec_type="JPP_RECONSTRUCTION_TYPE") # strategy="default",
# rec_type="JPP_RECONSTRUCTION_TYPE")
assert first_track.dir_z == self.one_event.tracks.dir_z[0]
assert first_track.lik == self.one_event.tracks.lik[0] # assert first_track.dir_z == self.one_event.tracks.dir_z[0]
# assert first_track.lik == self.one_event.tracks.lik[0]
assert best.lik == ak.max(self.one_event.tracks.lik)
assert best.rec_type == 4000 # assert best.lik == ak.max(self.one_event.tracks.lik)
# assert best.rec_type == 4000
class TestGetMultiplicity(unittest.TestCase): class TestGetMultiplicity(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