90 lines
1.8 KiB
Python
90 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Simple test to verify merge works
|
|
"""
|
|
from airtest.core.api import *
|
|
from PIL import Image
|
|
import numpy as np
|
|
|
|
def init_device():
|
|
dev = connect_device("android://127.0.0.1:5037/emulator-5554")
|
|
print(f"Device connected")
|
|
return dev
|
|
|
|
def capture_screen(filename="screen.png"):
|
|
screen = device().snapshot()
|
|
img = Image.fromarray(screen)
|
|
img.save(filename)
|
|
print(f"Screenshot: {filename}")
|
|
return screen
|
|
|
|
# Main
|
|
print("=" * 50)
|
|
print("Simple Merge Test")
|
|
print("=" * 50)
|
|
|
|
init_device()
|
|
sleep(2)
|
|
|
|
# Go to merge screen
|
|
touch([810, 2300]) # Puzzle tab
|
|
sleep(2)
|
|
capture_screen("step1_merge.png")
|
|
|
|
# Select 2x2
|
|
touch([180, 200])
|
|
sleep(1)
|
|
|
|
# Add images
|
|
touch([270, 500])
|
|
sleep(2)
|
|
|
|
# Select 4 images from gallery
|
|
touch([540, 800]) # Gallery tab in picker
|
|
sleep(1)
|
|
touch([200, 400]) # Image 1
|
|
sleep(0.5)
|
|
touch([540, 400]) # Image 2
|
|
sleep(0.5)
|
|
touch([880, 400]) # Image 3
|
|
sleep(0.5)
|
|
touch([200, 700]) # Image 4
|
|
sleep(0.5)
|
|
touch([900, 2300]) # Confirm
|
|
sleep(2)
|
|
|
|
capture_screen("step2_selected.png")
|
|
|
|
# Click preview
|
|
touch([300, 2200]) # Preview button
|
|
sleep(3)
|
|
|
|
capture_screen("step3_preview.png")
|
|
|
|
# Check preview result
|
|
img = Image.open("step3_preview.png")
|
|
img_array = np.array(img)
|
|
h, w = img_array.shape[:2]
|
|
center_region = img_array[h//4:3*h//4, w//4:3*w//4]
|
|
avg_color = np.mean(center_region, axis=(0, 1))
|
|
print(f"Preview average color: {avg_color}")
|
|
|
|
# Close preview
|
|
touch([540, 1500])
|
|
sleep(1)
|
|
|
|
# Save
|
|
touch([800, 2200]) # Save button
|
|
sleep(2)
|
|
touch([700, 1400]) # Confirm save
|
|
sleep(3)
|
|
|
|
capture_screen("step4_saved.png")
|
|
|
|
print("\n" + "=" * 50)
|
|
if avg_color[0] < 250 and avg_color[1] < 250 and avg_color[2] < 250:
|
|
print("RESULT: Merge works! Preview shows merged images.")
|
|
else:
|
|
print("RESULT: Merge may have issues.")
|
|
print("=" * 50)
|