Skip to content
Snippets Groups Projects

Add first test for slicing

Merged Tamas Gal requested to merge slicing into master
1 file
+ 6
0
Compare changes
  • Side-by-side
  • Inline
+ 19
0
@@ -306,6 +306,25 @@ class TestOfflineTracks(unittest.TestCase):
self.assertListEqual([0.230189, 0.230189, 0.218663],
list(dir_z[0][:3]))
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])
)
class TestOfflineTrack(unittest.TestCase):
def setUp(self):
Loading