summaryrefslogtreecommitdiff
path: root/networkx/algorithms/approximation/tests/test_approx_clust_coeff.py
blob: 5eab5c1ee79408c9f90a1993415a6c3d7d957141 (plain)
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
import networkx as nx
from networkx.algorithms.approximation import average_clustering

# This approximation has to be exact in regular graphs
# with no triangles or with all possible triangles.


def test_petersen():
    # Actual coefficient is 0
    G = nx.petersen_graph()
    assert average_clustering(G, trials=len(G) // 2) == nx.average_clustering(G)


def test_petersen_seed():
    # Actual coefficient is 0
    G = nx.petersen_graph()
    assert average_clustering(G, trials=len(G) // 2, seed=1) == nx.average_clustering(G)


def test_tetrahedral():
    # Actual coefficient is 1
    G = nx.tetrahedral_graph()
    assert average_clustering(G, trials=len(G) // 2) == nx.average_clustering(G)


def test_dodecahedral():
    # Actual coefficient is 0
    G = nx.dodecahedral_graph()
    assert average_clustering(G, trials=len(G) // 2) == nx.average_clustering(G)


def test_empty():
    G = nx.empty_graph(5)
    assert average_clustering(G, trials=len(G) // 2) == 0


def test_complete():
    G = nx.complete_graph(5)
    assert average_clustering(G, trials=len(G) // 2) == 1
    G = nx.complete_graph(7)
    assert average_clustering(G, trials=len(G) // 2) == 1