"""Tests for the run_fusion agent tool."""
from __future__ import annotations

from src.agents.tools import build_default_tools


class TestRunFusionTool:
    def test_fusion_tool_is_registered(self) -> None:
        tools = build_default_tools(rag_index_dir=None)
        names = [t.name for t in tools]
        assert "run_fusion" in names

    def test_fusion_tool_executes_with_only_clinical(self) -> None:
        tools = {t.name: t for t in build_default_tools(rag_index_dir=None)}
        tool = tools["run_fusion"]
        out = tool.execute(tool.input_model.model_validate({
            "clinical": {"mmse": 12.0, "age_years": 78.0},
        }))
        assert out.top_disease in {"alzheimers", "parkinsons", "other"}
        assert any(d.disease == "alzheimers" for d in out.diseases)