formatting

pull/837/head
Alrik Olson 2023-04-13 11:36:48 -07:00
parent 94845aa0e1
commit 8186ccb56a
2 changed files with 6 additions and 3 deletions

View File

@ -1,5 +1,6 @@
from promptgenerator import PromptGenerator
def get_prompt():
"""
This function generates a prompt string that includes various constraints, commands, resources, and performance evaluations.

View File

@ -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):
@ -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()