fix: fix test

JuliaGraphs · Oct 11, 2024 · 9b8301a · 9b8301a
1 parent ed513dc
commit 9b8301a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/GNNGraphs/src/sampling.jl b/GNNGraphs/src/sampling.jl
@@ -114,7 +114,7 @@ function sample_neighbors(g::GNNGraph{<:COO_T}, nodes, K = -1;
                         graph_indicator,
                         ndata, edata, g.gdata)
     end
-    return gnew
+    return gne
 end
 
 """

diff --git a/GNNGraphs/test/sampling.jl b/GNNGraphs/test/sampling.jl
@@ -64,11 +64,10 @@ if GRAPH_T == :coo
         nodes = [1, 2]
         subgraph = Graphs.induced_subgraph(graph, nodes)
 
-        @test subgraph.num_nodes == 3  
-        @test subgraph.num_edges == 2  
-        @test subgraph.ndata.x == graph.ndata.x
-        @test subgraph.ndata.y == graph.ndata.y
-        @test subgraph.edata == graph.edata
+        @test subgraph.num_nodes == 2 
+        @test subgraph.num_edges == 1 
+        @test subgraph.ndata == getobs(g.ndata, [1, 2])
+        @test subgraph.edata == getobs(graph.edata, 1)
 
         graph = GNNGraph(2)
         graph = add_edges(graph, ([2], [1]))