44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
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)
|
||
|
|
im2 = Image.open(fn2)
|
||
|
|
diff = 0
|
||
|
|
pixels = im1.size[0] * im1.size[1]
|
||
|
|
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
|
||
|
|
if len(errors) != 0:
|
||
|
|
print "-"*80
|
||
|
|
print "Summary:"
|
||
|
|
for error in errors:
|
||
|
|
print "%s failed: %d different pixels" % error
|
||
|
|
print "-"*80
|
||
|
|
sys.exit(1)
|