#import math, operator import Image import sys COMPUTE_THRESHOLD = 16 errors = [] # returns true if pixels are not identical def compare_pixels(pixel1, pixel2): r_diff = abs(pixel1[0] - pixel2[0]) g_diff = abs(pixel1[1] - pixel2[1]) b_diff = abs(pixel1[2] - pixel2[2]) if(r_diff > COMPUTE_THRESHOLD or g_diff > COMPUTE_THRESHOLD or b_diff > COMPUTE_THRESHOLD): return True else: return False # compare tow images and return number of different pixels def compare(fn1, fn2): global errors im1 = Image.open(fn1) try: im2 = Image.open(fn2) except IOError: errors.append((fn1, None)) return -1 diff = 0 pixels = im1.size[0] * im1.size[1] delta_pixels = im2.size[0] * im2.size[1] - pixels if delta_pixels != 0: errors.append((fn1, delta_pixels)) return delta_pixels im1 = im1.getdata() im2 = im2.getdata() for i in range(3, pixels - 1, 3): if(compare_pixels(im1[i], im2[i])): diff = diff + 1 if diff != 0: errors.append((fn1, diff)) return diff def summary(): global errors print "-"*80 print "Summary:" if len(errors) != 0: for error in errors: if (error[1] is None): print "Could not verify %s: No reference image found!" % error[0] else: print "%s failed: %d different pixels" % error print "-"*80 sys.exit(1) else: print 'No errors detected!' print "-"*80 sys.exit(0)