cleanup

Author: CarloLucibello

src/GNNGraphs/utils.jl

@@ -1,5 +1,5 @@
 function check_num_nodes(g::GNNGraph, x::AbstractArray)
-    @assert g.num_nodes == size(x, ndims(x)) "Got $(size(x, ndims(x))) as last dimension size instead of num_edges=$(g.num_nodes)"
+    @assert g.num_nodes == size(x, ndims(x)) "Got $(size(x, ndims(x))) as last dimension size instead of num_nodes=$(g.num_nodes)"
     return true
 end
 function check_num_nodes(g::GNNGraph, x::Union{Tuple,NamedTuple})

test/msgpass.jl

@@ -138,10 +138,12 @@
     end
 
     @testset "aggregate_neighbors" begin
-        m = rand(2, g.num_edges-1)
-        @test_throws AssertionError aggregate_neighbors(g, +, m)
+        @testset "sizecheck" begin
+            m = rand(2, g.num_edges-1)
+            @test_throws AssertionError aggregate_neighbors(g, +, m)
 
-        m = (a=rand(2, g.num_edges+1), b=nothing)
-        @test_throws AssertionError aggregate_neighbors(g, +, m)        
+            m = (a=rand(2, g.num_edges+1), b=nothing)
+            @test_throws AssertionError aggregate_neighbors(g, +, m)
+        end        
     end
 end