[CLEANUP][Multi agent debate]

pull/233/head
Kye 1 year ago
parent 4275efaa94
commit 2b6b0506a5

@ -1,66 +0,0 @@
from unittest.mock import patch
from swarms.swarms.multi_agent_debate import (
MultiAgentDebate,
Worker,
select_speaker,
)
def test_multiagentdebate_initialization():
multiagentdebate = MultiAgentDebate(
agents=[Worker] * 5, selection_func=select_speaker
)
assert isinstance(multiagentdebate, MultiAgentDebate)
assert len(multiagentdebate.agents) == 5
assert multiagentdebate.selection_func == select_speaker
@patch("swarms.workers.Worker.reset")
def test_multiagentdebate_reset_agents(mock_reset):
multiagentdebate = MultiAgentDebate(
agents=[Worker] * 5, selection_func=select_speaker
)
multiagentdebate.reset_agents()
assert mock_reset.call_count == 5
def test_multiagentdebate_inject_agent():
multiagentdebate = MultiAgentDebate(
agents=[Worker] * 5, selection_func=select_speaker
)
multiagentdebate.inject_agent(Worker)
assert len(multiagentdebate.agents) == 6
@patch("swarms.workers.Worker.run")
def test_multiagentdebate_run(mock_run):
multiagentdebate = MultiAgentDebate(
agents=[Worker] * 5, selection_func=select_speaker
)
results = multiagentdebate.run("Write a short story.")
assert len(results) == 5
assert mock_run.call_count == 5
def test_multiagentdebate_update_task():
multiagentdebate = MultiAgentDebate(
agents=[Worker] * 5, selection_func=select_speaker
)
multiagentdebate.update_task("Write a short story.")
assert multiagentdebate.task == "Write a short story."
def test_multiagentdebate_format_results():
multiagentdebate = MultiAgentDebate(
agents=[Worker] * 5, selection_func=select_speaker
)
results = [
{"agent": "Agent 1", "response": "Hello, world!"},
{"agent": "Agent 2", "response": "Goodbye, world!"},
]
formatted_results = multiagentdebate.format_results(results)
assert (
formatted_results
== "Agent Agent 1 responded: Hello, world!\nAgent Agent 2"
" responded: Goodbye, world!"
)
Loading…
Cancel
Save