aegis / tests /unit /test_models.py
jayc1978's picture
Upload initial Aegis project
8a87cbb verified
Raw History Blame Contribute Delete
790 Bytes
from aegis.core import AgentContext, AgentResult, Task, TaskStatus
def test_task_defaults_to_pending() -> None:
task = Task(name="Plan", description="Create an execution plan.")
assert task.status is TaskStatus.PENDING
assert task.name == "Plan"
def test_agent_context_contains_task() -> None:
task = Task(name="Research", description="Retrieve relevant evidence.")
context = AgentContext(task=task, inputs={"query": "AI agents"})
assert context.task == task
assert context.inputs["query"] == "AI agents"
def test_agent_result_success() -> None:
result = AgentResult(
agent_name="planner",
success=True,
output={"steps": ["inspect", "plan", "execute"]},
)
assert result.success is True
assert result.errors == []