OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/python |
| 2 |
| 3 """ |
| 4 Copyright 2013 Google Inc. |
| 5 |
| 6 Use of this source code is governed by a BSD-style license that can be |
| 7 found in the LICENSE file. |
| 8 |
| 9 Test imagediffdb.py |
| 10 """ |
| 11 |
| 12 # System-level imports |
| 13 import shutil |
| 14 import tempfile |
| 15 import unittest |
| 16 |
| 17 # Local imports |
| 18 import imagediffdb |
| 19 |
| 20 |
| 21 IMG_URL_BASE = ('http://chromium-skia-gm.commondatastorage.googleapis.com/gm/' |
| 22 'bitmap-64bitMD5/') |
| 23 |
| 24 |
| 25 class ImageDiffDbTest(unittest.TestCase): |
| 26 |
| 27 def setUp(self): |
| 28 self.temp_dir = tempfile.mkdtemp() |
| 29 self.maxDiff = None |
| 30 |
| 31 def tearDown(self): |
| 32 shutil.rmtree(self.temp_dir) |
| 33 |
| 34 def shortDescription(self): |
| 35 """Tell unittest framework to not print docstrings for test cases.""" |
| 36 return None |
| 37 |
| 38 def test_sanitize_locator(self): |
| 39 """Test _sanitize_locator().""" |
| 40 # pylint: disable=W0212 |
| 41 self.assertEqual(imagediffdb._sanitize_locator('simple'), 'simple') |
| 42 self.assertEqual(imagediffdb._sanitize_locator(1234), '1234') |
| 43 self.assertEqual(imagediffdb._sanitize_locator('one/two'), 'one_two') |
| 44 self.assertEqual(imagediffdb._sanitize_locator('one\\two'), 'one_two') |
| 45 self.assertEqual(imagediffdb._sanitize_locator('one_two'), 'one_two') |
| 46 |
| 47 def test_simple(self): |
| 48 """Test ImageDiffDB, downloading real known images from Google Storage. |
| 49 |
| 50 TODO(epoger): Instead of hitting Google Storage, we should read image |
| 51 files from local disk using a file:// IMG_URL_BASE. |
| 52 """ |
| 53 # params for each self-test: |
| 54 # 0. expected image locator |
| 55 # 1. expected image URL |
| 56 # 2. actual image locator |
| 57 # 3. actual image URL |
| 58 # 4. expected percent_pixels_differing (as a string, to 4 decimal places) |
| 59 # 5. expected perceptual difference (as a string, to 4 decimal places) |
| 60 # 6. expected max_diff_per_channel |
| 61 selftests = [ |
| 62 [ |
| 63 'arcofzorro/16206093933823793653', |
| 64 IMG_URL_BASE + 'arcofzorro/16206093933823793653.png', |
| 65 'arcofzorro/13786535001616823825', |
| 66 IMG_URL_BASE + 'arcofzorro/13786535001616823825.png', |
| 67 '0.0662', '0.0662', [255, 255, 247], |
| 68 ], |
| 69 [ |
| 70 'gradients_degenerate_2pt/10552995703607727960', |
| 71 IMG_URL_BASE + 'gradients_degenerate_2pt/10552995703607727960.png', |
| 72 'gradients_degenerate_2pt/11198253335583713230', |
| 73 IMG_URL_BASE + 'gradients_degenerate_2pt/11198253335583713230.png', |
| 74 '100.0000', '100.0000', [255, 0, 255], |
| 75 ], |
| 76 ] |
| 77 |
| 78 # Add all image pairs to the database |
| 79 db = imagediffdb.ImageDiffDB(self.temp_dir) |
| 80 for selftest in selftests: |
| 81 db.add_image_pair( |
| 82 expected_image_locator=selftest[0], expected_image_url=selftest[1], |
| 83 actual_image_locator=selftest[2], actual_image_url=selftest[3]) |
| 84 |
| 85 # Fetch each image pair from the database |
| 86 for selftest in selftests: |
| 87 record = db.get_diff_record(expected_image_locator=selftest[0], |
| 88 actual_image_locator=selftest[2]) |
| 89 self.assertEqual('%.4f' % record.get_percent_pixels_differing(), |
| 90 selftest[4]) |
| 91 self.assertEqual('%.4f' % record.get_perceptual_difference(), selftest[5]) |
| 92 self.assertEqual(record.get_max_diff_per_channel(), selftest[6]) |
| 93 |
| 94 |
| 95 def main(): |
| 96 suite = unittest.TestLoader().loadTestsFromTestCase(ImageDiffDbTest) |
| 97 unittest.TextTestRunner(verbosity=2).run(suite) |
| 98 |
| 99 |
| 100 if __name__ == '__main__': |
| 101 main() |
OLD | NEW |