Skip to content

Commit 9ca0835

Browse files
committed
Add GlobalPool test
1 parent a46a756 commit 9ca0835

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

GNNLux/test/layers/pool.jl

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
@testitem "Pooling" setup=[TestModuleLux] begin
2+
using .TestModuleLux
3+
@testset "GlobalPool" begin
4+
5+
rng = StableRNG(1234)
6+
g = rand_graph(rng, 10, 40)
7+
in_dims = 3
8+
x = randn(rng, Float32, in_dims, 10)
9+
10+
@testset "GCNConv" begin
11+
l = GlobalPool(mean)
12+
test_lux_layer(rng, l, g, x, sizey=(in_dims,1))
13+
end
14+
end
15+
end

0 commit comments

Comments
 (0)