delete doctest.testmod() if __name__ == "__main__"

materialsproject · Jul 25, 2023 · 26b3b5b · 26b3b5b
1 parent 1d3ff84
commit 26b3b5b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/pymatgen/core/composition.py b/pymatgen/core/composition.py
@@ -1295,9 +1295,3 @@ def __repr__(self):
 
 class CompositionError(Exception):
     """Exception class for composition errors."""
-
-
-if __name__ == "__main__":
-    import doctest
-
-    doctest.testmod()
diff --git a/pymatgen/core/units.py b/pymatgen/core/units.py
@@ -789,9 +789,9 @@ def get_mass():
 
     """
 
-    def wrap(f):
+    def wrap(func):
         def wrapped_f(*args, **kwargs):
-            val = f(*args, **kwargs)
+            val = func(*args, **kwargs)
             unit_type = _UNAME2UTYPE[unit]
 
             if isinstance(val, (FloatWithUnit, ArrayWithUnit)):
@@ -816,9 +816,3 @@ def wrapped_f(*args, **kwargs):
         return wrapped_f
 
     return wrap
-
-
-if __name__ == "__main__":
-    import doctest
-
-    doctest.testmod()