Index: tools/telemetry/telemetry/util/image_util_unittest.py |
diff --git a/tools/telemetry/telemetry/util/image_util_unittest.py b/tools/telemetry/telemetry/util/image_util_unittest.py |
deleted file mode 100644 |
index 4cb10e6e51cec77976f5897dafff88c680f62f33..0000000000000000000000000000000000000000 |
--- a/tools/telemetry/telemetry/util/image_util_unittest.py |
+++ /dev/null |
@@ -1,128 +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 tempfile |
-import unittest |
- |
-from telemetry.core import util |
-from telemetry.util import image_util |
-from telemetry.util import rgba_color |
- |
-# 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 testWritePngWithoutPngSuffixThrows(self): |
- orig = image_util.FromPngFile(test_png_path) |
- temp_file = tempfile.NamedTemporaryFile().name |
- self.assertRaises(AssertionError, image_util.WritePngFile, |
- orig, temp_file) |
- |
- 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, rgba_color.RgbaColor(1, 0, 0)) |
- self.assertEquals(box, (1, 1, 2, 1)) |
- self.assertEquals(count, 2) |
- |
- box, count = image_util.GetBoundingBox(bmp, rgba_color.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])) |