File size: 790 Bytes
8a87cbb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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 == []