76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
from collections import defaultdict
|
|
|
|
import pytest
|
|
numpy = pytest.importorskip('numpy')
|
|
scipy = pytest.importorskip('scipy')
|
|
|
|
import networkx as nx
|
|
from networkx.testing import almost_equal
|
|
from networkx.algorithms.communicability_alg import *
|
|
|
|
|
|
class TestCommunicability:
|
|
|
|
def test_communicability(self):
|
|
answer = {0: {0: 1.5430806348152435,
|
|
1: 1.1752011936438012
|
|
},
|
|
1: {0: 1.1752011936438012,
|
|
1: 1.5430806348152435
|
|
}
|
|
}
|
|
# answer={(0, 0): 1.5430806348152435,
|
|
# (0, 1): 1.1752011936438012,
|
|
# (1, 0): 1.1752011936438012,
|
|
# (1, 1): 1.5430806348152435}
|
|
|
|
result = communicability(nx.path_graph(2))
|
|
for k1, val in result.items():
|
|
for k2 in val:
|
|
assert almost_equal(answer[k1][k2], result[k1][k2], places=7)
|
|
|
|
def test_communicability2(self):
|
|
|
|
answer_orig = {('1', '1'): 1.6445956054135658,
|
|
('1', 'Albert'): 0.7430186221096251,
|
|
('1', 'Aric'): 0.7430186221096251,
|
|
('1', 'Dan'): 1.6208126320442937,
|
|
('1', 'Franck'): 0.42639707170035257,
|
|
('Albert', '1'): 0.7430186221096251,
|
|
('Albert', 'Albert'): 2.4368257358712189,
|
|
('Albert', 'Aric'): 1.4368257358712191,
|
|
('Albert', 'Dan'): 2.0472097037446453,
|
|
('Albert', 'Franck'): 1.8340111678944691,
|
|
('Aric', '1'): 0.7430186221096251,
|
|
('Aric', 'Albert'): 1.4368257358712191,
|
|
('Aric', 'Aric'): 2.4368257358712193,
|
|
('Aric', 'Dan'): 2.0472097037446457,
|
|
('Aric', 'Franck'): 1.8340111678944691,
|
|
('Dan', '1'): 1.6208126320442937,
|
|
('Dan', 'Albert'): 2.0472097037446453,
|
|
('Dan', 'Aric'): 2.0472097037446457,
|
|
('Dan', 'Dan'): 3.1306328496328168,
|
|
('Dan', 'Franck'): 1.4860372442192515,
|
|
('Franck', '1'): 0.42639707170035257,
|
|
('Franck', 'Albert'): 1.8340111678944691,
|
|
('Franck', 'Aric'): 1.8340111678944691,
|
|
('Franck', 'Dan'): 1.4860372442192515,
|
|
('Franck', 'Franck'): 2.3876142275231915}
|
|
|
|
answer = defaultdict(dict)
|
|
for (k1, k2), v in answer_orig.items():
|
|
answer[k1][k2] = v
|
|
|
|
G1 = nx.Graph([('Franck', 'Aric'), ('Aric', 'Dan'), ('Dan', 'Albert'),
|
|
('Albert', 'Franck'), ('Dan', '1'), ('Franck', 'Albert')])
|
|
|
|
result = communicability(G1)
|
|
for k1, val in result.items():
|
|
for k2 in val:
|
|
assert almost_equal(answer[k1][k2], result[k1][k2], places=7)
|
|
|
|
result = communicability_exp(G1)
|
|
for k1, val in result.items():
|
|
for k2 in val:
|
|
assert almost_equal(answer[k1][k2], result[k1][k2], places=7)
|