1 files changed, 59 insertions, 0 deletions
diff --git a/gm/rebaseline_server/imagediffdb_test.py b/gm/rebaseline_server/imagediffdb_test.py
new file mode 100755
index 0000000000..034ac51b3e
--- /dev/null
+++ b/gm/rebaseline_server/imagediffdb_test.py
@@ -0,0 +1,59 @@
+#!/usr/bin/python
+
+"""
+Copyright 2013 Google Inc.
+
+Use of this source code is governed by a BSD-style license that can be
+found in the LICENSE file.
+
+Test imagediffdb.py
+
+TODO(epoger): Modify to use Python's unittest framework.
+"""
+
+# System-level imports
+import logging
+
+# Local imports
+import imagediffdb
+
+
+IMAGE_URL_BASE = 'http://chromium-skia-gm.commondatastorage.googleapis.com/gm/bitmap-64bitMD5/'
+
+def main():
+  logging.basicConfig(level=logging.INFO)
+
+  # params for each self-test:
+  # 0. expected image locator
+  # 1. expected image URL
+  # 2. actual image locator
+  # 3. actual image URL
+  # 4. expected percent_pixels_differing (as a string, to 4 decimal places)
+  # 5. expected weighted_diff_measure (as a string, to 4 decimal places)
+  selftests = [
+      ['16206093933823793653',
+       IMAGE_URL_BASE + 'arcofzorro/16206093933823793653.png',
+       '13786535001616823825',
+       IMAGE_URL_BASE + 'arcofzorro/13786535001616823825.png',
+       '0.0653', '0.0113'],
+  ]
+
+  # Add all image pairs to the database
+  db = imagediffdb.ImageDiffDB('/tmp/ImageDiffDB')
+  for selftest in selftests:
+    retval = db.add_image_pair(
+        expected_image_locator=selftest[0], expected_image_url=selftest[1],
+        actual_image_locator=selftest[2],   actual_image_url=selftest[3])
+
+  # Fetch each image pair from the database
+  for selftest in selftests:
+    record = db.get_diff_record(expected_image_locator=selftest[0],
+                                actual_image_locator=selftest[2])
+    assert (('%.4f' % record.get_percent_pixels_differing()) == selftest[4])
+    assert (('%.4f' % record.get_weighted_diff_measure()) == selftest[5])
+
+  logging.info("Self-test completed successfully!")
+
+
+if __name__ == '__main__':
+  main()