def test_image_match():
class MockDevice():
def __init__(self):
self.x = None
self.y = None
def click(self, x, y):
self.x = x
self.y = y
def screenshot(self, *args, **kwargs):
return cv2.imread(TESTDIR + "/screenshot.jpg")
d = MockDevice()
ix = u2image.ImageX(d)
template = Image.open(TESTDIR + "/template.jpg")
res = ix.match(template)
x, y = res['point']
assert (x, y) == (409, 659), "Match position is wrong"
ix.click(template)
assert d.x == 409
assert d.y == 659
if False: # show position
pim = Image.open(TESTDIR + "/screenshot.jpg")
nim = u2image.draw_point(pim, x, y)
nim.show()
def image(self):
from uiautomator2 import image as _image
return _image.ImageX(self)