fix: fix test

JuliaGraphs · Oct 11, 2024 · a2d506b · a2d506b
1 parent 2f528aa
commit a2d506b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/GNNGraphs/test/sampling.jl b/GNNGraphs/test/sampling.jl
@@ -61,6 +61,14 @@ if GRAPH_T == :coo
         @test subgraph.ndata.y == graph.ndata.y
         @test subgraph.edata == graph.edata
 
+        nodes = [1, 2]
+        subgraph = Graphs.induced_subgraph(graph, nodes)
+
+        @test subgraph.num_nodes == 2 
+        @test subgraph.num_edges == 1 
+        @test subgraph.ndata == getobs(graph.ndata, [1, 2])
+        @test subgraph.edata.e == getobs(graph.edata, 1)
+
         graph = GNNGraph(2)
         graph = add_edges(graph, ([2], [1]))
         nodes = [1]