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
|
#!/usr/bin/env python
from nose.tools import *
import networkx as nx
class TestVitality:
def test_closeness_vitality_unweighted(self):
G=nx.cycle_graph(3)
v=nx.closeness_vitality(G)
assert_equal(v,{0:4.0, 1:4.0, 2:4.0})
def test_closeness_vitality_weighted(self):
G=nx.Graph()
G.add_cycle([0,1,2],weight=2)
v=nx.closeness_vitality(G,weight='weight')
assert_equal(v,{0:8.0, 1:8.0, 2:8.0})
def test_closeness_vitality_unweighted_digraph(self):
G=nx.DiGraph()
G.add_cycle([0,1,2])
v=nx.closeness_vitality(G)
assert_equal(v,{0:8.0, 1:8.0, 2:8.0})
def test_closeness_vitality_weighted_digraph(self):
G=nx.DiGraph()
G.add_cycle([0,1,2],weight=2)
v=nx.closeness_vitality(G,weight='weight')
assert_equal(v,{0:16.0, 1:16.0, 2:16.0})
|
