22 lines
685 B
Python
22 lines
685 B
Python
|
# MIT License
|
|||
|
|
# Copyright (c) 2024
|
|||
|
|
"""最小端到端测试：Dummy LLM 推理。"""
|
|||
|
|
|
|||
|
|
from autodemo.llm import DummyLLM
|
|||
|
|
from autodemo.schema import ControlSnapshot, EventRecord, Rect
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_dummy_llm_generate() -> None:
|
|||
|
|
llm = DummyLLM()
|
|||
|
|
ev = EventRecord(
|
|||
|
|
kind="mouse_click",
|
|||
|
|
timestamp=1.0,
|
|||
|
|
data={"x": 1, "y": 2},
|
|||
|
|
control=ControlSnapshot(
|
|||
|
|
AutomationId="btn1", Name="按钮", ClassName="Button", ControlType="Button", BoundingRectangle=Rect(left=0, top=0, right=10, bottom=10)
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
spec = llm.generate([ev])
|
|||
|
|
assert spec.steps[0]["action"] == "click"
|
|||
|
|
assert spec.steps[0]["target"]["AutomationId"] == "btn1"
|