mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-24 01:14:22 +01:00
dafbd11686
* Rearrange tests into unit/integration/challenge categories * Fix linting + `tests.challenges` imports * Fix obscured duplicate test in test_url_validation.py * Move VCR conftest to tests.vcr * Specify tests to run & their order (unit -> integration -> challenges) in CI * Fail Docker CI when tests fail * Fix import & linting errors in tests * Fix `get_text_summary` * Fix linting errors * Clean up pytest args in CI * Remove bogus tests from GoCodeo
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from autogpt.config.prompt_config import PromptConfig
|
|
|
|
"""
|
|
Test cases for the PromptConfig class, which handles loads the Prompts configuration
|
|
settings from a YAML file.
|
|
"""
|
|
|
|
|
|
def test_prompt_config_loading(tmp_path):
|
|
"""Test if the prompt configuration loads correctly"""
|
|
|
|
yaml_content = """
|
|
constraints:
|
|
- A test constraint
|
|
- Another test constraint
|
|
- A third test constraint
|
|
resources:
|
|
- A test resource
|
|
- Another test resource
|
|
- A third test resource
|
|
performance_evaluations:
|
|
- A test performance evaluation
|
|
- Another test performance evaluation
|
|
- A third test performance evaluation
|
|
"""
|
|
config_file = tmp_path / "test_prompt_settings.yaml"
|
|
config_file.write_text(yaml_content)
|
|
|
|
prompt_config = PromptConfig(config_file)
|
|
|
|
assert len(prompt_config.constraints) == 3
|
|
assert prompt_config.constraints[0] == "A test constraint"
|
|
assert prompt_config.constraints[1] == "Another test constraint"
|
|
assert prompt_config.constraints[2] == "A third test constraint"
|
|
assert len(prompt_config.resources) == 3
|
|
assert prompt_config.resources[0] == "A test resource"
|
|
assert prompt_config.resources[1] == "Another test resource"
|
|
assert prompt_config.resources[2] == "A third test resource"
|
|
assert len(prompt_config.performance_evaluations) == 3
|
|
assert prompt_config.performance_evaluations[0] == "A test performance evaluation"
|
|
assert (
|
|
prompt_config.performance_evaluations[1]
|
|
== "Another test performance evaluation"
|
|
)
|
|
assert (
|
|
prompt_config.performance_evaluations[2]
|
|
== "A third test performance evaluation"
|
|
)
|