aboutsummaryrefslogtreecommitdiff
path: root/autogpts/autogpt/tests/unit/test_prompt_config.py
diff options
context:
space:
mode:
Diffstat (limited to 'autogpts/autogpt/tests/unit/test_prompt_config.py')
-rw-r--r--autogpts/autogpt/tests/unit/test_prompt_config.py42
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"