mirror of
https://github.com/johndoe6345789/AutoMetabuilder.git
synced 2026-04-25 22:35:01 +00:00
16 lines
432 B
Python
16 lines
432 B
Python
import os
|
|
import unittest
|
|
|
|
from autometabuilder.utils import load_prompt_yaml
|
|
|
|
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()
|