wehub-resource-sync
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
import unittest
|
|
|
|
import easygraph as eg
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
|
|
|
|
class TestPositioning(unittest.TestCase):
|
|
def setUp(self):
|
|
self.ds = eg.datasets.get_graph_karateclub()
|
|
self.edges = [
|
|
(1, 4),
|
|
(2, 4),
|
|
("String", "Bool"),
|
|
(4, 1),
|
|
(0, 4),
|
|
(4, 256),
|
|
((1, 2), (3, 4)),
|
|
]
|
|
self.test_graphs = [eg.Graph(), eg.DiGraph()]
|
|
self.test_graphs.append(eg.classes.DiGraph(self.edges))
|
|
self.shs = eg.common_greedy(self.ds, int(len(self.ds.nodes) / 3))
|
|
|
|
def test_random_position(self):
|
|
print()
|
|
for i in self.test_graphs:
|
|
print(eg.random_position(i))
|
|
|
|
def test_circular_position(self):
|
|
print()
|
|
for i in self.test_graphs:
|
|
print(eg.circular_position(i))
|
|
|
|
def test_shell_position(self):
|
|
print()
|
|
for i in self.test_graphs:
|
|
print(eg.shell_position(i))
|
|
|
|
def test_rescale_position(self):
|
|
print()
|
|
for i in self.test_graphs:
|
|
try:
|
|
pos = eg.random_position(i)
|
|
obj = np.array(list(pos.values()))
|
|
print(eg.rescale_position(obj))
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
def test_kamada_kawai_layout(self):
|
|
print()
|
|
for i in self.test_graphs:
|
|
print(eg.kamada_kawai_layout(i))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|