Update jit_graph and tests

1 files changed, 18 insertions, 6 deletions

diff --git a/networkx/readwrite/json_graph/tests/test_jit.py b/networkx/readwrite/json_graph/tests/test_jit.py
index 3e4482b8..00397fcc 100644
--- a/networkx/readwrite/json_graph/tests/test_jit.py
+++ b/networkx/readwrite/json_graph/tests/test_jit.py
@@ -3,15 +3,27 @@ from nose.tools import assert_true
 import networkx as nx
 from networkx.readwrite.json_graph import jit_data, jit_graph
 
-class TestJIT:
 
-    def test_jit_graph(self):
+class TestJIT(object):
+    def test_jit(self):
         G = nx.Graph()
-        G.add_node('Node1', node_data = 'foobar')
-        G.add_node('Node3', node_data = 'bar')
-        G.add_edge('Node1', 'Node2', weight = 9, something='isSomething')
-        G.add_edge('Node2', 'Node3', weight = 4, something='isNotSomething')
+        G.add_node('Node1', node_data='foobar')
+        G.add_node('Node3', node_data='bar')
+        G.add_node('Node4')
+        G.add_edge('Node1', 'Node2', weight=9, something='isSomething')
+        G.add_edge('Node2', 'Node3', weight=4, something='isNotSomething')
         G.add_edge('Node1', 'Node2')
         d = jit_data(G)
         K = jit_graph(json.loads(d))
         assert_true(nx.is_isomorphic(G, K))
+
+    def test_jit_2(self):
+        G = nx.Graph()
+        G.add_node(1, node_data=3)
+        G.add_node(3, node_data=0)
+        G.add_edge(1, 2, weight=9, something=0)
+        G.add_edge(2, 3, weight=4, something=3)
+        G.add_edge(1, 2)
+        d = jit_data(G)
+        K = jit_graph(json.loads(d))
+        assert_true(nx.is_isomorphic(G, K))