Update sde_nondiag_stratonovich.jl

SciML · Jan 10, 2023 · 34c1809 · 34c1809
1 parent d3e0de2
commit 34c1809
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/test/sde_nondiag_stratonovich.jl b/test/sde_nondiag_stratonovich.jl
@@ -161,19 +161,18 @@ end
                                                         dt = dtnd, adaptive = false,
                                                         sensealg = BacksolveAdjoint(autojacvec = ZygoteVJP()))
 
-        @test all(isapprox(res_sde_u0a, res_sde_u0, rtol = 1e-6),isapprox(res_sde_pa, res_sde_p, rtol = 1e-6))
+        @test all((isapprox(res_sde_u0a, res_sde_u0, rtol = 1e-6),isapprox(res_sde_pa, res_sde_p, rtol = 1e-6)))
     end
 
     @info res_sde_pa
 
-    @test_broken begin
-        res_sde_u0a, res_sde_pa = adjoint_sensitivities(sol, EulerHeun(), t = Array(t),
+    res_sde_u0a, res_sde_pa = adjoint_sensitivities(sol, EulerHeun(), t = Array(t),
                                                     dgdu_discrete = dg!,
                                                     dt = dtnd, adaptive = false,
                                                     sensealg = BacksolveAdjoint(autojacvec = false))
 
-        @test all(isapprox(res_sde_u0a, res_sde_u0, rtol = 1e-6),isapprox(res_sde_pa, res_sde_p, rtol = 1e-6))
-    end
+    @test isapprox(res_sde_u0a, res_sde_u0, rtol = 1e-6)
+    @test isapprox(res_sde_pa, res_sde_p, rtol = 1e-6)
 
     @info res_sde_pa
 
@@ -195,7 +194,7 @@ end
                                                         dt = dtnd, adaptive = false,
                                                         sensealg = InterpolatingAdjoint(autojacvec = ZygoteVJP()))
 
-        @test all(isapprox(res_sde_u0a, res_sde_u0, rtol = 1e-5),isapprox(res_sde_pa, res_sde_p, rtol = 1e-4))
+        @test all((isapprox(res_sde_u0a, res_sde_u0, rtol = 1e-5),isapprox(res_sde_pa, res_sde_p, rtol = 1e-4)))
     end
 
     @info res_sde_pa