diff --git a/tests/test_agent.py b/tests/test_agent.py index f91be64..299a0f7 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -104,9 +104,9 @@ async def test_plan_never_executes_and_execution_dry_run_default(ctx, workspace) approved_step_ids=[plan["steps"][0]["id"]], approval_acknowledgement=APPROVAL_ACK, ) - assert result.status == "dry_run" - assert result.dry_run is True - assert result.step_results[0].status == "dry_run" + assert result["status"] == "dry_run" + assert result["dry_run"] is True + assert result["step_results"][0]["status"] == "dry_run" @pytest.mark.asyncio