add more tests

Author: pzimbrod

test/complexweights.jl

@@ -0,0 +1,13 @@
+using Test, Random, Flux
+
+@testset "Weights" begin
+    @testset "uniform" begin
+        @test size(OperatorLearning.cglorot_uniform(128,64,10)) == (128, 64, 10)
+        @test eltype(OperatorLearning.cglorot_uniform(128,64,10)) == ComplexF32
+    end
+
+    @testset "normal" begin
+        @test size(OperatorLearning.cglorot_normal(128,64,10)) == (128, 64, 10)
+        @test eltype(OperatorLearning.cglorot_normal(128,64,10)) == ComplexF32
+    end
+end

test/fourierlayer.jl

@@ -1,15 +1,31 @@
 using Test, Random, Flux
 
 @testset "FourierLayer" begin
-    # Test the proper construction
-    @test size(FourierLayer(128, 64, 100, 20).Wf) == (128, 64, 51)
-    @test size(FourierLayer(128, 64, 100, 20).Wl) == (64, 128)
-    #@test size(FourierLayer(10, 100).bias_f) == (51,)
-    #@test size(FourierLayer(10, 100).bias_l) == (100,)
+    @testset "dimensions" begin
+        # Test the proper construction
+        @test size(FourierLayer(128, 64, 100, 20).Wf) == (128, 64, 51)
+        @test size(FourierLayer(128, 64, 100, 20).Wl) == (64, 128)
+        @test size(FourierLayer(128, 64, 100, 20).bf) == (1, 64, 51)
+        @test size(FourierLayer(128, 64, 100, 20).bl) == (1, 64, 100)
+    end
+
+    # Test proper parameter assignment
+    # We only use a subset of the weight tensors for training
+    @testset "parameters" begin
+        # Wf
+        @test size(params(FourierLayer(128, 64, 100, 20))[1]) == (128, 64, 20)
+        # Wl
+        @test size(params(FourierLayer(128, 64, 100, 20))[2]) == (128, 64, 20)
+        # bf
+        @test size(params(FourierLayer(128, 64, 100, 20))[3]) == (1, 64, 20)
+        # bl
+        @test size(params(FourierLayer(128, 64, 100, 20))[4]) == (1, 64, 100)
+    end
 
     # Accept only Int as architecture parameters
     @test_throws MethodError FourierLayer(128.5, 64, 100, 20)
     @test_throws MethodError FourierLayer(128.5, 64, 100, 20, tanh)
+    # Test max amount of modes
     @test_throws AssertionError FourierLayer(100, 100, 100, 60, σ)
     @test_throws AssertionError FourierLayer(100, 100, 100, 60)
 end

test/runtests.jl

@@ -7,3 +7,7 @@ Random.seed!(0)
 @testset "FourierLayer" begin
     include("fourierlayer.jl")
 end
+
+@testset "Weights" begin
+    include("complexweights.jl")
+end