diff options
Diffstat (limited to 'tests/integration/test_setup.py')
-rw-r--r-- | tests/integration/test_setup.py | 73 |
1 files changed, 0 insertions, 73 deletions
diff --git a/tests/integration/test_setup.py b/tests/integration/test_setup.py deleted file mode 100644 index b74eebafc..000000000 --- a/tests/integration/test_setup.py +++ /dev/null @@ -1,73 +0,0 @@ -from unittest.mock import patch - -import pytest - -from autogpt.config.ai_config import AIConfig -from autogpt.setup import generate_aiconfig_automatic, prompt_user - - -@pytest.mark.vcr -@pytest.mark.requires_openai_api_key -def test_generate_aiconfig_automatic_default(patched_api_requestor, config): - user_inputs = [""] - with patch("autogpt.utils.session.prompt", side_effect=user_inputs): - ai_config = prompt_user(config) - - assert isinstance(ai_config, AIConfig) - assert ai_config.ai_name is not None - assert ai_config.ai_role is not None - assert 1 <= len(ai_config.ai_goals) <= 5 - - -@pytest.mark.vcr -@pytest.mark.requires_openai_api_key -def test_generate_aiconfig_automatic_typical(patched_api_requestor, config): - user_prompt = "Help me create a rock opera about cybernetic giraffes" - ai_config = generate_aiconfig_automatic(user_prompt, config) - - assert isinstance(ai_config, AIConfig) - assert ai_config.ai_name is not None - assert ai_config.ai_role is not None - assert 1 <= len(ai_config.ai_goals) <= 5 - - -@pytest.mark.vcr -@pytest.mark.requires_openai_api_key -def test_generate_aiconfig_automatic_fallback(patched_api_requestor, config): - user_inputs = [ - "T&GF£OIBECC()!*", - "Chef-GPT", - "an AI designed to browse bake a cake.", - "Purchase ingredients", - "Bake a cake", - "", - "", - ] - with patch("autogpt.utils.session.prompt", side_effect=user_inputs): - ai_config = prompt_user(config) - - assert isinstance(ai_config, AIConfig) - assert ai_config.ai_name == "Chef-GPT" - assert ai_config.ai_role == "an AI designed to browse bake a cake." - assert ai_config.ai_goals == ["Purchase ingredients", "Bake a cake"] - - -@pytest.mark.vcr -@pytest.mark.requires_openai_api_key -def test_prompt_user_manual_mode(patched_api_requestor, config): - user_inputs = [ - "--manual", - "Chef-GPT", - "an AI designed to browse bake a cake.", - "Purchase ingredients", - "Bake a cake", - "", - "", - ] - with patch("autogpt.utils.session.prompt", side_effect=user_inputs): - ai_config = prompt_user(config) - - assert isinstance(ai_config, AIConfig) - assert ai_config.ai_name == "Chef-GPT" - assert ai_config.ai_role == "an AI designed to browse bake a cake." - assert ai_config.ai_goals == ["Purchase ingredients", "Bake a cake"] |