Skip to content
Snippets Groups Projects

Add first test for slicing

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