Mercurial > repos > shellac > sam_consensus_v3
comparison env/lib/python3.9/site-packages/networkx/algorithms/tests/test_smallworld.py @ 0:4f3585e2f14b draft default tip
"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author | shellac |
---|---|
date | Mon, 22 Mar 2021 18:12:50 +0000 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:4f3585e2f14b |
---|---|
1 import pytest | |
2 | |
3 numpy = pytest.importorskip("numpy") | |
4 | |
5 import random | |
6 | |
7 from networkx import random_reference, lattice_reference, sigma, omega | |
8 import networkx as nx | |
9 | |
10 rng = random.Random(0) | |
11 rng = 42 | |
12 | |
13 | |
14 def test_random_reference(): | |
15 G = nx.connected_watts_strogatz_graph(50, 6, 0.1, seed=rng) | |
16 Gr = random_reference(G, niter=1, seed=rng) | |
17 C = nx.average_clustering(G) | |
18 Cr = nx.average_clustering(Gr) | |
19 assert C > Cr | |
20 | |
21 pytest.raises(nx.NetworkXError, random_reference, nx.Graph()) | |
22 pytest.raises(nx.NetworkXNotImplemented, random_reference, nx.DiGraph()) | |
23 | |
24 H = nx.Graph(((0, 1), (2, 3))) | |
25 Hl = random_reference(H, niter=1, seed=rng) | |
26 | |
27 | |
28 def test_lattice_reference(): | |
29 G = nx.connected_watts_strogatz_graph(50, 6, 1, seed=rng) | |
30 Gl = lattice_reference(G, niter=1, seed=rng) | |
31 L = nx.average_shortest_path_length(G) | |
32 Ll = nx.average_shortest_path_length(Gl) | |
33 assert Ll > L | |
34 | |
35 pytest.raises(nx.NetworkXError, lattice_reference, nx.Graph()) | |
36 pytest.raises(nx.NetworkXNotImplemented, lattice_reference, nx.DiGraph()) | |
37 | |
38 H = nx.Graph(((0, 1), (2, 3))) | |
39 Hl = lattice_reference(H, niter=1) | |
40 | |
41 | |
42 def test_sigma(): | |
43 Gs = nx.connected_watts_strogatz_graph(50, 6, 0.1, seed=rng) | |
44 Gr = nx.connected_watts_strogatz_graph(50, 6, 1, seed=rng) | |
45 sigmas = sigma(Gs, niter=1, nrand=2, seed=rng) | |
46 sigmar = sigma(Gr, niter=1, nrand=2, seed=rng) | |
47 assert sigmar < sigmas | |
48 | |
49 | |
50 def test_omega(): | |
51 Gl = nx.connected_watts_strogatz_graph(50, 6, 0, seed=rng) | |
52 Gr = nx.connected_watts_strogatz_graph(50, 6, 1, seed=rng) | |
53 Gs = nx.connected_watts_strogatz_graph(50, 6, 0.1, seed=rng) | |
54 omegal = omega(Gl, niter=1, nrand=1, seed=rng) | |
55 omegar = omega(Gr, niter=1, nrand=1, seed=rng) | |
56 omegas = omega(Gs, niter=1, nrand=1, seed=rng) | |
57 print("omegas, omegal, omegar") | |
58 print(omegas, omegal, omegar) | |
59 assert omegal < omegas and omegas < omegar |