default prob_choice to 1

Author: ChrisRackauckas

Project.toml

@@ -1,7 +1,7 @@
 name = "DiffEqDevTools"
 uuid = "f3b72e0c-5b89-59e1-b016-84e28bfd966d"
 authors = ["Chris Rackauckas <[email protected]>"]
-version = "2.28.0"
+version = "2.29.0"
 
 [deps]
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"

src/benchmark.jl

@@ -46,6 +46,8 @@ function Shootout(prob,setups;appxsol=nothing,names=nothing,error_estimate=:fina
 
     if :prob_choice ∈ keys(setups[i])
       cur_appxsol = appxsol[setups[i][:prob_choice]]
+    elseif prob isa AbstractArray
+      cur_appxsol = appxsol[1]
     else
       cur_appxsol = appxsol
     end
@@ -61,6 +63,8 @@ function Shootout(prob,setups;appxsol=nothing,names=nothing,error_estimate=:fina
 
     if haskey(setups[i], :prob_choice)
       _prob = prob[setups[i][:prob_choice]]
+    elseif prob isa AbstractArray
+      _prob = prob[1]
     else
       _prob = prob
     end
@@ -171,6 +175,8 @@ function WorkPrecision(prob,alg,abstols,reltols,dts=nothing;
 
   if haskey(kwargs, :prob_choice)
     _prob = prob[kwargs[:prob_choice]]
+  elseif prob isa AbstractArray
+    _prob = prob[1]
   else
     _prob = prob
   end
@@ -191,6 +197,8 @@ function WorkPrecision(prob,alg,abstols,reltols,dts=nothing;
 
       if haskey(kwargs, :prob_choice)
         cur_appxsol = appxsol[kwargs[:prob_choice]]
+      elseif prob isa AbstractArray
+        cur_appxsol = appxsol[1]
       else
         cur_appxsol = appxsol
       end

test/benchmark_tests.jl

@@ -106,7 +106,7 @@ wp = WorkPrecisionSet(prob,abstols,reltols,setups;appxsol=test_sol,save_everyste
 # Dual Problem
 
 probs = [prob,prob_ode_2Dlinear]
-setups = [Dict(:alg=>DP5(),:prob_choice => 1)
+setups = [Dict(:alg=>DP5())
           Dict(:alg=>Tsit5(), :prob_choice => 2)
           Dict(:alg=>Vern6(), :prob_choice => 2)
           ]