diff --git a/tests/visualization/reproduce_notebooks.py b/tests/visualization/reproduce_notebooks.py
index db5ca9ac..752fa836 100644
--- a/tests/visualization/reproduce_notebooks.py
+++ b/tests/visualization/reproduce_notebooks.py
@@ -6,7 +6,6 @@ from datetime import datetime
 import logging
 
 import pytest
-
 # Add project root to path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
 
@@ -24,7 +23,7 @@ from semantica.ontology import OntologyGenerator
 from semantica.embeddings import EmbeddingGenerator
 
 pytestmark = pytest.mark.integration
-
+# Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("reproduce_notebooks")
 
diff --git a/tests/visualization/test_visualization_comprehensive.py b/tests/visualization/test_visualization_comprehensive.py
index 82b79dd0..6f0d6997 100644
--- a/tests/visualization/test_visualization_comprehensive.py
+++ b/tests/visualization/test_visualization_comprehensive.py
@@ -5,7 +5,7 @@ import numpy as np
 from pathlib import Path
 
 import pytest
-
+# Mock heavy libraries before importing visualization modules
 sys.modules['matplotlib'] = MagicMock()
 sys.modules['matplotlib.pyplot'] = MagicMock()
 sys.modules['matplotlib.colors'] = MagicMock()
@@ -33,7 +33,6 @@ from semantica.visualization.temporal_visualizer import TemporalVisualizer
 from semantica.visualization.utils.color_schemes import ColorScheme
 
 pytestmark = pytest.mark.integration
-
 class TestVisualizationComprehensive(unittest.TestCase):
 
     def setUp(self):