test: add UpwindTest

junyixu · Aug 25, 2024 · edcf42b · edcf42b
1 parent 849927d
commit edcf42b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/test/UpwindTest.jl b/test/UpwindTest.jl
@@ -4,8 +4,8 @@ using TrixiEnzyme
 using TrixiEnzyme: upwind!
 using ForwardDiff
 
-x = -1:0.005:1
-batch_size = 2
+x = -1:0.01:1
+batch_size = 11
 jacobian_enzyme_forward(TrixiEnzyme.upwind!, x, N=batch_size)
 
 Δt₁ = @elapsed  J1 = jacobian_enzyme_forward(TrixiEnzyme.upwind!, x)
@@ -24,7 +24,7 @@ end;
 
 end; # 0.326764 seconds (1.09 M allocations: 75.013 MiB, 7.28% gc time, 99.87% compilation time)
 
-@test J1 = J2
+@test J1 == J2
 
 @info "Compare the time consumed by Enzyme.jl Δt and ForwardDiff.jl..."
 @info "Enzyme: $Δt₁ seconds | ForwardDiff: $Δt₂ seconds."

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -7,6 +7,11 @@ using Test, TrixiEnzyme
         t = @elapsed include("CommonTest.jl")
         println("##### done (took $t seconds).")
     end
+    @testset "Upwind" begin
+        println("##### Testing Upwind...")
+        t = @elapsed include("UpwindTest.jl")
+        println("##### done (took $t seconds).")
+    end
     @testset "Semi" begin
         println("##### Testing semi...")
         t = @elapsed include("SemiTest.jl")