"""
Tests for DomainExpertAgent.
"""

import pytest
from dcos.agents.domain import DomainExpertAgent


class TestDomainExpertAgent:
    def test_domain_initialization(self):
        agent = DomainExpertAgent(name="medical", domain="medicine")
        assert agent.domain == "medicine"

    def test_answer_question(self):
        agent = DomainExpertAgent(domain="physics")
        answer = agent.answer("What is gravity?")
        assert "[physics]" in answer

    def test_act(self):
        agent = DomainExpertAgent(domain="chemistry")
        result = agent.act({"question": "What is water?"})
        assert result["domain"] == "chemistry"