Update tests/test_utils_logcfg.py

tests/test_utils_logcfg.py

@@ -7,23 +7,23 @@ import os
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 from slic.utils.logcfg import *
 
-# Test : add_log_Level()
-
-# Fixture: Register custom log levels once for the module
-@pytest.fixture(scope="module", autouse=True)
-def setup_custom_levels():
-    # Add custom log levels with names, numeric values, functions, and colors
-    add_log_Level(log, "LONG", logging.INFO + 5, func_name="long", color="cyan")
-    add_log_Level(log, "ENLARGE", logging.INFO + 6, func_name="enlarge", color="green")
-    # Set default log level (will display LONG, ENLARGE and higher)
-    logcfg("LONG")
-
-# ANSI escape codes to verify colors
+# ANSI escape codes used for color testing
 ANSI_CYAN = "\x1b[36m"
 ANSI_GREEN = "\x1b[32m"
+ANSI_MAGENTA = "\x1b[35m"
 ANSI_RESET = "\x1b[0m"
 
-# Parametrized test: custom log levels
+# Register custom log levels once for the test module
+@pytest.fixture(scope="module", autouse=True)
+def setup_custom_levels():
+    # Add custom levels
+    add_log_Level(log, "LONG", logging.INFO + 5, func_name="long", color="cyan")
+    add_log_Level(log, "ENLARGE", logging.INFO + 6, func_name="enlarge", color="green")
+    add_log_Level(log, "TRACE", logging.DEBUG - 1, func_name="trace", color="magenta")
+    
+    logcfg("TRACE")  # Enable all levels >= TRACE
+
+# Test: add_log_Level()
 @pytest.mark.parametrize("levelname, logfunc, message, ansi_color", [
     ("LONG", lambda msg: log.long(msg), "This is a LONG message", ANSI_CYAN),
     ("ENLARGE", lambda msg: log.enlarge(msg), "Please ENLARGE this!", ANSI_GREEN),
@@ -31,46 +31,41 @@ ANSI_RESET = "\x1b[0m"
 def test_custom_levels_are_logged(levelname, logfunc, message, ansi_color, capsys):
     # Call the custom log function
     logfunc(message)
+    
     # Capture stderr where logzero outputs messages
     captured = capsys.readouterr().err
 
+    # Assert color, message, and ANSI reset code are present
     assert ansi_color in captured, f"Expected color code {ansi_color} not found in log output"
-    assert message in captured, "Expected log message not found in output"
+    assert message in captured, f"Expected log message '{message}' not found in output"
     assert captured.strip().endswith(ANSI_RESET), "Missing ANSI reset code at end of log line"
 
-# Test : setup_import_logging()
-
+# Test: setup_import_logging()
 def test_import_logging_multiple_modules(capfd):
-    # Setup custom TRACE log level with color
-    add_log_Level(log, "TRACE", logging.DEBUG - 1, color="magenta")
 
-    # Initialize both logzero and logging configurations to use TRACE level
-    logcfg("TRACE")
-
-    # Fuction to test
     setup_import_logging()
 
-    # Perform various imports
+    # Perform imports 
     import json
     import math
-    import io        
-    import json      # Duplicate
-    import math      # Duplicate
+    import io
+    import json   # Duplicate
+    import math   # Duplicate
     import random
-    import random    # Duplicate
+    import random # Duplicate
 
-    # Capture the printed output
+    # Capture stdout/stderr output
     out, err = capfd.readouterr()
 
-    # Check that each unique module import appears once in the logs
+    # Expected modules to be logged once
     expected_imports = {"json", "math", "io", "random"}
     for mod in expected_imports:
-        assert f"importing: {mod}" in out
+        assert f"importing: {mod}" in out, f"Missing import log for {mod}"
 
-    # Ensure duplicates are only logged once
+    # Check duplicates only logged once
     assert out.count("importing: json") == 1
     assert out.count("importing: math") == 1
     assert out.count("importing: random") == 1
 
-    # Verify that our custom log level is actually used
+    # Check custom TRACE level is used
     assert "TRACE" in out.upper()