formatting
parent
94845aa0e1
commit
8186ccb56a
|
@ -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
|
||||
|
|
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue