Raised error value changed form Attribute to RunTime

ubermag · Jan 30, 2024 · d99176a · d99176a
1 parent 38a823a
commit d99176a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/oommfc/drivers/hysteresisdriver.py b/oommfc/drivers/hysteresisdriver.py
@@ -73,6 +73,11 @@ def _checkargs(self, **kwargs):
         if n - 1 <= 0:  # OOMMF counts steps, not points (n -> n-1)
             msg = f"Cannot drive with {n=}."
             raise ValueError(msg)
+
+    def _check_system(self, system):
+        """Checks the system has energy in it"""
+        if len(system.energy) == 0:
+            raise RuntimeError("System's energy is not defined")
 
     @property
     def _x(self):

diff --git a/oommfc/drivers/mindriver.py b/oommfc/drivers/mindriver.py
@@ -62,7 +62,7 @@ def _checkargs(self, **kwargs):
     def _check_system(self, system):
         """Checks the system has energy in it"""
         if len(system.energy) == 0:
-            raise AttributeError("System's energy is not defined")
+            raise RuntimeError("System's energy is not defined")
 
     @property
     def _x(self):

diff --git a/oommfc/drivers/timedriver.py b/oommfc/drivers/timedriver.py
@@ -70,9 +70,9 @@ def _checkargs(self, **kwargs):
     def _check_system(self, system):
         """Checks the system has dynamics in it"""
         if len(system.dynamics) == 0:
-            raise AttributeError("System's dynamics is not defined")
+            raise RuntimeError("System's dynamics is not defined")
         if len(system.energy) == 0:
-            raise AttributeError("System's energy is not defined")
+            raise RuntimeError("System's energy is not defined")
 
     @property
     def _x(self):