MEGAINFRA-WEBAPP/scripts/site-model/test_model_keys.py

21 lines
639 B
Python
Raw Normal View History

import importlib.util
import json
import unittest
from pathlib import Path
HERE = Path(__file__).resolve().parent
FIXTURE = HERE.parent.parent / "src/lib/site-map/__fixtures__/mini-model"
spec = importlib.util.spec_from_file_location("model_keys", HERE / "model_keys.py")
model_keys = importlib.util.module_from_spec(spec)
spec.loader.exec_module(model_keys)
class ModelKeysTests(unittest.TestCase):
def test_same_keys_as_typescript(self):
expected = json.loads((FIXTURE / "expected-keys.json").read_text())
self.assertEqual(model_keys.model_keys(FIXTURE), expected)
if __name__ == "__main__":
unittest.main()