[Telemetry] Migrate bitmap.py from bitmaptools.cc to numpy

tools/perf/metrics/speedindex_unittest.py

 # Copyright 2013 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.
 
 # These tests access private methods in the speedindex module.
 # pylint: disable=W0212
 
 import json
 import os
 import unittest
 
-from telemetry.core import bitmap
+from telemetry.image_processing import histogram
+from telemetry.image_processing import rgba_color
 from telemetry.timeline import inspector_timeline_data
 from telemetry.timeline import model
 from metrics import speedindex
 
 # Sample timeline data in the json format provided by devtools.
 # The sample events will be used in several tests below.
 _TEST_DIR = os.path.join(os.path.dirname(__file__), 'unittest_data')
 _SAMPLE_DATA = json.load(open(os.path.join(_TEST_DIR, 'sample_timeline.json')))
 _SAMPLE_TIMELINE_DATA = inspector_timeline_data.InspectorTimelineData(
     _SAMPLE_DATA)
 _SAMPLE_EVENTS = model.TimelineModel(
     timeline_data=_SAMPLE_TIMELINE_DATA).GetAllEvents()
 
 
+class FakeImageUtil(object):
+  # pylint: disable=W0613
+  def GetColorHistogram(self, image, ignore_color=None, tolerance=None):
+    return image.ColorHistogram()
+
 class FakeTimelineModel(object):
-
   def __init__(self):
     self._events = []
 
   def SetAllEvents(self, events):
     self._events = events
 
   def GetAllEvents(self, recursive=True):
     assert recursive == True
     return self._events
 
 
 class FakeVideo(object):
-
   def __init__(self, frames):
     self._frames = frames
 
   def GetVideoFrameIter(self):
     for frame in self._frames:
       yield frame
 
 class FakeBitmap(object):
-
   def __init__(self, r, g, b):
-    self._histogram = bitmap.ColorHistogram(r, g, b, bitmap.WHITE)
+    self._histogram = histogram.ColorHistogram(r, g, b, rgba_color.WHITE)
 
   # pylint: disable=W0613
   def ColorHistogram(self, ignore_color=None, tolerance=None):
     return self._histogram
 
 
 class FakeTab(object):
-
   def __init__(self, video_capture_result=None):
     self._timeline_model = FakeTimelineModel()
     self._javascript_result = None
     self._video_capture_result = FakeVideo(video_capture_result)
 
   @property
   def timeline_model(self):
     return self._timeline_model
 
   @property
@@ skipping 45 matching lines @@
         (0.0, FakeBitmap([ 0, 0, 0,10], [ 0, 0, 0,10], [ 0, 0, 0,10])),
         (0.1, FakeBitmap([10, 0, 0, 0], [10, 0, 0, 0], [10, 0, 0, 0])),
         (0.2, FakeBitmap([ 0, 0, 2, 8], [ 0, 0, 4, 6], [ 0, 0, 1, 9])),
         (0.3, FakeBitmap([ 0, 3, 2, 5], [ 2, 1, 0, 7], [ 0, 3, 0, 7])),
         (0.4, FakeBitmap([ 0, 0, 1, 0], [ 0, 0, 1, 0], [ 0, 0, 1, 0])),
         (0.5, FakeBitmap([ 0, 4, 6, 0], [ 0, 4, 6, 0], [ 0, 4, 6, 0])),
     ]
     max_distance = 42.
 
     tab = FakeTab(frames)
-    impl = speedindex.VideoSpeedIndexImpl()
+    impl = speedindex.VideoSpeedIndexImpl(FakeImageUtil())
     impl.Start(tab)
     impl.Stop(tab)
     time_completeness = impl.GetTimeCompletenessList(tab)
     self.assertEqual(len(time_completeness), 6)
     self.assertEqual(time_completeness[0], (0.0, 0))
     self.assertTimeCompleteness(
         time_completeness[1], 0.1, 1 - (16 + 16 + 16) / max_distance)
     self.assertTimeCompleteness(
         time_completeness[2], 0.2, 1 - (12 + 10 + 13) / max_distance)
     self.assertTimeCompleteness(
         time_completeness[3], 0.3, 1 - (6 + 10 + 8) / max_distance)
     self.assertTimeCompleteness(
         time_completeness[4], 0.4, 1 - (4 + 4 + 4) / max_distance)
     self.assertEqual(time_completeness[5], (0.5, 1))
 
   def testBlankPage(self):
     frames = [
         (0.0, FakeBitmap([0, 0, 0, 1], [0, 0, 0, 1], [0, 0, 0, 1])),
         (0.1, FakeBitmap([0, 0, 0, 1], [0, 0, 0, 1], [0, 0, 0, 1])),
         (0.2, FakeBitmap([1, 0, 0, 0], [0, 0, 0, 1], [0, 0, 0, 1])),
         (0.3, FakeBitmap([0, 0, 0, 1], [0, 0, 0, 1], [0, 0, 0, 1])),
     ]
     tab = FakeTab(frames)
-    impl = speedindex.VideoSpeedIndexImpl()
+    impl = speedindex.VideoSpeedIndexImpl(FakeImageUtil())
     impl.Start(tab)
     impl.Stop(tab)
     time_completeness = impl.GetTimeCompletenessList(tab)
     self.assertEqual(len(time_completeness), 4)
     self.assertEqual(time_completeness[0], (0.0, 1.0))
     self.assertEqual(time_completeness[1], (0.1, 1.0))
     self.assertEqual(time_completeness[2], (0.2, 0.0))
     self.assertEqual(time_completeness[3], (0.3, 1.0))
 
   def assertTimeCompleteness(self, time_completeness, time, completeness):
@@ skipping 64 matching lines @@
 
   def test2ch(self):
     # Page: http://2ch.net/
     # This page has several paint events, including nested paint events.
     self._TestJsonTimelineExpectation(
         '2ch_repeat_timeline.json', (997, 650), 674.58)
 
 
 if __name__ == "__main__":
   unittest.main()