code review

BlueBrain · Oct 2, 2024 · 55bc1be · 55bc1be
1 parent fb6870c
commit 55bc1be
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/multi_agents/test_supervisor_multi_agent.py b/tests/multi_agents/test_supervisor_multi_agent.py
@@ -8,13 +8,12 @@
 from src.neuroagent.multi_agents import SupervisorMultiAgent
 
 
-def test_create_main_agent():
+def test_create_main_agent_initialization():
     mock_llm = MagicMock()
     bind_function_result = MagicMock()
     bind_function_result.__ror__.return_value = {}
     mock_llm.bind_functions.return_value = bind_function_result
     data = {"llm": mock_llm, "agents": [("agent1",)]}
-    from src.neuroagent.multi_agents import SupervisorMultiAgent
 
     result = SupervisorMultiAgent.create_main_agent(data)
     assert "main_agent" in result
@@ -34,7 +33,6 @@ async def mock_ainvoke(_):
     agent_state = MagicMock()
     agent = MagicMock()
     agent.ainvoke = mock_ainvoke
-    from src.neuroagent.multi_agents import SupervisorMultiAgent
 
     agent_node_test = await SupervisorMultiAgent.agent_node(
         agent_state, agent, "test_agent"