Add unittest to ensure state classes are imported by states.__init__.py

xnx · Jun 12, 2024 · dbfa0ee · dbfa0ee
1 parent d12e699
commit dbfa0ee
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_imports.py b/tests/test_imports.py
@@ -1,9 +1,15 @@
 import unittest
 
 from pyvalem import states
+from pyvalem.states._state_parser import STATES
 
 
 class StatesImportTest(unittest.TestCase):
     def test_state_types_import(self):
         self.assertIn("AtomicTermSymbol", states.__dict__)
         self.assertIn("CompoundLSCoupling", states.__dict__)
+
+class StatesInitFileTest(unittest.TestCase):
+    def test___init___imports(self):
+        for StateClass in STATES.keys():
+            self.assertIn(StateClass.__name__, states.__dict__)