tests/test_agent_params.py

import unittest

from src.agent_params import get_agent_params


class TestAgentParams(unittest.TestCase):
    def test_known_agent_temperature_is_loaded(self):
        params = get_agent_params("PatientLLM")
        self.assertEqual(params["temperature"], 0.2)

    def test_unknown_agent_returns_empty_dict(self):
        self.assertEqual(get_agent_params("UnknownAgent"), {})


if __name__ == "__main__":
    unittest.main()