mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-08-29 04:26:20 +00:00
Fix dependency issues, align GraphRAG notebook, and update changelog
This commit is contained in:
KaifAhmad1
@@ -1,227 +0,0 @@
|
||||
import unittest
|
||||
from unittest.mock import MagicMock, patch
|
||||
import sys
|
||||
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()
|
||||
sys.modules['matplotlib.patches'] = MagicMock()
|
||||
sys.modules['plotly'] = MagicMock()
|
||||
sys.modules['plotly.express'] = MagicMock()
|
||||
sys.modules['plotly.graph_objects'] = MagicMock()
|
||||
sys.modules['plotly.subplots'] = MagicMock()
|
||||
sys.modules['seaborn'] = MagicMock()
|
||||
sys.modules['umap'] = MagicMock()
|
||||
sys.modules['sklearn'] = MagicMock()
|
||||
sys.modules['sklearn.decomposition'] = MagicMock()
|
||||
sys.modules['sklearn.manifold'] = MagicMock()
|
||||
sys.modules['networkx'] = MagicMock()
|
||||
sys.modules['graphviz'] = MagicMock()
|
||||
|
||||
# Import visualizers
|
||||
from semantica.visualization.kg_visualizer import KGVisualizer
|
||||
from semantica.visualization.ontology_visualizer import OntologyVisualizer
|
||||
from semantica.visualization.embedding_visualizer import EmbeddingVisualizer
|
||||
from semantica.visualization.semantic_network_visualizer import SemanticNetworkVisualizer
|
||||
from semantica.visualization.analytics_visualizer import AnalyticsVisualizer
|
||||
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):
|
||||
self.mock_logger = MagicMock()
|
||||
self.mock_tracker = MagicMock()
|
||||
|
||||
# Patch dependencies for all visualizers
|
||||
self.patchers = [
|
||||
patch('semantica.visualization.kg_visualizer.get_logger', return_value=self.mock_logger),
|
||||
patch('semantica.visualization.kg_visualizer.get_progress_tracker', return_value=self.mock_tracker),
|
||||
patch('semantica.visualization.ontology_visualizer.get_logger', return_value=self.mock_logger),
|
||||
patch('semantica.visualization.ontology_visualizer.get_progress_tracker', return_value=self.mock_tracker),
|
||||
patch('semantica.visualization.embedding_visualizer.get_logger', return_value=self.mock_logger),
|
||||
patch('semantica.visualization.embedding_visualizer.get_progress_tracker', return_value=self.mock_tracker),
|
||||
patch('semantica.visualization.semantic_network_visualizer.get_logger', return_value=self.mock_logger),
|
||||
patch('semantica.visualization.semantic_network_visualizer.get_progress_tracker', return_value=self.mock_tracker),
|
||||
patch('semantica.visualization.analytics_visualizer.get_logger', return_value=self.mock_logger),
|
||||
patch('semantica.visualization.analytics_visualizer.get_progress_tracker', return_value=self.mock_tracker),
|
||||
patch('semantica.visualization.temporal_visualizer.get_logger', return_value=self.mock_logger),
|
||||
patch('semantica.visualization.temporal_visualizer.get_progress_tracker', return_value=self.mock_tracker),
|
||||
# Mock Layouts
|
||||
patch('semantica.visualization.kg_visualizer.ForceDirectedLayout', MagicMock()),
|
||||
patch('semantica.visualization.kg_visualizer.HierarchicalLayout', MagicMock()),
|
||||
patch('semantica.visualization.kg_visualizer.CircularLayout', MagicMock()),
|
||||
patch('semantica.visualization.ontology_visualizer.HierarchicalLayout', MagicMock()),
|
||||
patch('semantica.visualization.semantic_network_visualizer.ForceDirectedLayout', MagicMock()),
|
||||
]
|
||||
|
||||
for p in self.patchers:
|
||||
p.start()
|
||||
|
||||
# Reset plotly mocks
|
||||
import plotly.graph_objects as go
|
||||
import plotly.express as px
|
||||
go.Figure.reset_mock()
|
||||
px.bar.reset_mock()
|
||||
px.scatter.reset_mock()
|
||||
|
||||
def tearDown(self):
|
||||
for p in self.patchers:
|
||||
p.stop()
|
||||
|
||||
# --- KGVisualizer Tests ---
|
||||
def test_kg_visualizer(self):
|
||||
viz = KGVisualizer()
|
||||
graph = {
|
||||
"entities": [{"id": "e1", "label": "E1", "type": "T1"}, {"id": "e2", "label": "E2", "type": "T2"}],
|
||||
"relationships": [{"source": "e1", "target": "e2", "type": "R1"}]
|
||||
}
|
||||
|
||||
# Test visualize_network
|
||||
viz.visualize_network(graph)
|
||||
|
||||
# Test visualize_communities
|
||||
communities = {"node_assignments": {"e1": 0, "e2": 1}, "num_communities": 2}
|
||||
viz.visualize_communities(graph, communities)
|
||||
|
||||
# Test visualize_centrality
|
||||
centrality = {"centrality": {"e1": 0.5, "e2": 0.3}}
|
||||
viz.visualize_centrality(graph, centrality)
|
||||
|
||||
# Test visualize_entity_types
|
||||
viz.visualize_entity_types(graph)
|
||||
|
||||
# Test visualize_relationship_matrix
|
||||
viz.visualize_relationship_matrix(graph)
|
||||
|
||||
# --- OntologyVisualizer Tests ---
|
||||
def test_ontology_visualizer(self):
|
||||
viz = OntologyVisualizer()
|
||||
ontology = {
|
||||
"classes": [
|
||||
{"name": "C1", "label": "Class 1", "parent": None},
|
||||
{"name": "C2", "label": "Class 2", "parent": "C1"}
|
||||
],
|
||||
"properties": [
|
||||
{"name": "P1", "label": "Prop 1", "domain": "C1", "range": "C2"}
|
||||
]
|
||||
}
|
||||
|
||||
# Test visualize_hierarchy
|
||||
viz.visualize_hierarchy(ontology)
|
||||
|
||||
# Test visualize_properties
|
||||
viz.visualize_properties(ontology)
|
||||
|
||||
# Test visualize_structure
|
||||
viz.visualize_structure(ontology)
|
||||
|
||||
# Test visualize_class_property_matrix
|
||||
viz.visualize_class_property_matrix(ontology)
|
||||
|
||||
# Test visualize_metrics
|
||||
viz.visualize_metrics(ontology)
|
||||
|
||||
# Test visualize_semantic_model (mocking extract classes)
|
||||
semantic_model = {"nodes": [{"id": "n1", "type": "T1"}], "edges": []}
|
||||
viz.visualize_semantic_model(semantic_model)
|
||||
|
||||
# --- SemanticNetworkVisualizer Tests ---
|
||||
def test_semantic_network_visualizer(self):
|
||||
viz = SemanticNetworkVisualizer()
|
||||
semantic_network = {
|
||||
"nodes": [{"id": "n1", "label": "N1", "type": "T1"}],
|
||||
"edges": [{"source": "n1", "target": "n1", "label": "R1"}]
|
||||
}
|
||||
|
||||
# Test visualize_network
|
||||
with patch('semantica.visualization.kg_visualizer.KGVisualizer') as MockKG:
|
||||
viz.visualize_network(semantic_network)
|
||||
MockKG.return_value.visualize_network.assert_called()
|
||||
|
||||
# Test visualize_node_types
|
||||
viz.visualize_node_types(semantic_network)
|
||||
|
||||
# Test visualize_edge_types
|
||||
viz.visualize_edge_types(semantic_network)
|
||||
|
||||
# --- AnalyticsVisualizer Tests ---
|
||||
def test_analytics_visualizer(self):
|
||||
viz = AnalyticsVisualizer()
|
||||
graph = {"entities": [], "relationships": []}
|
||||
|
||||
# Test visualize_centrality_rankings
|
||||
centrality = {"rankings": [{"node": "n1", "score": 0.9}]}
|
||||
viz.visualize_centrality_rankings(centrality)
|
||||
|
||||
# Test visualize_community_structure
|
||||
communities = {"node_assignments": {}}
|
||||
with patch('semantica.visualization.kg_visualizer.KGVisualizer') as MockKG:
|
||||
viz.visualize_community_structure(graph, communities)
|
||||
|
||||
# Test visualize_connectivity
|
||||
connectivity = {"is_connected": True, "num_components": 1, "component_sizes": [10]}
|
||||
viz.visualize_connectivity(connectivity)
|
||||
|
||||
# Test visualize_degree_distribution
|
||||
viz.visualize_degree_distribution(graph)
|
||||
|
||||
# Test visualize_metrics_dashboard
|
||||
metrics = {"num_nodes": 10, "num_edges": 20, "density": 0.1}
|
||||
viz.visualize_metrics_dashboard(metrics)
|
||||
|
||||
# Test visualize_centrality_comparison
|
||||
results = {"degree": {"rankings": [{"node": "n1", "score": 0.9}]}}
|
||||
viz.visualize_centrality_comparison(results)
|
||||
|
||||
# --- TemporalVisualizer Tests ---
|
||||
def test_temporal_visualizer(self):
|
||||
viz = TemporalVisualizer()
|
||||
|
||||
# Test visualize_timeline
|
||||
temporal_data = {"events": [{"timestamp": "2023-01-01", "type": "create", "label": "E1"}], "timestamps": ["2023-01-01"]}
|
||||
viz.visualize_timeline(temporal_data)
|
||||
|
||||
# Test visualize_temporal_patterns
|
||||
patterns = [{"pattern_type": "trend", "start_time": "2023", "end_time": "2024", "entities": ["e1"]}]
|
||||
viz.visualize_temporal_patterns(patterns)
|
||||
|
||||
# Test visualize_snapshot_comparison
|
||||
snapshots = {"2023": {"entities": ["e1"], "relationships": []}}
|
||||
viz.visualize_snapshot_comparison(snapshots)
|
||||
|
||||
# Test visualize_version_history
|
||||
history = [{"version": "v1", "date": "2023-01-01"}]
|
||||
viz.visualize_version_history(history)
|
||||
|
||||
# Test visualize_metrics_evolution
|
||||
metrics_history = {"nodes": [10, 20]}
|
||||
timestamps = ["2023", "2024"]
|
||||
viz.visualize_metrics_evolution(metrics_history, timestamps)
|
||||
|
||||
# --- EmbeddingVisualizer Tests ---
|
||||
def test_embedding_visualizer(self):
|
||||
viz = EmbeddingVisualizer()
|
||||
embeddings = np.random.rand(10, 10)
|
||||
|
||||
# Test visualize_2d_projection (mock UMAP/PCA)
|
||||
with patch('semantica.visualization.embedding_visualizer.umap.UMAP') as MockUMAP:
|
||||
MockUMAP.return_value.fit_transform.return_value = np.random.rand(10, 2)
|
||||
viz.visualize_2d_projection(embeddings)
|
||||
|
||||
# Test visualize_similarity_heatmap
|
||||
viz.visualize_similarity_heatmap(embeddings[:5]) # smaller for heatmap
|
||||
|
||||
# Test visualize_clustering
|
||||
clusters = [0, 1, 0, 1, 0, 1, 0, 1, 0, 1]
|
||||
with patch('semantica.visualization.embedding_visualizer.umap.UMAP') as MockUMAP:
|
||||
MockUMAP.return_value.fit_transform.return_value = np.random.rand(10, 2)
|
||||
viz.visualize_clustering(embeddings, clusters)
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user