scikit-network 0.33.3__cp312-cp312-macosx_10_13_x86_64.whl

sknetwork/visualization/tests/__init__.py

@@ -0,0 +1 @@
+"""tests for visualization"""

sknetwork/visualization/tests/test_dendrograms.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""tests for visualization of dendrograms"""
+
+import tempfile
+import unittest
+
+import numpy as np
+
+from sknetwork.data.toy_graphs import karate_club, painters
+from sknetwork.hierarchy import Paris
+from sknetwork.visualization.dendrograms import visualize_dendrogram, svg_dendrogram, svg_dendrogram_top
+
+
+# noinspection DuplicatedCode
+class TestVisualization(unittest.TestCase):
+
+    def test_undirected(self):
+        adjacency = karate_club()
+        paris = Paris()
+        dendrogram = paris.fit_transform(adjacency)
+        image = svg_dendrogram(dendrogram)
+        self.assertEqual(image[1:4], 'svg')
+        n = adjacency.shape[0]
+        image = visualize_dendrogram(dendrogram, names=np.arange(n), width=200, height=200, margin=10, margin_text=5,
+                                     scale=3, n_clusters=3, color='green', colors=['red', 'blue'], font_size=14,
+                                     reorder=True, rotate=True)
+        self.assertEqual(image[1:4], 'svg')
+        image = svg_dendrogram(dendrogram, names=np.arange(n), width=200, height=200, margin=10, margin_text=5, scale=3,
+                               n_clusters=3, color='green', colors={0: 'red', 1: 'blue'}, font_size=14, reorder=False,
+                               rotate=True)
+        self.assertEqual(image[1:4], 'svg')
+        svg_dendrogram_top(dendrogram, names=np.arange(n), width=200, height=200, margin=10, margin_text=5, scale=3,
+                           n_clusters=3, color='green', colors=np.array(['red', 'black', 'blue']), font_size=14,
+                           reorder=False, rotate_names=True, line_width=0.1)
+
+    def test_directed(self):
+        graph = painters(True)
+        adjacency = graph.adjacency
+        names = graph.names
+        paris = Paris()
+        dendrogram = paris.fit_transform(adjacency)
+        image = visualize_dendrogram(dendrogram)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_dendrogram(dendrogram, names=names, width=200, height=200, margin=10, margin_text=5, scale=3,
+                                     n_clusters=3, color='green', font_size=14, reorder=True, rotate=True)
+        self.assertEqual(image[1:4], 'svg')
+
+        filename = tempfile.gettempdir() + '/image'
+        _ = visualize_dendrogram(dendrogram, filename=filename)
+        with open(filename + '.svg', 'r') as f:
+            row = f.readline()
+            self.assertEqual(row[1:4], 'svg')

sknetwork/visualization/tests/test_graphs.py

@@ -0,0 +1,176 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""tests for visualization of graphs"""
+
+import tempfile
+import unittest
+
+import numpy as np
+from scipy import sparse
+
+from sknetwork.data.test_graphs import test_disconnected_graph, test_bigraph_disconnect
+from sknetwork.data.toy_graphs import karate_club, painters, movie_actor, bow_tie, star_wars
+from sknetwork.visualization.graphs import visualize_graph, visualize_bigraph, svg_graph, svg_bigraph, svg_text, rescale
+
+
+# noinspection DuplicatedCode
+class TestVisualization(unittest.TestCase):
+
+    def test_undirected(self):
+        graph = karate_club(True)
+        adjacency = graph.adjacency
+        position = graph.position
+        labels = graph.labels
+        image = visualize_graph(adjacency, position, labels=labels)
+        self.assertEqual(image[1:4], 'svg')
+        # alias
+        image = svg_graph(adjacency, position, labels=labels)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, labels=list(labels))
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, display_edges=False)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, height=None)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, height=300, width=None)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, height=None, width=200)
+        self.assertEqual(image[1:4], 'svg')
+        n = adjacency.shape[0]
+        edge_labels = [(0, 1, 0), (1, 1, 1), (3, 10, 2)]
+        image = visualize_graph(adjacency, position=None, names=np.arange(n), labels=np.arange(n), scores=np.arange(n),
+                                seeds=[0, 1], width=200, height=200, margin=10, margin_text=5, scale=3,
+                                node_order=np.flip(np.arange(n)),
+                                node_size=5, node_size_min=2, node_size_max=6, display_node_weight=True,
+                                node_weights=np.arange(n),
+                                node_width=2, node_width_max=5, node_color='red', edge_width=2, edge_width_min=2,
+                                edge_width_max=4, edge_color='blue', edge_labels=edge_labels, display_edge_weight=True,
+                                font_size=14)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position=None, labels={0: 0})
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position=None, scores={0: 0})
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency=None, position=position)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency=None, position=position, edge_labels=edge_labels)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, labels, label_colors={0: "red", 1: "blue"})
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, labels, label_colors=["red", "blue"])
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, labels, node_weights=np.arange(adjacency.shape[0]))
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, scores=list(np.arange(n)))
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, seeds={0: 1, 2: 1})
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, labels=np.arange(n), name_position='left')
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, scale=2, labels=np.arange(n), name_position='left')
+        self.assertEqual(image[1:4], 'svg')
+        with self.assertRaises(ValueError):
+            visualize_graph(adjacency, position, labels=[0, 1])
+        with self.assertRaises(ValueError):
+            visualize_graph(adjacency, position, scores=[0, 1])
+        visualize_graph(adjacency, position, scale=2, labels=np.arange(n), name_position='left')
+
+    def test_directed(self):
+        graph = painters(True)
+        adjacency = graph.adjacency
+        position = graph.position
+        names = graph.names
+        image = visualize_graph(adjacency, position, names=names)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_graph(adjacency, position, display_edges=False)
+        self.assertEqual(image[1:4], 'svg')
+        n = adjacency.shape[0]
+        image = visualize_graph(adjacency, position=None, names=np.arange(n), labels=np.arange(n), scores=np.arange(n),
+                                name_position='below', seeds=[0, 1], width=200, height=200, margin=10, margin_text=5,
+                                scale=3, node_order=np.flip(np.arange(n)),
+                                node_size=5, node_size_min=2, node_size_max=6, display_node_weight=True,
+                                node_weights=np.arange(n),
+                                node_width=2, node_width_max=5, node_color='red', edge_width=2, edge_width_min=2,
+                                edge_width_max=4, edge_color='blue', display_edge_weight=True, font_size=14)
+        self.assertEqual(image[1:4], 'svg')
+
+    def test_bipartite(self):
+        graph = movie_actor(True)
+        biadjacency = graph.biadjacency
+        names_row = graph.names_row
+        names_col = graph.names_col
+        image = visualize_bigraph(biadjacency, names_row, names_col)
+        self.assertEqual(image[1:4], 'svg')
+        # alias
+        image = svg_bigraph(biadjacency, names_row, names_col)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_bigraph(biadjacency, display_edges=False)
+        self.assertEqual(image[1:4], 'svg')
+        image = visualize_bigraph(biadjacency, reorder=False)
+        self.assertEqual(image[1:4], 'svg')
+        n_row, n_col = biadjacency.shape
+        position_row = np.random.random((n_row, 2))
+        position_col = np.random.random((n_col, 2))
+        edge_labels = [(0, 1, 0), (1, 1, 1), (3, 10, 2)]
+        image = visualize_bigraph(biadjacency=biadjacency, names_row=np.arange(n_row), names_col=np.arange(n_col),
+                                  labels_row=np.arange(n_row), labels_col=np.arange(n_col), scores_row=np.arange(n_row),
+                                  scores_col=np.arange(n_col), seeds_row=[0, 1], seeds_col=[1, 2],
+                                  position_row=position_row, position_col=position_col, color_row='red', color_col='white',
+                                  width=200, height=200, margin=10, margin_text=5, scale=3, node_size=5,
+                                  node_size_min=1, node_size_max=30, node_weights_row=np.arange(n_row),
+                                  node_weights_col=np.arange(n_col), display_node_weight=True, node_width=2, node_width_max=5,
+                                  edge_labels=edge_labels, edge_width=2, edge_width_min=0.3, edge_width_max=4,
+                                  edge_color='red', display_edge_weight=True, font_size=14)
+        self.assertEqual(image[1:4], 'svg')
+
+    def test_disconnect(self):
+        adjacency = test_disconnected_graph()
+        position = np.random.random((adjacency.shape[0], 2))
+        image = visualize_graph(adjacency, position)
+        self.assertEqual(image[1:4], 'svg')
+        biadjacency = test_bigraph_disconnect()
+        image = visualize_bigraph(biadjacency)
+        self.assertEqual(image[1:4], 'svg')
+
+    def test_probs(self):
+        adjacency = bow_tie()
+        probs = np.array([[.5, .5], [0, 0], [1, 0], [0, 1], [0, 1]])
+        image = visualize_graph(adjacency, probs=probs)
+        self.assertEqual(image[1:4], 'svg')
+        probs = sparse.csr_matrix(probs)
+        image = visualize_graph(adjacency, probs=probs)
+        self.assertEqual(image[1:4], 'svg')
+        biadjacency = star_wars()
+        probs_row = sparse.csr_matrix([[.5, .5], [0, 0], [1, 0], [0, 1]])
+        probs_col = sparse.csr_matrix([[.5, .5], [0, 0], [1, 0]])
+        image = visualize_bigraph(biadjacency, probs_row=probs_row, probs_col=probs_col)
+        self.assertEqual(image[1:4], 'svg')
+
+    def test_labels(self):
+        adjacency = bow_tie()
+        names = ["aa", "bb", "<>", "a&b", ""]
+        image = visualize_graph(adjacency, names=names)
+        self.assertEqual(image[1:4], 'svg')
+
+    def test_text(self):
+        image = svg_text(np.array([0, 0]), 'foo', 0.1, 16, 'above')
+        self.assertEqual(image[1:5], 'text')
+
+    def test_rescale(self):
+        output = rescale(np.array([[0, 0]]), width=4, height=6, margin=2, node_size=10, node_size_max=20,
+                         display_node_weight=True, names=np.array(['foo']), name_position='left')
+        self.assertEqual(len(output), 3)
+
+    def test_write(self):
+        filename = tempfile.gettempdir() + '/image'
+        graph = karate_club(True)
+        adjacency = graph.adjacency
+        position = graph.position
+        _ = visualize_graph(adjacency, position, filename=filename)
+        with open(filename + '.svg', 'r') as f:
+            row = f.readline()
+            self.assertEqual(row[1:4], 'svg')
+        _ = visualize_bigraph(adjacency, position, filename=filename)
+        with open(filename + '.svg', 'r') as f:
+            row = f.readline()
+            self.assertEqual(row[1:4], 'svg')