shrink test

Author: foldfelis

test/fourier.jl

@@ -8,12 +8,11 @@
     )
     @test ndims(SpectralConv(ch, modes)) == 1
 
-    𝐱, _ = get_burgers_data(n=1000)
-    @test size(m(𝐱)) == (64, 1024, 1000)
+    𝐱, _ = get_burgers_data(n=5)
+    @test size(m(𝐱)) == (64, 1024, 5)
 
-    T = Float32
     loss(x, y) = Flux.mse(m(x), y)
-    data = [(T.(𝐱[:, :, 1:5]), rand(T, 64, 1024, 5))]
+    data = [(𝐱, rand(Float32, 64, 1024, 5))]
     Flux.train!(loss, params(m), data, Flux.ADAM())
 end
 
@@ -26,11 +25,11 @@ end
         FourierOperator(ch, modes)
     )
 
-    𝐱, _ = get_burgers_data(n=1000)
-    @test size(m(𝐱)) == (64, 1024, 1000)
+    𝐱, _ = get_burgers_data(n=5)
+    @test size(m(𝐱)) == (64, 1024, 5)
 
     loss(x, y) = Flux.mse(m(x), y)
-    data = [(Float32.(𝐱[:, :, 1:5]), rand(Float32, 64, 1024, 5))]
+    data = [(𝐱, rand(Float32, 64, 1024, 5))]
     Flux.train!(loss, params(m), data, Flux.ADAM())
 end
 
@@ -44,12 +43,11 @@ end
     )
     @test ndims(SpectralConv(ch, modes)) == 2
 
-    𝐱, _ , _, _ = get_darcy_flow_data()
-    @test size(m(𝐱)) == (64, 85, 85, 1024)
+    𝐱, _, _, _ = get_darcy_flow_data(n=5, Δsamples=20)
+    @test size(m(𝐱)) == (64, 22, 22, 5)
 
-    T = Float32
     loss(x, y) = Flux.mse(m(x), y)
-    data = [(T.(𝐱[:, :, :, 1:5]), rand(T, 64, 85, 85, 5))]
+    data = [(𝐱, rand(Float32, 64, 22, 22, 5))]
     Flux.train!(loss, params(m), data, Flux.ADAM())
 end
 
@@ -62,10 +60,10 @@ end
         FourierOperator(ch, modes)
     )
 
-    𝐱, _ , _, _ = get_darcy_flow_data()
-    @test size(m(𝐱)) == (64, 85, 85, 1024)
+    𝐱, _, _, _ = get_darcy_flow_data(n=5, Δsamples=20)
+    @test size(m(𝐱)) == (64, 22, 22, 5)
 
     loss(x, y) = Flux.mse(m(x), y)
-    data = [(Float32.(𝐱[:, :, :, 1:5]), rand(Float32, 64, 85, 85, 5))]
+    data = [(𝐱, rand(Float32, 64, 22, 22, 5))]
     Flux.train!(loss, params(m), data, Flux.ADAM())
 end