mirror of
https://github.com/johndoe6345789/AutoMetabuilder.git
synced 2026-04-24 13:54:59 +00:00
15 lines
430 B
Python
15 lines
430 B
Python
import unittest
|
|
from autometabuilder.main import load_prompt_yaml
|
|
import os
|
|
|
|
class TestMain(unittest.TestCase):
|
|
def test_load_prompt_yaml(self):
|
|
# This test assumes prompt.yml exists in the root
|
|
if os.path.exists("prompt.yml"):
|
|
config = load_prompt_yaml()
|
|
self.assertIsInstance(config, dict)
|
|
self.assertIn("messages", config)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|