added feature to remove transitions related to removed level.

doc/source/changelog.rst

@@ -4,6 +4,14 @@ Changelog
 All notable changes to this project will be documented in this file.  This
 project adheres to `Semantic Versioning <http://semver.org/spec/v2.0.0.html>`_.
 
+Version 3.0.3
+-------------
+
+Fix:
+
+  * Removing a level now removes all transitions to and from the level before removing 
+    the level itself. This alleviates any issues that arise after removing the level.
+
 Version 3.0.2
 -------------
 

lvlspy/__about__.py

@@ -10,6 +10,6 @@
 
 __title__ = "lvlspy"
 __summary__ = "Python project to work with quantum level system data"
-__version__ = "3.0.2"
+__version__ = "3.0.3"
 __author__ = "Clemson University"
 __copyright__ = "Clemson University, 2022-2023"

lvlspy/species.py

@@ -69,9 +69,23 @@ def remove_level(self, level):
             ``level`` (:obj:`lvlspy.level.Level`) The level to be removed.
 
         Return:
-            On successful return, the level has been removed.
+            On successful return, the level and all connected transitions have been removed.
 
         """
+        upper_conn = self.get_upper_linked_levels(level)
+        lower_conn = self.get_lower_linked_levels(level)
+
+        for l in upper_conn:
+            t = self.get_level_to_level_transition(l,level)
+            if t is None:
+                continue
+            self.remove_transition(t)
+
+        for l in lower_conn:
+            t = self.get_level_to_level_transition(level,l)
+            if t is None:
+                continue
+            self.remove_transition(t)
 
         self.levels.remove(level)