在UI自动化测试的结果验证过程中,不免会用到截图对比这个方式来判断是否测试通过。以下是一个简单的实现,使用第三方库:Pillow,精确度可能不会很高,但足够应付简单的验证。

from PIL import Image
#使用第三方库:Pillow
import math
import operator
from functools import reduce

image1=Image.open('C:\\Python35\\work\\1.png')
image3=Image.open('C:\\Python35\\work\\3.png')
#把图像对象转换为直方图数据,存在list h1、h2 中
h1=image1.histogram()
h2=image3.histogram()

result = math.sqrt(reduce(operator.add,  list(map(lambda a,b: (a-b)**2, h1, h2)))/len(h1) )
'''
sqrt:计算平方根,reduce函数:前一次调用的结果和sequence的下一个元素传递给operator.add
operator.add(x,y)对应表达式:x+y
这个函数是方差的数学公式:S^2= ∑(X-Y) ^2 / (n-1)
'''
print(result)
#result的值越大,说明两者的差别越大;如果result=0,则说明两张图一模一样