fix tests

Author: CarloLucibello

src/GNNGraphs/query.jl

@@ -182,15 +182,15 @@ function Graphs.degree(g::GNNGraph{<:ADJMAT_T}, T=nothing; dir=:out, edge_weight
                   vec(sum(A, dims=1)) .+ vec(sum(A, dims=2)) 
 end
 
-function Graphs.laplacian_matrix(g::GNNGraph, T::DataType=Float32; dir::Symbol=:out)
+function Graphs.laplacian_matrix(g::GNNGraph, T::DataType=nodetype(g); dir::Symbol=:out)
     A = adjacency_matrix(g, T; dir=dir)
     D = Diagonal(vec(sum(A; dims=2)))
     return D - A
 end
 
 
 """
-    normalized_laplacian(g, T=nothing; add_self_loops=false, dir=:out)
+    normalized_laplacian(g, T=Float32; add_self_loops=false, dir=:out)
 
 Normalized Laplacian matrix of graph `g`.
 
@@ -219,7 +219,7 @@ function normalized_adjacency(g::GNNGraph, T::DataType=Float32;
 end
 
 @doc raw"""
-    scaled_laplacian(g, T=nothing; dir=:out)
+    scaled_laplacian(g, T=Float32; dir=:out)
 
 Scaled Laplacian matrix of graph `g`,
 defined as ``\hat{L} = \frac{2}{\lambda_{max}} L - I`` where ``L`` is the normalized Laplacian matrix.

test/GNNGraphs/query.jl

@@ -52,7 +52,7 @@
                 @test Array(d_gpu) == d
             end
         end
-        
+
         @testset "weighted" begin
             # weighted degree
             s = [1, 1, 2, 3]
@@ -84,4 +84,13 @@
             end
         end
     end
+
+    @testset "laplacian_matrix" begin
+        g = rand_graph(10, 30, graph_type=GRAPH_T)
+        A = adjacency_matrix(g)
+        D = Diagonal(vec(sum(A, dims=2)))
+        L = laplacian_matrix(g)
+        @test eltype(L) == GNNGraphs.nodetype(g)
+        @test L ≈ D - A
+    end
 end

test/layers/conv.jl

@@ -40,7 +40,7 @@
     end
 
     @testset "ChebConv" begin
-        k = 3
+        k = 2
         l = ChebConv(in_channel => out_channel, k)
         @test size(l.weight) == (out_channel, in_channel, k)
         @test size(l.bias) == (out_channel,)