wehub-resource-sync
106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
import unittest
|
|
|
|
import easygraph as eg
|
|
|
|
from easygraph.functions.structural_holes.metrics import nodes_of_max_cc_without_shs
|
|
from easygraph.functions.structural_holes.metrics import structural_hole_influence_index
|
|
from easygraph.functions.structural_holes.metrics import sum_of_shortest_paths
|
|
|
|
|
|
class TestStructuralHoleMetrics(unittest.TestCase):
|
|
def setUp(self):
|
|
self.G = eg.datasets.get_graph_karateclub()
|
|
self.shs = [3, 9, 20]
|
|
self.communities = [
|
|
[1, 2, 3, 4, 5, 6, 7, 8, 11, 12, 13, 14, 17, 18, 20, 22],
|
|
[9, 10, 15, 16, 19, 21, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34],
|
|
]
|
|
|
|
def test_sum_of_shortest_paths_valid(self):
|
|
result = sum_of_shortest_paths(self.G, self.shs)
|
|
self.assertIsInstance(result, (int, float))
|
|
|
|
def test_sum_of_shortest_paths_empty(self):
|
|
result = sum_of_shortest_paths(self.G, [])
|
|
self.assertEqual(result, 0)
|
|
|
|
def test_nodes_of_max_cc_without_shs(self):
|
|
result = nodes_of_max_cc_without_shs(self.G, self.shs)
|
|
self.assertIsInstance(result, int)
|
|
self.assertLessEqual(result, self.G.number_of_nodes())
|
|
|
|
def test_nodes_of_max_cc_without_all_nodes(self):
|
|
result = nodes_of_max_cc_without_shs(self.G, list(self.G.nodes))
|
|
self.assertEqual(result, 0)
|
|
|
|
def test_structural_hole_influence_index_IC(self):
|
|
result = structural_hole_influence_index(
|
|
self.G,
|
|
self.shs,
|
|
self.communities,
|
|
model="IC",
|
|
Directed=False,
|
|
seedRatio=0.1,
|
|
randSeedIter=2,
|
|
countIterations=5,
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
self.assertTrue(all(isinstance(k, int) for k in result))
|
|
self.assertTrue(all(isinstance(v, float) for v in result.values()))
|
|
|
|
def test_structural_hole_influence_index_LT(self):
|
|
result = structural_hole_influence_index(
|
|
self.G,
|
|
self.shs,
|
|
self.communities,
|
|
model="LT",
|
|
Directed=False,
|
|
seedRatio=0.1,
|
|
randSeedIter=2,
|
|
countIterations=5,
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
|
|
def test_structural_hole_influence_index_variant_LT(self):
|
|
result = structural_hole_influence_index(
|
|
self.G,
|
|
self.shs,
|
|
self.communities,
|
|
model="LT",
|
|
variant=True,
|
|
Directed=False,
|
|
seedRatio=0.1,
|
|
randSeedIter=2,
|
|
countIterations=5,
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
|
|
def test_structural_hole_influence_index_empty_shs(self):
|
|
result = structural_hole_influence_index(
|
|
self.G, [], self.communities, model="IC", Directed=False
|
|
)
|
|
self.assertEqual(result, {})
|
|
|
|
def test_structural_hole_influence_index_directed_flag(self):
|
|
result = structural_hole_influence_index(
|
|
self.G,
|
|
self.shs,
|
|
self.communities,
|
|
model="IC",
|
|
Directed=True,
|
|
seedRatio=0.1,
|
|
randSeedIter=2,
|
|
countIterations=5,
|
|
)
|
|
self.assertIsInstance(result, dict)
|
|
|
|
def test_structural_hole_influence_index_no_shs_in_any_community(self):
|
|
result = structural_hole_influence_index(
|
|
self.G, [34], self.communities, model="LT", Directed=False
|
|
)
|
|
self.assertIn(34, result)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|