PIL截取图片进行对比

from PIL import ImageGrab
from PIL import Image
import os
from os.path import *
# 获取对应的路径信息
def picture_path():
    path = dirname(dirname(abspath(__file__)))
    pict_path = os.path.join(path,"picture")
    return pict_path
# 进行封装
class Jt():
    def __init__(self):
        self.tp_path = picture_path()

        # 截图操作
    def scr_shoot(self,tp_name):
        image = ImageGrab.grab() # 使用ImageGrab.grab()进行截图操作
        image.save(self.tp_path+"//{}.png".format(tp_name)) # 保存到对应的路径下


     # 图形对比
    def gc(self,tp_name_one,tp_name_two):
        image = Image.open(self.tp_path+"//{}.png".format(tp_name_one))
        image_two = Image.open(self.tp_path+"//{}.png".format(tp_name_two))  # 打开对应位置的图片信息
        res_one_picture = image.histogram()
        res_picture = image_two.histogram()  # image.histogram()为list数据，当两张图片
        assert res_one_picture == res_picture   # assert进行对比