| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 import unittest | 4 import unittest |
| 5 | 5 |
| 6 from telemetry import png_bitmap | 6 from telemetry import png_bitmap |
| 7 | 7 |
| 8 # This is a simple base64 encoded 2x2 PNG which contains, in order, a single | 8 # This is a simple base64 encoded 2x2 PNG which contains, in order, a single |
| 9 # Red, Yellow, Blue, and Green pixel. | 9 # Red, Yellow, Blue, and Green pixel. |
| 10 test_png = """ | 10 test_png = """ |
| 11 iVBORw0KGgoAAAANSUhEUgAAAAIAAAACCAIAAAD91 | 11 iVBORw0KGgoAAAANSUhEUgAAAAIAAAACCAIAAAD91 |
| 12 JpzAAAAAXNSR0IArs4c6QAAAAlwSFlzAAALEwAACx | 12 JpzAAAAAXNSR0IArs4c6QAAAAlwSFlzAAALEwAACx |
| 13 MBAJqcGAAAABZJREFUCNdj/M/AwPCfgYGB4T/DfwY | 13 MBAJqcGAAAABZJREFUCNdj/M/AwPCfgYGB4T/DfwY |
| 14 AHAAD/iOWZXsAAAAASUVORK5CYII= | 14 AHAAD/iOWZXsAAAAASUVORK5CYII= |
| 15 """ | 15 """ |
| 16 | 16 |
| 17 class PngBitmapTest(unittest.TestCase): | 17 class PngBitmapTest(unittest.TestCase): |
| 18 def testRead(self): | 18 def testRead(self): |
| 19 png = png_bitmap.PngBitmap(test_png) | 19 png = png_bitmap.PngBitmap(test_png) |
| 20 | 20 |
| 21 self.assertEquals(2, png.width) | 21 self.assertEquals(2, png.width) |
| 22 self.assertEquals(2, png.height) | 22 self.assertEquals(2, png.height) |
| 23 | 23 |
| 24 png.GetPixelColor(0, 0).AssertIsRGB(255, 0, 0) | 24 png.GetPixelColor(0, 0).AssertIsRGB(255, 0, 0) |
| 25 png.GetPixelColor(1, 1).AssertIsRGB(0, 255, 0) | 25 png.GetPixelColor(1, 1).AssertIsRGB(0, 255, 0) |
| 26 png.GetPixelColor(0, 1).AssertIsRGB(0, 0, 255) | 26 png.GetPixelColor(0, 1).AssertIsRGB(0, 0, 255) |
| 27 png.GetPixelColor(1, 0).AssertIsRGB(255, 255, 0) | 27 png.GetPixelColor(1, 0).AssertIsRGB(255, 255, 0) |
| OLD | NEW |