-
Notifications
You must be signed in to change notification settings - Fork 26
/
test_utils.py
70 lines (53 loc) · 1.98 KB
/
test_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import pytest
from xgi.utils import convert_labels_to_integers, get_dual, load_xgi_data
@pytest.mark.webtest
@pytest.mark.slow
def test_load_xgi_data():
H = load_xgi_data("email-enron")
assert H.num_nodes == 148
assert H.num_edges == 10885
assert H["name"] == "email-Enron"
assert H.nodes["4"]["name"] == "robert.badeer@enron.com"
assert H.edges["0"]["timestamp"] == "2000-01-11T10:29:00"
def test_get_dual(dict5):
dual = get_dual(dict5)
assert dual[0] == [0]
assert dual[1] == [0]
assert dual[2] == [0]
assert dual[3] == [0]
assert dual[4] == [1]
assert dual[5] == [2]
assert dual[6] == [2, 3]
assert dual[7] == [3]
assert dual[8] == [3]
def test_convert_labels_to_integers(hypergraph1, hypergraph2):
H1 = convert_labels_to_integers(hypergraph1)
H2 = convert_labels_to_integers(hypergraph2)
assert set(H1.nodes) == {0, 1, 2}
assert set(H1.edges) == {0, 1, 2}
assert H1.nodes[0]["label"] == "a"
assert H1.nodes[1]["label"] == "b"
assert H1.nodes[2]["label"] == "c"
assert H1.edges[0]["label"] == "e1"
assert H1.edges[1]["label"] == "e2"
assert H1.edges[2]["label"] == "e3"
assert H1.edges.members(0) == [0, 1]
assert H1.edges.members(1) == [0, 1, 2]
assert H1.edges.members(2) == [2]
assert H1.nodes.memberships(0) == [0, 1]
assert H1.nodes.memberships(1) == [0, 1]
assert H1.nodes.memberships(2) == [1, 2]
assert set(H2.nodes) == {0, 1, 2}
assert set(H2.edges) == {0, 1, 2}
assert H2.nodes[0]["label"] == 0
assert H2.nodes[1]["label"] == "b"
assert H2.nodes[2]["label"] == "c"
assert H2.edges[0]["label"] == "e1"
assert H2.edges[1]["label"] == "e2"
assert H2.edges[2]["label"] == "e3"
assert H2.edges.members(0) == [0, 1]
assert H2.edges.members(1) == [0, 2]
assert H2.edges.members(2) == [0, 1, 2]
assert H2.nodes.memberships(0) == [0, 1, 2]
assert H2.nodes.memberships(1) == [0, 2]
assert H2.nodes.memberships(2) == [1, 2]