diff --git a/scripts/prompt.py b/scripts/prompt.py index 286002ee..188603a3 100644 --- a/scripts/prompt.py +++ b/scripts/prompt.py @@ -1,9 +1,10 @@ from promptgenerator import PromptGenerator + def get_prompt(): """ This function generates a prompt string that includes various constraints, commands, resources, and performance evaluations. - + Returns: str: The generated prompt string. """ @@ -58,5 +59,5 @@ def get_prompt(): # Generate the prompt string prompt_string = prompt_generator.generate_prompt_string() - + return prompt_string diff --git a/tests/promptgenerator_tests.py b/tests/promptgenerator_tests.py index ac5c3a79..181fdea6 100644 --- a/tests/promptgenerator_tests.py +++ b/tests/promptgenerator_tests.py @@ -7,6 +7,7 @@ import os sys.path.append(os.path.abspath("../scripts")) from promptgenerator import PromptGenerator + # Create a test class for the PromptGenerator, subclassed from unittest.TestCase class promptgenerator_tests(unittest.TestCase): @@ -81,7 +82,7 @@ class promptgenerator_tests(unittest.TestCase): self.assertIn(constraint, prompt_string) for command in commands: self.assertIn(command["name"], prompt_string) - + # Check for each key-value pair in the command args dictionary for key, value in command["args"].items(): self.assertIn(f'"{key}": "{value}"', prompt_string) @@ -94,6 +95,7 @@ class promptgenerator_tests(unittest.TestCase): self.assertIn("resources", prompt_string.lower()) self.assertIn("performance evaluation", prompt_string.lower()) + # Run the tests when this script is executed if __name__ == '__main__': unittest.main()