Spaces:
Runtime error
Runtime error
import pytest | |
from unittest.mock import Mock | |
from swarms.workers.worker_agent_ultra import WorkerUltraNode # import your module here | |
def test_create_agent(): | |
mock_llm = Mock() | |
mock_toolset = { 'test_toolset': Mock() } | |
mock_vectorstore = Mock() | |
worker = WorkerUltraNode(mock_llm, mock_toolset, mock_vectorstore) | |
worker.create_agent() | |
assert worker.agent is not None | |
def test_add_toolset_invalid(invalid_toolset): | |
mock_llm = Mock() | |
mock_toolset = { 'test_toolset': Mock() } | |
mock_vectorstore = Mock() | |
worker = WorkerUltraNode(mock_llm, mock_toolset, mock_vectorstore) | |
with pytest.raises(TypeError): | |
worker.add_toolset(invalid_toolset) | |
def test_run_invalid_prompt(invalid_prompt): | |
mock_llm = Mock() | |
mock_toolset = { 'test_toolset': Mock() } | |
mock_vectorstore = Mock() | |
worker = WorkerUltraNode(mock_llm, mock_toolset, mock_vectorstore) | |
with pytest.raises((TypeError, ValueError)): | |
worker.run(invalid_prompt) | |
def test_run_valid_prompt(mocker): | |
mock_llm = Mock() | |
mock_toolset = { 'test_toolset': Mock() } | |
mock_vectorstore = Mock() | |
worker = WorkerUltraNode(mock_llm, mock_toolset, mock_vectorstore) | |
mocker.patch.object(worker, 'create_agent') | |
assert worker.run('Test prompt') == 'Task completed by WorkerNode' | |
def test_worker_node(): | |
worker = worker_ultra_node('test-key') | |
assert isinstance(worker, WorkerUltraNode) | |
def test_worker_node_no_key(): | |
with pytest.raises(ValueError): | |
worker_ultra_node(None) | |