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

remove tracks slicing

parent 3ed967b4
No related branches found
No related tags found
No related merge requests found
Pipeline #8909 passed
......@@ -970,12 +970,8 @@ class OfflineTracks:
setattr(self, k, v)
def __getitem__(self, item):
if isinstance(item, int):
return OfflineTrack(self._keys, [v[item] for v in self._values],
fitparameters=self._fitparameters)
else:
return OfflineTracks(self._keys, [v[item] for v in self._values],
fitparameters=self._fitparameters)
return OfflineTrack(self._keys, [v[item] for v in self._values],
fitparameters=self._fitparameters)
def __len__(self):
try:
......
......@@ -452,19 +452,19 @@ class TestOfflineTracks(unittest.TestCase):
def test_slicing(self):
tracks = self.tracks
assert 10 == len(tracks)
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 TestOfflineTrack(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