diff options
Diffstat (limited to 'autogpts/autogpt/tests/unit/test_prompt_config.py')
-rw-r--r-- | autogpts/autogpt/tests/unit/test_prompt_config.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/autogpts/autogpt/tests/unit/test_prompt_config.py b/autogpts/autogpt/tests/unit/test_prompt_config.py new file mode 100644 index 000000000..ccadb191d --- /dev/null +++ b/autogpts/autogpt/tests/unit/test_prompt_config.py @@ -0,0 +1,42 @@ +from autogpt.config.ai_directives import AIDirectives + +""" +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 +best_practices: +- A test best-practice +- Another test best-practice +- A third test best-practice +""" + prompt_settings_file = tmp_path / "test_prompt_settings.yaml" + prompt_settings_file.write_text(yaml_content) + + prompt_config = AIDirectives.from_file(prompt_settings_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.best_practices) == 3 + assert prompt_config.best_practices[0] == "A test best-practice" + assert prompt_config.best_practices[1] == "Another test best-practice" + assert prompt_config.best_practices[2] == "A third test best-practice" |