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
|
# -*- coding: utf-8 -*-
import json
from nose.tools import assert_equal, assert_raises, assert_not_equal, assert_true, raises
import networkx as nx
from networkx.readwrite.json_graph import *
class TestNodeLink:
def test_graph(self):
G = nx.path_graph(4)
H = node_link_graph(node_link_data(G))
nx.is_isomorphic(G,H)
def test_graph_attributes(self):
G = nx.path_graph(4)
G.add_node(1,color='red')
G.add_edge(1,2,width=7)
G.graph[1]='one'
G.graph['foo']='bar'
H = node_link_graph(node_link_data(G))
assert_equal(H.graph['foo'],'bar')
assert_equal(H.node[1]['color'],'red')
assert_equal(H[1][2]['width'],7)
d = json.dumps(node_link_data(G))
H = node_link_graph(json.loads(d))
assert_equal(H.graph['foo'],'bar')
assert_equal(H.graph['1'],'one')
assert_equal(H.node[1]['color'],'red')
assert_equal(H[1][2]['width'],7)
def test_digraph(self):
G = nx.DiGraph()
H = node_link_graph(node_link_data(G))
assert_true(H.is_directed())
def test_multigraph(self):
G = nx.MultiGraph()
G.add_edge(1,2,key='first')
G.add_edge(1,2,key='second',color='blue')
H = node_link_graph(node_link_data(G))
nx.is_isomorphic(G,H)
assert_equal(H[1][2]['second']['color'],'blue')
def test_unicode_keys(self):
try:
q = unicode("qualité",'utf-8')
except NameError:
q = "qualité"
G = nx.Graph()
G.add_node(1, {q:q})
s = node_link_data(G)
output = json.dumps(s, ensure_ascii=False)
data = json.loads(output)
H = node_link_graph(data)
assert_equal(H.node[1][q], q)
@raises(nx.NetworkXError)
def test_exception(self):
G = nx.MultiDiGraph()
attrs = dict(id='id', source='node', target='node', key='node')
node_link_data(G, attrs)
|
