TL: conveniency changes on utilities

Author: tlunet

qmat/qcoeff/__init__.py

@@ -216,7 +216,7 @@ def register(cls: type[T]) -> type[T]:
     """Class decorator to register a specialized :class:`QGenerator` class in qmat"""
     # Check for correct overriding
     for name in ["nodes", "Q", "weights", "order"]:
-        checkOverriding(cls, name)
+        checkOverriding(cls, name, isProperty=True)
     # Check that TEST_PARAMS are given and valid if no default constructor
     try:
         cls()

qmat/qdelta/__init__.py

@@ -217,7 +217,7 @@ def genCoeffs(self, k=None, form="Z2N", dTau=False):
 def register(cls: type[T]) -> type[T]:
     """Class decorator to register a specialized :class:`QDeltaGenerator` class in `qmat`"""
     checkGenericConstr(cls)
-    checkOverriding(cls, "computeQDelta", isProperty=False)
+    checkOverriding(cls, "computeQDelta")
     try:
         sig = inspect.signature(cls.computeQDelta)
         par = sig.parameters["k"]

qmat/solvers/generic/diffops.py

@@ -18,7 +18,7 @@
 
 def registerDiffOp(cls: type[T]) -> type[T]:
     """Class decorator to register a specialized :class:`DiffOp` class in `qmat`"""
-    checkOverriding(cls, "evalF", isProperty=False)
+    checkOverriding(cls, "evalF")
     storeClass(cls, DIFFOPS)
     return cls
 

qmat/utils/__init__.py

@@ -14,10 +14,10 @@
 from time import time
 
 
-def checkOverriding(cls, name, isProperty=True):
+def checkOverriding(cls, name, isProperty=False, mroIndex=-2):
     """Check if a class overrides a method with a given name"""
     method = getattr(cls, name)
-    parent = getattr(cls.mro()[-2], name)
+    parent = getattr(cls.mro()[mroIndex], name)
     assert method != parent, \
         f"{name} method must be overriden in {cls.__name__} class"
     if isProperty: