diff --git a/core.py b/core.py index eaf7986..0a960f0 100644 --- a/core.py +++ b/core.py @@ -25,7 +25,7 @@ def __eq__(self, other): def getEdgeAdjacentToVertex(self,v): for edge in self.edges: - if edge.v2==v or edge.v1==v: + if edge.v2 is v or edge.v1 is v: return edge return None