Spaces:
Running
Running
File size: 603 Bytes
fd1472e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
"""Tests for Search Agent domain support."""
from unittest.mock import MagicMock
from src.agents.search_agent import SearchAgent
from src.config.domain import SEXUAL_HEALTH_CONFIG, ResearchDomain
class TestSearchAgentDomain:
def test_search_agent_accepts_domain(self):
mock_handler = MagicMock()
store = {"current": []}
agent = SearchAgent(
search_handler=mock_handler, evidence_store=store, domain=ResearchDomain.SEXUAL_HEALTH
)
# Verify description updated
assert agent.description == SEXUAL_HEALTH_CONFIG.search_agent_description
|