Skip to content
This repository was archived by the owner on Sep 28, 2024. It is now read-only.

Commit 0185b1f

Browse files
committed
Enhance test for OperatorConv
1 parent 6a0a238 commit 0185b1f

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

test/fourier.jl

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,3 +145,12 @@ end
145145
data = [(𝐱, rand(Float32, 22, 22, 64, 5))]
146146
Flux.train!(loss, params(m), data, Flux.ADAM())
147147
end
148+
149+
@testset "SpectralConv" begin
150+
modes = (16, 16)
151+
ch = 64 => 64
152+
permuted = false
153+
154+
@test SpectralConv(ch, modes) isa OperatorConv
155+
@test SpectralConv(ch, modes).transform isa FourierTransform
156+
end

0 commit comments

Comments
 (0)