[pypy-svn] r74119 - pypy/branch/blackhole-improvement/pypy/tool/algo

[arigo at codespeak.net]

Author: arigo
Date: Tue Apr 27 19:25:07 2010
New Revision: 74119

Modified:
   pypy/branch/blackhole-improvement/pypy/tool/algo/color.py
Log:
Accept duplicate add_node()s.


Modified: pypy/branch/blackhole-improvement/pypy/tool/algo/color.py
==============================================================================
--- pypy/branch/blackhole-improvement/pypy/tool/algo/color.py	(original)
+++ pypy/branch/blackhole-improvement/pypy/tool/algo/color.py	Tue Apr 27 19:25:07 2010
@@ -6,9 +6,9 @@
         self.neighbours = {}
 
     def add_node(self, v):
-        assert v not in self.neighbours, "duplicate vertex %r" % (v,)
-        self._all_nodes.append(v)
-        self.neighbours[v] = set()
+        if v not in self.neighbours:
+            self._all_nodes.append(v)
+            self.neighbours[v] = set()
 
     def add_edge(self, v1, v2):
         assert v1 != v2