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

tools/telemetry/telemetry/image_processing/image_util_unittest.py

Index: tools/telemetry/telemetry/image_processing/image_util_unittest.py
diff --git a/tools/telemetry/telemetry/image_processing/image_util_unittest.py b/tools/telemetry/telemetry/image_processing/image_util_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..0102e214b278aec268c7e1eca64a7f494c012226
--- /dev/null
+++ b/tools/telemetry/telemetry/image_processing/image_util_unittest.py
@@ -0,0 +1,123 @@
+# 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 tempfile
+import unittest
+
+from telemetry import benchmark
+from telemetry.core import util
+from telemetry.image_processing import image_util
+from telemetry.image_processing.rgba_color import RgbaColor
+
+# This is a simple base64 encoded 2x2 PNG which contains, in order, a single
+# Red, Yellow, Blue, and Green pixel.
+test_png = """
+ iVBORw0KGgoAAAANSUhEUgAAAAIAAAACCAIAAAD91
+ JpzAAAAFklEQVR4Xg3EAQ0AAABAMP1LY3YI7l8l6A
+ T8tgwbJAAAAABJRU5ErkJggg==
+"""
+test_png_path = os.path.join(util.GetUnittestDataDir(), 'test_png.png')
+test_png_2_path = os.path.join(util.GetUnittestDataDir(), 'test_png_2.png')
+
+class ImageUtilTest(unittest.TestCase):
+  def testReadFromBase64Png(self):
+    bmp = image_util.FromBase64Png(test_png)
+
+    self.assertEquals(2, image_util.Width(bmp))
+    self.assertEquals(2, image_util.Height(bmp))
+
+    image_util.GetPixelColor(bmp, 0, 0).AssertIsRGB(255, 0, 0)
+    image_util.GetPixelColor(bmp, 1, 1).AssertIsRGB(0, 255, 0)
+    image_util.GetPixelColor(bmp, 0, 1).AssertIsRGB(0, 0, 255)
+    image_util.GetPixelColor(bmp, 1, 0).AssertIsRGB(255, 255, 0)
+
+  def testReadFromPngFile(self):
+    file_bmp = image_util.FromPngFile(test_png_path)
+
+    self.assertEquals(2, image_util.Width(file_bmp))
+    self.assertEquals(2, image_util.Height(file_bmp))
+
+    image_util.GetPixelColor(file_bmp, 0, 0).AssertIsRGB(255, 0, 0)
+    image_util.GetPixelColor(file_bmp, 1, 1).AssertIsRGB(0, 255, 0)
+    image_util.GetPixelColor(file_bmp, 0, 1).AssertIsRGB(0, 0, 255)
+    image_util.GetPixelColor(file_bmp, 1, 0).AssertIsRGB(255, 255, 0)
+
+  def testWritePngToPngFile(self):
+    orig = image_util.FromPngFile(test_png_path)
+    temp_file = tempfile.NamedTemporaryFile(suffix='.png').name
+    image_util.WritePngFile(orig, temp_file)
+    new_file = image_util.FromPngFile(temp_file)
+    self.assertTrue(image_util.AreEqual(orig, new_file, likely_equal=True))
+
+  def testWriteCroppedBmpToPngFile(self):
+    pixels = [255,0,0, 255,255,0, 0,0,0,
+              255,255,0, 0,255,0, 0,0,0]
+    orig = image_util.FromRGBPixels(3, 2, pixels)
+    orig = image_util.Crop(orig, 0, 0, 2, 2)
+    temp_file = tempfile.NamedTemporaryFile(suffix='.png').name
+    image_util.WritePngFile(orig, temp_file)
+    new_file = image_util.FromPngFile(temp_file)
+    self.assertTrue(image_util.AreEqual(orig, new_file, likely_equal=True))
+
+  def testIsEqual(self):
+    bmp = image_util.FromBase64Png(test_png)
+    file_bmp = image_util.FromPngFile(test_png_path)
+    self.assertTrue(image_util.AreEqual(bmp, file_bmp, likely_equal=True))
+
+  def testDiff(self):
+    file_bmp = image_util.FromPngFile(test_png_path)
+    file_bmp_2 = image_util.FromPngFile(test_png_2_path)
+
+    diff_bmp = image_util.Diff(file_bmp, file_bmp)
+
+    self.assertEquals(2, image_util.Width(diff_bmp))
+    self.assertEquals(2, image_util.Height(diff_bmp))
+
+    image_util.GetPixelColor(diff_bmp, 0, 0).AssertIsRGB(0, 0, 0)
+    image_util.GetPixelColor(diff_bmp, 1, 1).AssertIsRGB(0, 0, 0)
+    image_util.GetPixelColor(diff_bmp, 0, 1).AssertIsRGB(0, 0, 0)
+    image_util.GetPixelColor(diff_bmp, 1, 0).AssertIsRGB(0, 0, 0)
+
+    diff_bmp = image_util.Diff(file_bmp, file_bmp_2)
+
+    self.assertEquals(3, image_util.Width(diff_bmp))
+    self.assertEquals(3, image_util.Height(diff_bmp))
+
+    image_util.GetPixelColor(diff_bmp, 0, 0).AssertIsRGB(0, 255, 255)
+    image_util.GetPixelColor(diff_bmp, 1, 1).AssertIsRGB(255, 0, 255)
+    image_util.GetPixelColor(diff_bmp, 0, 1).AssertIsRGB(255, 255, 0)
+    image_util.GetPixelColor(diff_bmp, 1, 0).AssertIsRGB(0, 0, 255)
+
+    image_util.GetPixelColor(diff_bmp, 0, 2).AssertIsRGB(255, 255, 255)
+    image_util.GetPixelColor(diff_bmp, 1, 2).AssertIsRGB(255, 255, 255)
+    image_util.GetPixelColor(diff_bmp, 2, 0).AssertIsRGB(255, 255, 255)
+    image_util.GetPixelColor(diff_bmp, 2, 1).AssertIsRGB(255, 255, 255)
+    image_util.GetPixelColor(diff_bmp, 2, 2).AssertIsRGB(255, 255, 255)
+
+  def testGetBoundingBox(self):
+    pixels = [0,0,0, 0,0,0, 0,0,0, 0,0,0,
+              0,0,0, 1,0,0, 1,0,0, 0,0,0,
+              0,0,0, 0,0,0, 0,0,0, 0,0,0]
+    bmp = image_util.FromRGBPixels(4, 3, pixels)
+    box, count = image_util.GetBoundingBox(bmp, RgbaColor(1, 0, 0))
+    self.assertEquals(box, (1, 1, 2, 1))
+    self.assertEquals(count, 2)
+
+    box, count = image_util.GetBoundingBox(bmp, RgbaColor(0, 1, 0))
+    self.assertEquals(box, None)
+    self.assertEquals(count, 0)
+
+  def testCrop(self):
+    pixels = [0,0,0, 1,0,0, 2,0,0, 3,0,0,
+              0,1,0, 1,1,0, 2,1,0, 3,1,0,
+              0,2,0, 1,2,0, 2,2,0, 3,2,0]
+    bmp = image_util.FromRGBPixels(4, 3, pixels)
+    bmp = image_util.Crop(bmp, 1, 2, 2, 1)
+
+    self.assertEquals(2, image_util.Width(bmp))
+    self.assertEquals(1, image_util.Height(bmp))
+    image_util.GetPixelColor(bmp, 0, 0).AssertIsRGB(1, 2, 0)
+    image_util.GetPixelColor(bmp, 1, 0).AssertIsRGB(2, 2, 0)
+    self.assertEquals(image_util.Pixels(bmp), bytearray([1,2,0, 2,2,0]))