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…
Reference in new issue