From 873482902ce9f25e5d772a0fb7b04f69080a1008 Mon Sep 17 00:00:00 2001 From: Siddharth Narayanan Date: Thu, 5 Sep 2024 20:03:14 +0000 Subject: [PATCH] bring back test_agent_config --- tests/test_agents.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/test_agents.py b/tests/test_agents.py index f86e017e..e2447226 100644 --- a/tests/test_agents.py +++ b/tests/test_agents.py @@ -631,3 +631,11 @@ async def test_lifecycle(self, dummy_env: DummyEnv) -> None: with patch("httpx.AsyncClient.post", async_client.post), eval_mode(): # Check we can make a second sequential Agent decision without crashing await agent_client.get_asv(agent_state_1, obs) + + +@pytest.mark.parametrize("agent_cls", [SimpleAgent, MemoryAgent, ReActAgent]) +def test_agent_config(agent_cls: type[Agent]): + config = AgentConfig(agent_type=agent_cls.__name__) + assert isinstance(hash(config), int), "AgentConfig should be hashable" + agent = config.construct_agent() + assert isinstance(agent, agent_cls)