copy

Author: CarloLucibello

src/GNNGraphs/gnngraph.jl

@@ -200,6 +200,27 @@ function GNNGraph(g::GNNGraph; ndata=g.ndata, edata=g.edata, gdata=g.gdata, grap
             ndata, edata, gdata) 
 end
 
+
+"""
+    copy(g::GNNGraph; deep=false)
+
+Create a copy of `g`. If `deep` is `true`, then copy will be a deep copy (equivalent to `deepcopy(g)`),
+otherwise it will be a shallow copy with the same underlying graph data.
+"""
+function Base.copy(g::GNNGraph; deep=false)
+    if deep
+        GNNGraph(deepcopy(g.graph), 
+                g.num_nodes, g.num_edges, g.num_graphs, 
+                deepcopy(g.graph_indicator), 
+                deepcopy(g.ndata), deepcopy(g.edata), deepcopy(g.gdata))
+    else
+        GNNGraph(g.graph, 
+                g.num_nodes, g.num_edges, g.num_graphs, 
+                g.graph_indicator, 
+                g.ndata, g.edata, g.gdata)
+    end
+end
+
 function Base.show(io::IO, g::GNNGraph)
     print(io, "GNNGraph($(g.num_nodes), $(g.num_edges))")
 end

test/GNNGraphs/gnngraph.jl

@@ -313,6 +313,17 @@
         @test hash(g1) != hash(GNNGraph(g1, ndata=rand(5), graph_type=GRAPH_T))
         @test hash(g1) != hash(GNNGraph(g1, edata=rand(6), graph_type=GRAPH_T))
     end
+
+
+    @testset "copy" begin
+        g1 = rand_graph(10, 4, ndata=rand(2,10), graph_type=GRAPH_T)
+        g2 = copy(g1)
+        @test g1 === g2 # shallow copies are identical for immutable objects
+
+        g2 = copy(g1, deep=true)
+        @test g1 == g2
+        @test g1 !== g2 
+    end
 end