Fix bug in copy(pg::PartitionedGraph) (#50)

ITensor · Feb 15, 2024 · dd9c5cd · dd9c5cd
1 parent 08df103
commit dd9c5cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/Graphs/partitionedgraphs/partitionedgraph.jl b/src/Graphs/partitionedgraphs/partitionedgraph.jl
@@ -99,7 +99,7 @@ function copy(pg::PartitionedGraph)
     copy(unpartitioned_graph(pg)),
     copy(partitioned_graph(pg)),
     copy(partitioned_vertices(pg)),
-    copy(partitionvertex(pg)),
+    copy(which_partition(pg)),
   )
 end
 

diff --git a/test/test_partitionedgraph.jl b/test/test_partitionedgraph.jl
@@ -27,6 +27,8 @@ using Graphs
   @test is_tree(partitioned_graph(pg))
   @test nv(pg) == nx * ny
   @test nv(partitioned_graph(pg)) == nx
+  pg_c = copy(pg)
+  @test pg_c == pg
 
   #Same partitioning but with a dictionary constructor
   partition_dict = Dictionary([first(partition) for partition in partitions], partitions)