Add test for loss

Author: foldfelis

test/loss.jl

@@ -0,0 +1,9 @@
+@testset "loss" begin
+    𝐲 = rand(1, 3, 3, 5)
+    𝐲̂ = rand(1, 3, 3, 5)
+
+    feature_dims = 2:3
+    loss = mean(.√(sum(abs2, 𝐲̂-𝐲, dims=feature_dims)))
+
+    @test l₂loss(𝐲̂, 𝐲) ≈ loss/prod(feature_dims)
+end

test/runtests.jl

@@ -4,19 +4,21 @@ using Flux
 using GeometricFlux
 using Graphs
 using Zygote
+using Statistics
 using Test
 
 CUDA.allowscalar(false)
 
 cuda_tests = [
-    "cuda",
+    "cuda.jl",
 ]
 
 tests = [
-    "Transform/Transform",
-    "operator_kernel",
-    "model",
-    "deeponet",
+    "Transform/Transform.jl",
+    "operator_kernel.jl",
+    "loss.jl",
+    "model.jl",
+    "deeponet.jl",
 ]
 
 if CUDA.functional()
@@ -27,7 +29,7 @@ end
 
 @testset "NeuralOperators.jl" begin
     for t in tests
-        include("$(t).jl")
+        include(t)
     end
 end