added check for equality on edge

MaineKuehn · Jul 26, 2018 · d12edb8 · d12edb8
1 parent 83b22e8
commit d12edb8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/graphi/edge.py b/graphi/edge.py
@@ -84,6 +84,11 @@ def __iter__(self):
     def __str__(self):
         return '[%s:%s]' % (self.start, self.stop)
 
+    def __eq__(self, other):
+        if not isinstance(self, type(other)):
+            return NotImplemented
+        return self.start == other.start and self.stop == other.stop
+
     def __repr__(self):
         return '%s[%r:%r]' % (type(self).__name__, self.start, self.stop)