checks for system dynamic and system energy is added

ubermag · Jan 23, 2024 · 9aaeb9b · 9aaeb9b
1 parent 41d36ac
commit 9aaeb9b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/oommfc/drivers/mindriver.py b/oommfc/drivers/mindriver.py
@@ -58,7 +58,11 @@ class MinDriver(Driver):
 
     def _checkargs(self, **kwargs):
         pass  # no kwargs should be checked
-
+    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")
+
     @property
     def _x(self):
         return "iteration"
diff --git a/oommfc/drivers/timedriver.py b/oommfc/drivers/timedriver.py
@@ -66,7 +66,13 @@ def _checkargs(self, **kwargs):
         if n <= 0:
             msg = f"Cannot drive with {n=}."
             raise ValueError(msg)
-
+    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")
+        if len(system.energy)==0:
+            raise AttributeError("System's energy is not defined")
+
     @property
     def _x(self):
         return "t"