diff options
Diffstat (limited to 'tests/unit/test_agent.py')
-rw-r--r-- | tests/unit/test_agent.py | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/tests/unit/test_agent.py b/tests/unit/test_agent.py deleted file mode 100644 index 3fb896bad..000000000 --- a/tests/unit/test_agent.py +++ /dev/null @@ -1,47 +0,0 @@ -from unittest.mock import MagicMock - -import pytest - -from autogpt.agent import Agent -from autogpt.config import AIConfig -from autogpt.config.config import Config - - -@pytest.fixture -def agent(config: Config): - ai_name = "Test AI" - memory = MagicMock() - next_action_count = 0 - command_registry = MagicMock() - ai_config = AIConfig(ai_name=ai_name) - system_prompt = "System prompt" - triggering_prompt = "Triggering prompt" - workspace_directory = "workspace_directory" - - agent = Agent( - ai_name=ai_name, - memory=memory, - next_action_count=next_action_count, - command_registry=command_registry, - ai_config=ai_config, - config=config, - system_prompt=system_prompt, - triggering_prompt=triggering_prompt, - workspace_directory=workspace_directory, - ) - return agent - - -def test_agent_initialization(agent: Agent): - assert agent.ai_name == "Test AI" - assert agent.memory == agent.memory - assert agent.history.messages == [] - assert agent.next_action_count == 0 - assert agent.command_registry == agent.command_registry - assert agent.ai_config == agent.ai_config - assert agent.system_prompt == "System prompt" - assert agent.triggering_prompt == "Triggering prompt" - - -# More test methods can be added for specific agent interactions -# For example, mocking chat_with_ai and testing the agent's interaction loop |