| Index: tools/telemetry/telemetry/internal/image_processing/video_file_frame_generator_unittest.py
|
| diff --git a/tools/telemetry/telemetry/internal/image_processing/video_file_frame_generator_unittest.py b/tools/telemetry/telemetry/internal/image_processing/video_file_frame_generator_unittest.py
|
| deleted file mode 100644
|
| index 72632a25e544a53090dad6317565e6026464354b..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/telemetry/internal/image_processing/video_file_frame_generator_unittest.py
|
| +++ /dev/null
|
| @@ -1,86 +0,0 @@
|
| -# Copyright 2014 The Chromium Authors. All rights reserved.
|
| -# Use of this source code is governed by a BSD-style license that can be
|
| -# found in the LICENSE file.
|
| -
|
| -import os
|
| -import unittest
|
| -
|
| -from telemetry.core import util
|
| -from telemetry.internal.image_processing import frame_generator
|
| -from telemetry.internal.util import external_modules
|
| -
|
| -try:
|
| - cv2 = external_modules.ImportRequiredModule('cv2')
|
| -except (ImportError, NotImplementedError):
|
| - pass
|
| -else:
|
| - class VideoFileFrameGeneratorTest(unittest.TestCase):
|
| - def __init__(self, *args, **kwargs):
|
| - super(VideoFileFrameGeneratorTest, self).__init__(*args, **kwargs)
|
| - # Import modules with dependencies that may not be preset in test setup so
|
| - # that importing this unit test doesn't cause the test runner to raise an
|
| - # exception.
|
| - from telemetry.internal.image_processing import video_file_frame_generator
|
| - self.VideoFileFrameGenerator = \
|
| - video_file_frame_generator.VideoFileFrameGenerator
|
| -
|
| - def testVideoFileFrameGeneratorSuccess(self):
|
| - vid = os.path.join(util.GetUnittestDataDir(), 'screen_3_frames.mov')
|
| - fg = self.VideoFileFrameGenerator(vid)
|
| - timestamps = [0, 33.367, 66.733]
|
| - self.assertTrue(isinstance(fg, frame_generator.FrameGenerator))
|
| -
|
| - self.assertEqual(fg.CurrentFrameNumber, -1)
|
| - self.assertAlmostEqual(fg.CurrentTimestamp, 0, 3)
|
| - self.assertEqual(fg.Dimensions, (432, 320))
|
| - generator = fg.Generator
|
| - i = 0
|
| - for frame in generator:
|
| - self.assertEqual(fg.CurrentFrameNumber, i)
|
| - self.assertAlmostEqual(fg.CurrentTimestamp, timestamps[i], 3)
|
| - self.assertEqual(fg.Dimensions, (432, 320))
|
| - self.assertEqual(frame.shape[:2], (320, 432))
|
| - i += 1
|
| - self.assertEqual(i, 3)
|
| - try:
|
| - next(generator)
|
| - stopped = False
|
| - except StopIteration:
|
| - stopped = True
|
| - self.assertTrue(stopped)
|
| - try:
|
| - next(fg.Generator)
|
| - stopped = False
|
| - except StopIteration:
|
| - stopped = True
|
| - self.assertTrue(stopped)
|
| -
|
| - def testVideoFileFrameGeneratorSkipFrames(self):
|
| - vid = os.path.join(util.GetUnittestDataDir(), 'screen_3_frames.mov')
|
| - fg = self.VideoFileFrameGenerator(vid, 2)
|
| - self.assertEqual(fg.CurrentFrameNumber, 1)
|
| - self.assertAlmostEqual(fg.CurrentTimestamp, 33.367, 3)
|
| - self.assertEqual(fg.Dimensions, (432, 320))
|
| - next(fg.Generator)
|
| - try:
|
| - next(fg.Generator)
|
| - stopped = False
|
| - except StopIteration:
|
| - stopped = True
|
| - self.assertTrue(stopped)
|
| -
|
| - def testVideoFileFrameGeneratorFailure(self):
|
| - vid = os.path.join(util.GetUnittestDataDir(), 'screen_3_frames.mov')
|
| - try:
|
| - self.VideoFileFrameGenerator(vid, 4)
|
| - fail = False
|
| - except frame_generator.FrameReadError:
|
| - fail = True
|
| - self.assertTrue(fail)
|
| -
|
| - try:
|
| - self.VideoFileFrameGenerator('not_a_file', 0)
|
| - fail = False
|
| - except frame_generator.FrameReadError:
|
| - fail = True
|
| - self.assertTrue(fail)
|
|
|