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) |