multi agent

tests/test_agents_models.py

@@ -0,0 +1,77 @@
+from __future__ import annotations
+
+from agents.models import (
+    Observation,
+    ReflectionAction,
+    ReflectionOutcome,
+    VerificationVerdict,
+)
+
+
+def test_observation_round_trip() -> None:
+    obs = Observation(
+        scene_summary="Screen 1080x1920\n  [button] Send",
+        semantic_page="chat",
+        semantic_intents=["send_message"],
+        world_app="com.example.messenger",
+        world_page="conversation",
+        world_variables={"draft": "hello"},
+        raw_scene={"screen": {"width": 1080, "height": 1920}, "elements": []},
+    )
+    data = obs.to_dict()
+    restored = Observation.from_dict(data)
+    assert restored == obs
+
+
+def test_observation_minimal_round_trip() -> None:
+    obs = Observation(scene_summary="empty")
+    data = obs.to_dict()
+    restored = Observation.from_dict(data)
+    assert restored.scene_summary == "empty"
+    assert restored.semantic_page is None
+    assert restored.semantic_intents == []
+    assert restored.world_app is None
+    assert restored.world_page is None
+    assert restored.world_variables == {}
+    assert restored.raw_scene is None
+
+
+def test_verification_verdict_achieved_round_trip() -> None:
+    verdict = VerificationVerdict(result="achieved", reasoning="Effect observed.")
+    data = verdict.to_dict()
+    restored = VerificationVerdict.from_dict(data)
+    assert restored == verdict
+
+
+def test_verification_verdict_not_achieved_round_trip() -> None:
+    verdict = VerificationVerdict(result="not_achieved", reasoning="Scene unchanged.")
+    data = verdict.to_dict()
+    restored = VerificationVerdict.from_dict(data)
+    assert restored == verdict
+
+
+def test_reflection_action_round_trip() -> None:
+    action = ReflectionAction(
+        action="tap",
+        description="Tap the send button",
+        args={"element_id": "btn_send"},
+    )
+    data = action.to_dict()
+    restored = ReflectionAction.from_dict(data)
+    assert restored == action
+
+
+def test_reflection_outcome_with_action_round_trip() -> None:
+    action = ReflectionAction(action="tap", description="Retry tap", args={})
+    outcome = ReflectionOutcome(replan=False, action=action, reasoning="Try again.")
+    data = outcome.to_dict()
+    restored = ReflectionOutcome.from_dict(data)
+    assert restored == outcome
+
+
+def test_reflection_outcome_replan_round_trip() -> None:
+    outcome = ReflectionOutcome(replan=True, reasoning="No recovery possible.")
+    data = outcome.to_dict()
+    restored = ReflectionOutcome.from_dict(data)
+    assert restored == outcome
+    assert restored.action is None