Add GlobalPool test

JuliaGraphs · Dec 23, 2024 · 9ca0835 · 9ca0835
1 parent a46a756
commit 9ca0835
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/GNNLux/test/layers/pool.jl b/GNNLux/test/layers/pool.jl
@@ -0,0 +1,15 @@
+@testitem "Pooling" setup=[TestModuleLux] begin
+    using .TestModuleLux
+    @testset "GlobalPool" begin
+
+        rng = StableRNG(1234)
+        g = rand_graph(rng, 10, 40)
+        in_dims = 3
+        x = randn(rng, Float32, in_dims, 10)
+
+        @testset "GCNConv" begin
+            l = GlobalPool(mean)
+            test_lux_layer(rng, l, g, x, sizey=(in_dims,1))
+        end
+    end
+end