Merge pull request #242 from RazinShaikh/multigraph-self-loop

Multigraph self loop
zxcalc · Jun 27, 2024 · 39a7203 · 39a7203
2 parents cd0269d + 61ac585
commit 39a7203
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyzx/graph/multigraph.py b/pyzx/graph/multigraph.py
@@ -209,7 +209,7 @@ def remove_vertices(self, vertices):
                 e = self.graph[v][v1]
                 self.nedges -= e.s + e.h
                 del self.graph[v][v1]
-                del self.graph[v1][v]
+                if v != v1: del self.graph[v1][v]
             # remove the vertex
             del self.graph[v]
             del self.ty[v]
@@ -244,7 +244,7 @@ def remove_edge(self, edge):
 
         if e.is_empty():
             del self.graph[s][t]
-            del self.graph[t][s]
+            if s != t: del self.graph[t][s]
 
         self.nedges -= 1
 
@@ -270,7 +270,7 @@ def edges(self, s=None, t=None):
         if s == None:
             for v0,adj in self.graph.items():
                 for v1, e in adj.items():
-                    if v1 > v0:
+                    if v1 >= v0:
                         for _ in range(e.s): yield (v0, v1, EdgeType.SIMPLE)
                         for _ in range(e.h): yield (v0, v1, EdgeType.HADAMARD)
                         for _ in range(e.w_io): yield (v0, v1, EdgeType.W_IO)