mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-01-18 21:44:22 +01:00
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from autogpt.app.setup import (
|
|
apply_overrides_to_ai_settings,
|
|
interactively_revise_ai_settings,
|
|
)
|
|
from autogpt.config.ai_profile import AIProfile
|
|
from autogpt.config import AIDirectives, Config
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_overrides_to_ai_settings():
|
|
ai_profile = AIProfile(ai_name="Test AI", ai_role="Test Role")
|
|
directives = AIDirectives(resources=["Resource1"], constraints=["Constraint1"], best_practices=["BestPractice1"])
|
|
|
|
apply_overrides_to_ai_settings(ai_profile, directives, override_name="New AI", override_role="New Role", replace_directives=True, resources=["NewResource"], constraints=["NewConstraint"], best_practices=["NewBestPractice"])
|
|
|
|
assert ai_profile.ai_name == "New AI"
|
|
assert ai_profile.ai_role == "New Role"
|
|
assert directives.resources == ["NewResource"]
|
|
assert directives.constraints == ["NewConstraint"]
|
|
assert directives.best_practices == ["NewBestPractice"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_interactively_revise_ai_settings(config: Config):
|
|
ai_profile = AIProfile(ai_name="Test AI", ai_role="Test Role")
|
|
directives = AIDirectives(resources=["Resource1"], constraints=["Constraint1"], best_practices=["BestPractice1"])
|
|
|
|
user_inputs = [
|
|
"y",
|
|
"New AI",
|
|
"New Role",
|
|
"NewConstraint",
|
|
"",
|
|
"NewResource",
|
|
"",
|
|
"NewBestPractice",
|
|
"",
|
|
]
|
|
with patch("autogpt.app.utils.clean_input", side_effect=user_inputs):
|
|
ai_profile, directives = await interactively_revise_ai_settings(ai_profile, directives, config)
|
|
|
|
assert ai_profile.ai_name == "New AI"
|
|
assert ai_profile.ai_role == "New Role"
|
|
assert directives.resources == ["NewResource"]
|
|
assert directives.constraints == ["NewConstraint"]
|
|
assert directives.best_practices == ["NewBestPractice"]
|