wehub-resource-sync
25 lines
684 B
Python
25 lines
684 B
Python
import easygraph as eg
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture()
|
|
def g1():
|
|
e_list = [(0, 1, 2, 5), (0, 1), (2, 3, 4), (1, 2, 4)]
|
|
g = eg.Hypergraph(6, e_list=e_list)
|
|
return g
|
|
|
|
|
|
@pytest.fixture()
|
|
def g2():
|
|
e_list = [(1, 2, 3), (0, 1, 3), (0, 1), (2, 4, 3), (2, 3)]
|
|
e_weight = [0.5, 1, 0.5, 1, 0.5]
|
|
g = eg.Hypergraph(5, e_list=e_list, e_weight=e_weight)
|
|
return g
|
|
|
|
|
|
def test_degree_centrality(g1, g2):
|
|
print(eg.hyepergraph_degree_centrality(g1))
|
|
print(eg.hyepergraph_degree_centrality(g2))
|
|
assert eg.hyepergraph_degree_centrality(g1) == {0: 2, 1: 3, 2: 3, 3: 1, 4: 2, 5: 1}
|
|
assert eg.hyepergraph_degree_centrality(g2) == {0: 2, 1: 3, 2: 3, 3: 4, 4: 1}
|