From c75ff25940bdfb6d6d788445b6d15a947dacbf90 Mon Sep 17 00:00:00 2001 From: xiongyu Date: Thu, 27 Sep 2018 13:21:59 +0800 Subject: [PATCH] add more tests for video_reader slice --- tests/test_video.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/test_video.py b/tests/test_video.py index 1103e1e34..b6f9ef2e2 100644 --- a/tests/test_video.py +++ b/tests/test_video.py @@ -74,6 +74,27 @@ class TestVideo(object): imgs = v[-105:-103] assert int(round(imgs[0].mean())) == 94 assert int(round(imgs[1].mean())) == 205 + assert len(imgs) == 2 + imgs = v[63:65] + assert int(round(imgs[0].mean())) == 94 + assert int(round(imgs[1].mean())) == 205 + assert len(imgs) == 2 + imgs = v[64:62:-1] + assert int(round(imgs[0].mean())) == 205 + assert int(round(imgs[1].mean())) == 94 + assert len(imgs) == 2 + imgs = v[:5] + assert len(imgs) == 5 + for img in imgs: + assert int(round(img.mean())) == 94 + imgs = v[165:] + assert len(imgs) == 3 + for img in imgs: + assert int(round(img.mean())) == 0 + imgs = v[-3:] + assert len(imgs) == 3 + for img in imgs: + assert int(round(img.mean())) == 0 def test_current_frame(self): v = mmcv.VideoReader(self.video_path)