/
githubmirror
/
Multi-GPT
Обзор
Документация
Войти
/
githubmirror
/
Multi-GPT
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tests/unit/test_commands.py
18 строк
815 B
Itamar Friedman
fixing tests to fit latest merges into master
16 апр 2023, 01:19
16 апр 2023, 01:19
5a87000
Код
Авторство
О чём код?
import autogpt.agent.agent_manager as agent_manager from autogpt.app import start_agent, list_agents, execute_command import unittest from unittest.mock import patch, MagicMock class TestCommands(unittest.TestCase): def test_make_agent(self): with patch("openai.ChatCompletion.create") as mock: obj = MagicMock() obj.response.choices[0].messages[0].content = "Test message" mock.return_value = obj start_agent("Test Agent", "chat", "Hello, how are you?", "gpt2") agents = list_agents() self.assertEqual("List of agents:\n0: chat", agents) start_agent("Test Agent 2", "write", "Hello, how are you?", "gpt2") agents = list_agents() self.assertEqual("List of agents:\n0: chat\n1: write", agents)