X-iZhang
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""Shared fixtures for EvoScientist tests."""
|
|
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_tool_call():
|
|
"""A minimal tool call dict."""
|
|
return {"id": "tc_001", "name": "execute", "args": {"command": "ls -la"}}
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_tool_result():
|
|
"""A minimal tool result dict."""
|
|
return {"name": "execute", "content": "[OK] file1.py file2.py", "success": True}
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_events():
|
|
"""A sequence of stream event dicts covering common types."""
|
|
return [
|
|
{"type": "thinking", "content": "Let me think..."},
|
|
{"type": "text", "content": "Here is the answer."},
|
|
{"type": "tool_call", "id": "tc_001", "name": "execute", "args": {"command": "ls"}},
|
|
{"type": "tool_result", "name": "execute", "content": "[OK] done", "success": True},
|
|
{"type": "subagent_start", "name": "research-agent", "description": "Find papers"},
|
|
{"type": "subagent_tool_call", "subagent": "research-agent", "name": "tavily_search", "args": {"query": "test"}, "id": "tc_sa_001"},
|
|
{"type": "subagent_tool_result", "subagent": "research-agent", "name": "tavily_search", "content": "Results...", "success": True},
|
|
{"type": "subagent_end", "name": "research-agent"},
|
|
{"type": "done", "response": "Here is the answer."},
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def tmp_workspace(tmp_path):
|
|
"""Provide a temporary workspace directory path."""
|
|
ws = tmp_path / "workspace"
|
|
ws.mkdir()
|
|
return str(ws)
|