wehub-resource-sync
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import unittest
|
|
|
|
import easygraph as eg
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import statsmodels.api as sm
|
|
|
|
|
|
class TestPlot(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_plot_Followers(self):
|
|
eg.functions.plot_Followers(self.ds, self.shs)
|
|
|
|
def test_plot_Connected_Communities(self):
|
|
eg.functions.plot_Connected_Communities(self.ds, self.shs)
|
|
|
|
def test_plot_Neighborhood_Followers(self):
|
|
eg.functions.plot_Neighborhood_Followers(self.ds, self.shs)
|
|
|
|
def test_plot_Betweenness_Centrality(self):
|
|
eg.functions.plot_Betweenness_Centrality(self.ds, self.shs)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|