fix TestClass name, make module level variable all capital and other code cleanup

.github/workflows/lint.yml

@@ -26,5 +26,4 @@ jobs:
       - name: mypy
         run: |
           mypy --version
-          rm -rf .mypy_cache
           mypy src

.pre-commit-config.yaml

@@ -1,4 +1,4 @@
-exclude: ^(docs|tests/files|cmd_line|tasks.py)
+exclude: ^(docs|tests/test_files|tasks.py)
 
 ci:
   autoupdate_schedule: monthly

src/monty/subprocess.py

@@ -74,11 +74,9 @@ def run(self, timeout: Optional[float] = None, **kwargs) -> Self:
 
         def target(**kw):
             try:
-                # print('Thread started')
                 with Popen(self.command, **kw) as self.process:
                     self.output, self.error = self.process.communicate()
                     self.retcode = self.process.returncode
-                # print('Thread stopped')
             except Exception:
                 self.error = traceback.format_exc()
                 self.retcode = -1
@@ -96,7 +94,6 @@ def target(**kw):
         thread.join(timeout)
 
         if thread.is_alive() and self.process is not None:
-            # print("Terminating process")
             self.process.terminate()
             self.killed = True
             thread.join()

src/monty/termcolor.py

@@ -128,12 +128,7 @@ def cprint(
 
     It accepts arguments of print function.
     """
-    try:
-        print((colored(text, color, on_color, attrs)), **kwargs)
-    except TypeError:
-        # flush is not supported by py2.7
-        kwargs.pop("flush", None)
-        print((colored(text, color, on_color, attrs)), **kwargs)
+    print((colored(text, color, on_color, attrs)), **kwargs)
 
 
 def colored_map(text: str, cmap: dict) -> str:
@@ -162,12 +157,7 @@ def cprint_map(text: str, cmap: dict, **kwargs) -> None:
     Examples:
         cprint_map("Hello world", {"Hello": "red"})
     """
-    try:
-        print(colored_map(text, cmap), **kwargs)
-    except TypeError:
-        # flush is not supported by py2.7
-        kwargs.pop("flush", None)
-        print(colored_map(text, cmap), **kwargs)
+    print(colored_map(text, cmap), **kwargs)
 
 
 def get_terminal_size():

tests/test_design_patterns.py

@@ -53,32 +53,32 @@ def test_pickle(self):
 
 
 @cached_class
-class TestClass:
+class ClassForTest:
     def __init__(self, value: Any) -> None:
         self.value = value
 
 
 def test_caching():
     # Test that instances are cached
-    inst1 = TestClass(1)
-    inst2 = TestClass(1)
+    inst1 = ClassForTest(1)
+    inst2 = ClassForTest(1)
     assert inst1 is inst2
 
-    inst3 = TestClass(2)
+    inst3 = ClassForTest(2)
     assert inst1 is not inst3
 
 
 def test_picklability():
     # Test that instances can be pickled and unpickled
-    original = TestClass(42)
+    original = ClassForTest(42)
     pickled = pickle.dumps(original)
     unpickled = pickle.loads(pickled)
 
     # Check that the unpickled instance has the same value
     assert original.value == unpickled.value
 
     # Check that the unpickled instance is the same as a newly created instance
-    new_instance = TestClass(42)
+    new_instance = ClassForTest(42)
     assert unpickled is new_instance
 
 
@@ -98,8 +98,8 @@ def __init__(self):
 
 def test_class_identity():
     # Ensure the decorated class is still recognized as the original class
-    assert isinstance(TestClass(1), TestClass)
-    assert type(TestClass(1)) is TestClass
+    assert isinstance(ClassForTest(1), ClassForTest)
+    assert type(ClassForTest(1)) is ClassForTest
 
 
 def test_multiple_arguments():

tests/test_dev.py

@@ -143,7 +143,7 @@ class TestClassNew:
             """A dummy class for tests."""
 
             def __post_init__(self):
-                print("Hello.")
+                pass
 
             def method_a(self):
                 pass
@@ -156,7 +156,7 @@ class TestClassOld:
             class_attrib_old = "OLD_ATTRIB"
 
             def __post_init__(self):
-                print("Hello.")
+                pass
 
             def method_b(self):
                 """This is method_b."""
@@ -246,7 +246,7 @@ def test_requires(self):
 
         @requires(fictitious_mod is not None, err_msg)
         def use_fictitious_mod():
-            print("success")
+            pass
 
         with pytest.raises(RuntimeError, match=err_msg):
             use_fictitious_mod()

tests/test_json.py

@@ -678,7 +678,7 @@ def test_pint_quantity(self):
         cls = ClassContainingQuantity(qty=pint.Quantity("9.81 m/s**2"))
 
         d = json.loads(MontyEncoder().encode(cls))
-        print(d)
+        assert isinstance(d, dict)
 
         assert d["qty"]["@module"] == "pint"
         assert d["qty"]["@class"] == "Quantity"

tests/test_pprint.py

@@ -34,4 +34,4 @@ def __str__(self):
             ],
         )
 
-        print(draw_tree(root))
+        assert draw_tree(root)

tests/test_subprocess.py

@@ -8,8 +8,9 @@ def test_command():
     sleep05 = Command("sleep 0.5")
 
     sleep05.run(timeout=1)
-    print(sleep05)
-    assert sleep05.retcode == 0
+    # DEBUG: this unit test fail in Win CI sometimes
+    full_msg = f"{sleep05=}\n{sleep05.error=}\n{sleep05.output}"
+    assert sleep05.retcode == 0, full_msg
     assert not sleep05.killed
 
     sleep05.run(timeout=0.1)