diff --git a/test/GNNGraphs/query.jl b/test/GNNGraphs/query.jl index dcd4bcd3c..9b70ba2db 100644 --- a/test/GNNGraphs/query.jl +++ b/test/GNNGraphs/query.jl @@ -137,12 +137,10 @@ t = [2, 3, 4, 5, 1, 5, 1, 2, 3, 4] g = GNNGraph(s,t) @test laplacian_lambda_max(g) ≈ Float32(1.809017) - data1 = [g for i in 1:5] - gall1 = Flux.batch(data1) - @test laplacian_lambda_max(gall1) ≈ [Float32(1.809017) for i in 1:5] - data2 = [rand_graph(10,20) for i in 1:3] - gall2 = Flux.batch(data2) - @test length(laplacian_lambda_max(gall2)) == 3 + data = [g for i in 1:5] + gall = Flux.batch(data) + @test laplacian_lambda_max(gall) ≈ [Float32(1.809017) for i in 1:5] + @test length(laplacian_lambda_max(gall)) == 5 end @testset "adjacency_matrix" begin