From 22afca7b1d9931cd82d0e326b117ad3731637365 Mon Sep 17 00:00:00 2001 From: Christian Hill Date: Thu, 2 May 2024 09:58:33 +0200 Subject: [PATCH] Fix bug on states import --- setup.py | 2 +- src/pyvalem/states/__init__.py | 1 + tests/test_imports.py | 9 +++++++++ 3 files changed, 11 insertions(+), 1 deletion(-) create mode 100644 tests/test_imports.py diff --git a/setup.py b/setup.py index 23f8b11..0590d59 100644 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ setup( name="pyvalem", - version="2.6", + version="2.6.1", description="A package for managing simple chemical species and states", long_description=long_description, long_description_content_type="text/x-rst", diff --git a/src/pyvalem/states/__init__.py b/src/pyvalem/states/__init__.py index 05994e8..21d6cab 100644 --- a/src/pyvalem/states/__init__.py +++ b/src/pyvalem/states/__init__.py @@ -12,3 +12,4 @@ from .vibrational_state import VibrationalState, VibrationalStateError from .generic_excited_state import GenericExcitedState, GenericExcitedStateError from .key_value_pair import KeyValuePair, KeyValuePairError +from .compound_LS_coupling import CompoundLSCoupling, CompoundLSCouplingError diff --git a/tests/test_imports.py b/tests/test_imports.py new file mode 100644 index 0000000..9b40674 --- /dev/null +++ b/tests/test_imports.py @@ -0,0 +1,9 @@ +import unittest + +from pyvalem import states + + +class StatesImportTest(unittest.TestCase): + def test_state_types_import(self): + self.assertIn("AtomicTermSymbol", states.__dict__) + self.assertIn("CompoundLSCoupling", states.__dict__)