Merge pull request #70 from JuliaGNI/make_tests_quicker

Only computing two instances of initial conditions and parameters now.
JuliaGNI · Jul 8, 2024 · de55721 · de55721
2 parents 382b318 + 3fb5f60
commit de55721
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/test/test_hodeensembles.jl b/test/test_hodeensembles.jl
@@ -12,19 +12,20 @@ function test_multiple_initial_conditions(cho::Module)
     # ensemble solution
     esol = integrate(epr, ImplicitMidpoint())
 
-    sol = integrate(cho.hodeproblem(), ImplicitMidpoint())
-    @test esol.s[1].q.d.parent ≈ sol.q.d.parent
-    @test esol.s[2].q.d.parent ≉ sol.q.d.parent
+    @test esol.s[2].q.d.parent ≉ esol.s[1].q.d.parent
 end
 
 function test_multiple_parameters(cho::Module)
     params_vec = Vector{NamedTuple}()
 
     for i in 0:1
+        param_vals = ()
+
         param_vals = ()
         for key in keys(cho.default_parameters)
             param_vals = (param_vals..., cho.default_parameters[key] .+ 1. * i)
         end
+
         params_vec = push!(params_vec, NamedTuple{keys(cho.default_parameters)}(param_vals))
     end
 
@@ -33,9 +34,7 @@ function test_multiple_parameters(cho::Module)
     # ensemble solution
     esol = integrate(epr, ImplicitMidpoint())
 
-    sol = integrate(cho.hodeproblem(), ImplicitMidpoint())
-    @test esol.s[1].q.d.parent ≈ sol.q.d.parent
-    @test esol.s[2].q.d.parent ≉ sol.q.d.parent
+    @test esol.s[2].q.d.parent ≉ esol.s[1].q.d.parent
 end
 
 test_multiple_initial_conditions(cho)