Spaces:
Running
Running
File size: 1,187 Bytes
d18b34d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import os
import tempfile
import unittest
import yaml
from src.utils.config import Config, load_yaml
class TestConfig(unittest.TestCase):
def setUp(self):
self.temp_dir = tempfile.TemporaryDirectory()
self.yaml_path = os.path.join(self.temp_dir.name, "config.yaml")
def tearDown(self):
self.temp_dir.cleanup()
def test_load_yaml_valid(self):
data = {"key": "value", "nested": {"k": 1}}
with open(self.yaml_path, "w") as f:
yaml.dump(data, f)
config = load_yaml(self.yaml_path)
self.assertIsInstance(config, Config)
self.assertEqual(config.data["key"], "value")
self.assertEqual(config.data["nested"]["k"], 1)
def test_load_yaml_invalid_structure(self):
# List at root instead of dict
data = ["item1", "item2"]
with open(self.yaml_path, "w") as f:
yaml.dump(data, f)
with self.assertRaises(ValueError):
load_yaml(self.yaml_path)
def test_load_yaml_file_not_found(self):
with self.assertRaises(FileNotFoundError):
load_yaml("non_existent_file.yaml")
if __name__ == "__main__":
unittest.main()
|