chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user