98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
from core.models import Bounds, Scene, SceneElement
|
|
from semantic.models import SemanticScene, SemanticWidget
|
|
from world.models import WorldEvent, WorldState
|
|
|
|
|
|
def _scene() -> Scene:
|
|
return Scene(
|
|
width=10,
|
|
height=20,
|
|
elements=[
|
|
SceneElement(
|
|
id="send",
|
|
type="button",
|
|
text="Send",
|
|
bounds=Bounds(1, 2, 3, 4),
|
|
)
|
|
],
|
|
)
|
|
|
|
|
|
def test_world_event_and_state_to_dict() -> None:
|
|
semantic_scene = SemanticScene(
|
|
page="Chat",
|
|
intents=["send a message"],
|
|
widgets=[SemanticWidget(element_id="send", purpose="send message")],
|
|
)
|
|
event = WorldEvent(
|
|
action="tap",
|
|
success=True,
|
|
scene_summary=semantic_scene,
|
|
)
|
|
state = WorldState.with_history_bound(2)
|
|
state.current_app = "com.example.chat"
|
|
state.current_page = "Chat"
|
|
state.variables["contact"] = "Zhang San"
|
|
state.history.append(event)
|
|
|
|
data = state.to_dict()
|
|
|
|
assert data["current_app"] == "com.example.chat"
|
|
assert data["current_page"] == "Chat"
|
|
assert data["variables"] == {"contact": "Zhang San"}
|
|
assert data["history"][0]["scene_summary"] == semantic_scene.to_dict()
|
|
assert data["history"][0]["action"] == "tap"
|
|
assert data["history"][0]["success"] is True
|
|
assert data["history"][0]["timestamp"]
|
|
|
|
|
|
def test_world_state_history_evicts_oldest_entry_at_bound() -> None:
|
|
state = WorldState.with_history_bound(2)
|
|
|
|
state.history.append(WorldEvent(action="first", success=True))
|
|
state.history.append(WorldEvent(action="second", success=True))
|
|
state.history.append(WorldEvent(action="third", success=True))
|
|
|
|
assert len(state.history) == 2
|
|
assert [event.action for event in state.history] == ["second", "third"]
|
|
assert state.history.maxlen == 2
|
|
|
|
|
|
def test_world_event_with_rationale_thinking_page() -> None:
|
|
event = WorldEvent(
|
|
action="tap",
|
|
success=True,
|
|
rationale="Previous step opened settings. Now tapping account.",
|
|
thinking="I should navigate to account settings.",
|
|
arguments={"x": 12, "y": 34},
|
|
purpose="Open account settings.",
|
|
expected_outcome="The account settings page is visible.",
|
|
page="Settings",
|
|
)
|
|
data = event.to_dict()
|
|
|
|
assert data["rationale"] == "Previous step opened settings. Now tapping account."
|
|
assert data["thinking"] == "I should navigate to account settings."
|
|
assert data["arguments"] == {"x": 12, "y": 34}
|
|
assert data["purpose"] == "Open account settings."
|
|
assert data["expected_outcome"] == "The account settings page is visible."
|
|
assert data["page"] == "Settings"
|
|
assert data["scene_summary"] is None
|
|
|
|
|
|
def test_world_event_all_optional_fields_none() -> None:
|
|
event = WorldEvent(action="swipe", success=False)
|
|
data = event.to_dict()
|
|
|
|
assert data["rationale"] is None
|
|
assert data["thinking"] is None
|
|
assert data["arguments"] == {}
|
|
assert data["purpose"] is None
|
|
assert data["expected_outcome"] is None
|
|
assert data["page"] is None
|
|
assert data["scene_summary"] is None
|
|
assert data["action"] == "swipe"
|
|
assert data["success"] is False
|