117 lines
2.7 KiB
Python
117 lines
2.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
Android 模拟器图像识别测试脚本
|
|||
|
|
"""
|
|||
|
|
from airtest.core.api import *
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# 初始化设备连接
|
|||
|
|
def init_device():
|
|||
|
|
# 连接 Android 模拟器
|
|||
|
|
dev = connect_device("android://127.0.0.1:5037/emulator-5554")
|
|||
|
|
print(f"已连接设备: {dev}")
|
|||
|
|
return dev
|
|||
|
|
|
|||
|
|
# 截取当前屏幕
|
|||
|
|
def capture_screen(filename="screen.png"):
|
|||
|
|
dev = device()
|
|||
|
|
screen = dev.snapshot()
|
|||
|
|
|
|||
|
|
# numpy.ndarray 转换为 PIL Image 并保存
|
|||
|
|
from PIL import Image
|
|||
|
|
import numpy as np
|
|||
|
|
|
|||
|
|
img = Image.fromarray(screen)
|
|||
|
|
img.save(filename)
|
|||
|
|
print(f"截图已保存: {filename}")
|
|||
|
|
return screen
|
|||
|
|
|
|||
|
|
# 测试1: 打开相机应用
|
|||
|
|
def test_open_camera():
|
|||
|
|
print("测试1: 打开相机应用...")
|
|||
|
|
|
|||
|
|
# 使用 ADB 启动相机
|
|||
|
|
shell("am start -n com.inspection.camera/.ui.MainActivity")
|
|||
|
|
sleep(2)
|
|||
|
|
|
|||
|
|
# 截图
|
|||
|
|
capture_screen("test_open_camera.png")
|
|||
|
|
print("相机已打开")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 测试2: 模拟点击屏幕中心（测试触摸功能）
|
|||
|
|
def test_touch():
|
|||
|
|
print("测试2: 测试触摸功能...")
|
|||
|
|
|
|||
|
|
# 点击屏幕中心
|
|||
|
|
touch([540, 960]) # 模拟器分辨率 1080x1920 的中心点
|
|||
|
|
sleep(1)
|
|||
|
|
|
|||
|
|
capture_screen("test_touch.png")
|
|||
|
|
print("触摸测试完成")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 测试3: 滑动测试
|
|||
|
|
def test_swipe():
|
|||
|
|
print("测试3: 测试滑动功能...")
|
|||
|
|
|
|||
|
|
# 从上往下滑动
|
|||
|
|
swipe([540, 300], [540, 900])
|
|||
|
|
sleep(1)
|
|||
|
|
|
|||
|
|
capture_screen("test_swipe.png")
|
|||
|
|
print("滑动测试完成")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
# 测试4: 查找屏幕上是否有特定文字（使用OCR）
|
|||
|
|
def test_ocr():
|
|||
|
|
print("测试4: OCR文字识别测试...")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from airtest.aircv import aircv
|
|||
|
|
import numpy as np
|
|||
|
|
|
|||
|
|
# 截图
|
|||
|
|
screen = device().snapshot()
|
|||
|
|
|
|||
|
|
# 使用 PIL 显示图像信息
|
|||
|
|
from PIL import Image
|
|||
|
|
img = Image.fromarray(screen)
|
|||
|
|
print(f"屏幕分辨率: {img.size}")
|
|||
|
|
print(f"屏幕模式: {img.mode}")
|
|||
|
|
|
|||
|
|
capture_screen("test_ocr.png")
|
|||
|
|
print("OCR 测试完成")
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"OCR测试出错: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 主测试函数
|
|||
|
|
def main():
|
|||
|
|
print("=" * 50)
|
|||
|
|
print("Android 模拟器图像识别测试")
|
|||
|
|
print("=" * 50)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 初始化设备
|
|||
|
|
init_device()
|
|||
|
|
|
|||
|
|
# 执行测试
|
|||
|
|
test_open_camera()
|
|||
|
|
test_touch()
|
|||
|
|
test_swipe()
|
|||
|
|
test_ocr()
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 50)
|
|||
|
|
print("所有测试完成!")
|
|||
|
|
print("=" * 50)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"测试出错: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|