From dcfa630baa6e0eac095af6dea0feb042b9cd684f Mon Sep 17 00:00:00 2001
From: zineb aly <aly.zineb.az@gmail.com>
Date: Thu, 27 Feb 2020 14:55:36 +0100
Subject: [PATCH] remove tracks slicing
---
km3io/offline.py | 8 ++------
tests/test_offline.py | 26 +++++++++++++-------------
2 files changed, 15 insertions(+), 19 deletions(-)
diff --git a/km3io/offline.py b/km3io/offline.py
index 8a8175d..7da446d 100644
--- a/km3io/offline.py
+++ b/km3io/offline.py
@@ -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:
diff --git a/tests/test_offline.py b/tests/test_offline.py
index 1925349..74ea435 100644
--- a/tests/test_offline.py
+++ b/tests/test_offline.py
@@ -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):
--
GitLab