Index: tools/telemetry/telemetry/util/color_histogram_unittest.py |
diff --git a/tools/telemetry/telemetry/util/color_histogram_unittest.py b/tools/telemetry/telemetry/util/color_histogram_unittest.py |
deleted file mode 100644 |
index a853d534c901781ecd00f23db420bbf30d8189c0..0000000000000000000000000000000000000000 |
--- a/tools/telemetry/telemetry/util/color_histogram_unittest.py |
+++ /dev/null |
@@ -1,117 +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 unittest |
- |
-from telemetry.util import color_histogram |
-from telemetry.util import image_util |
-from telemetry.util import rgba_color |
- |
-class HistogramDistanceTest(unittest.TestCase): |
- def testNoData(self): |
- hist1 = [] |
- hist2 = [] |
- self.assertEqual(color_histogram.HistogramDistance(hist1, hist2), 0) |
- |
- hist1 = [0, 0, 0] |
- hist2 = [0, 0, 0] |
- self.assertRaises( |
- ValueError, lambda: color_histogram.HistogramDistance(hist1, hist2)) |
- |
- def testWrongSizes(self): |
- hist1 = [1] |
- hist2 = [1, 0] |
- self.assertRaises( |
- ValueError, lambda: color_histogram.HistogramDistance(hist1, hist2)) |
- |
- def testNoDistance(self): |
- hist1 = [2, 4, 1, 8, 0, 0] |
- hist2 = [2, 4, 1, 8, 0, 0] |
- self.assertEqual(color_histogram.HistogramDistance(hist1, hist2), 0) |
- |
- def testNormalizeCounts(self): |
- hist1 = [0, 0, 1, 0, 0] |
- hist2 = [0, 0, 0, 0, 7] |
- self.assertEqual(color_histogram.HistogramDistance(hist1, hist2), 2) |
- self.assertEqual(color_histogram.HistogramDistance(hist2, hist1), 2) |
- |
- def testDistance(self): |
- hist1 = [2, 0, 1, 3, 4] |
- hist2 = [3, 1, 2, 4, 0] |
- self.assertEqual(color_histogram.HistogramDistance(hist1, hist2), 1) |
- self.assertEqual(color_histogram.HistogramDistance(hist2, hist1), 1) |
- |
- hist1 = [0, 1, 3, 1] |
- hist2 = [2, 2, 1, 0] |
- self.assertEqual(color_histogram.HistogramDistance(hist1, hist2), 1.2) |
- self.assertEqual(color_histogram.HistogramDistance(hist2, hist1), 1.2) |
- |
- |
-class HistogramTest(unittest.TestCase): |
- def testHistogram(self): |
- pixels = [1, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, |
- 1, 2, 3, 8, 7, 6, 5, 4, 6, 1, 2, 3, |
- 1, 2, 3, 8, 7, 6, 5, 4, 6, 1, 2, 3] |
- bmp = image_util.FromRGBPixels(4, 3, pixels) |
- bmp = image_util.Crop(bmp, 1, 1, 2, 2) |
- |
- hist = image_util.GetColorHistogram(bmp) |
- for i in xrange(3): |
- self.assertEquals(sum(hist[i]), |
- image_util.Width(bmp) * image_util.Height(bmp)) |
- self.assertEquals(hist.r[1], 0) |
- self.assertEquals(hist.r[5], 2) |
- self.assertEquals(hist.r[8], 2) |
- self.assertEquals(hist.g[2], 0) |
- self.assertEquals(hist.g[4], 2) |
- self.assertEquals(hist.g[7], 2) |
- self.assertEquals(hist.b[3], 0) |
- self.assertEquals(hist.b[6], 4) |
- |
- def testHistogramIgnoreColor(self): |
- pixels = [1, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, |
- 1, 2, 3, 8, 7, 6, 5, 4, 6, 1, 2, 3, |
- 1, 2, 3, 8, 7, 6, 5, 4, 6, 1, 2, 3] |
- bmp = image_util.FromRGBPixels(4, 3, pixels) |
- |
- hist = image_util.GetColorHistogram( |
- bmp, ignore_color=rgba_color.RgbaColor(1, 2, 3)) |
- self.assertEquals(hist.r[1], 0) |
- self.assertEquals(hist.r[5], 2) |
- self.assertEquals(hist.r[8], 2) |
- self.assertEquals(hist.g[2], 0) |
- self.assertEquals(hist.g[4], 2) |
- self.assertEquals(hist.g[7], 2) |
- self.assertEquals(hist.b[3], 0) |
- self.assertEquals(hist.b[6], 4) |
- |
- def testHistogramIgnoreColorTolerance(self): |
- pixels = [1, 2, 3, 4, 5, 6, |
- 7, 8, 9, 8, 7, 6] |
- bmp = image_util.FromRGBPixels(2, 2, pixels) |
- |
- hist = image_util.GetColorHistogram( |
- bmp, ignore_color=rgba_color.RgbaColor(0, 1, 2), tolerance=1) |
- self.assertEquals(hist.r[1], 0) |
- self.assertEquals(hist.r[4], 1) |
- self.assertEquals(hist.r[7], 1) |
- self.assertEquals(hist.r[8], 1) |
- self.assertEquals(hist.g[2], 0) |
- self.assertEquals(hist.g[5], 1) |
- self.assertEquals(hist.g[7], 1) |
- self.assertEquals(hist.g[8], 1) |
- self.assertEquals(hist.b[3], 0) |
- self.assertEquals(hist.b[6], 2) |
- self.assertEquals(hist.b[9], 1) |
- |
- def testHistogramDistanceIgnoreColor(self): |
- pixels = [1, 2, 3, 1, 2, 3, |
- 1, 2, 3, 1, 2, 3] |
- bmp = image_util.FromRGBPixels(2, 2, pixels) |
- |
- hist1 = image_util.GetColorHistogram( |
- bmp, ignore_color=rgba_color.RgbaColor(1, 2, 3)) |
- hist2 = image_util.GetColorHistogram(bmp) |
- |
- self.assertEquals(hist1.Distance(hist2), 0) |