Merge pull request #2 from dkorduban/add_edge_constant_time

Make graph.add_edge run in O(1) instead of O(degree(V))

Author: agroszer

core/pygraph/classes/graph.py

@@ -78,7 +78,7 @@ def neighbors(self, node):
         @rtype:  list
         @return: List of nodes directly accessible from given node.
         """
-        return self.node_neighbors[node]
+        return list(self.node_neighbors[node])
 
     def edges(self):
         """
@@ -118,7 +118,7 @@ def add_node(self, node, attrs=None):
         if attrs is None:
             attrs = []
         if (not node in self.node_neighbors):
-            self.node_neighbors[node] = []
+            self.node_neighbors[node] = set()
             self.node_attr[node] = attrs
         else:
             raise AdditionError("Node %s already in graph" % node)
@@ -143,9 +143,9 @@ def add_edge(self, edge, wt=1, label='', attrs=[]):
         """
         u, v = edge
         if (v not in self.node_neighbors[u] and u not in self.node_neighbors[v]):
-            self.node_neighbors[u].append(v)
+            self.node_neighbors[u].add(v)
             if (u != v):
-                self.node_neighbors[v].append(u)
+                self.node_neighbors[v].add(u)
 
             self.add_edge_attributes((u,v), attrs)        
             self.set_edge_properties((u, v), label=label, weight=wt)

tests/unittests-graph.py

@@ -70,7 +70,7 @@ def test_raise_exception_when_edge_added_from_non_existing_node(self):
             pass
         else:
             fail()
-        assert gr.node_neighbors == {0: [], 1: []}
+        assert gr.node_neighbors == {0: set(), 1: set()}
 
     def test_raise_exception_when_edge_added_to_non_existing_node(self):
         gr = graph()
@@ -81,7 +81,7 @@ def test_raise_exception_when_edge_added_to_non_existing_node(self):
             pass
         else:
             fail()
-        assert gr.node_neighbors == {0: [], 1: []}
+        assert gr.node_neighbors == {0: set(), 1: set()}
 
     def test_remove_node(self):
         gr = testlib.new_graph()