Merge pull request #488 from frankschae/concrete_steady

SteadyStateSolution has no field t error

Author: ChrisRackauckas

src/solve.jl

@@ -210,6 +210,12 @@ function _concrete_solve(prob::DiffEqBase.DEProblem,alg::DiffEqBase.DEAlgorithm,
   RecursiveArrayTools.DiffEqArray(sol.u,sol.t)
 end
 
+function _concrete_solve(prob::DiffEqBase.SteadyStateProblem,alg::DiffEqBase.DEAlgorithm,
+                        u0=prob.u0,p=prob.p,args...;kwargs...)
+  sol = solve(remake(prob,u0=u0,p=p),alg,args...;kwargs...)
+  RecursiveArrayTools.VectorOfArray(sol.u)
+end
+
 function ChainRulesCore.frule(::typeof(concrete_solve),prob,alg,u0,p,args...;
                      sensealg=nothing,kwargs...)
   _concrete_solve_forward(prob,alg,sensealg,u0,p,args...;kwargs...)

test/downstream/Project.toml

@@ -3,4 +3,5 @@ DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 DiffEqCallbacks = "459566f4-90b8-5000-8ac3-15dfb0a30def"
 DiffEqProblemLibrary = "a077e3f3-b75c-5d7f-a0c6-6bc4c8ec64a9"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
+SteadyStateDiffEq = "9672c7b4-1e72-59bd-8a11-6ac3964bc41f"
 StochasticDiffEq = "789caeaf-c7a9-5a7d-9973-96adeb23e2a0"

test/downstream/concrete_solve_tests.jl

@@ -0,0 +1,18 @@
+using OrdinaryDiffEq, StochasticDiffEq, SteadyStateDiffEq, RecursiveArrayTools, Test
+
+function f!(du,u,p,t)
+  du[1] = p[1] + p[2]*u[1]
+  du[2] = p[3]*u[1] + p[4]*u[2]
+end
+u0 = zeros(2)
+p = [2.0,-2.0,1.0,-4.0]
+
+probODE = ODEProblem(f!,u0,(0.0,10.0),p)
+probSS = SteadyStateProblem(f!,u0,p)
+
+solODE = concrete_solve(probODE,Tsit5(), abstol=1e-14,reltol=1e-14)
+solSS = concrete_solve(probSS,DynamicSS(Rodas5()))
+
+@test solSS.u ≈ solODE.u[end] rtol = 1e-8
+@test length(fieldnames(typeof(solODE))) == 2
+@test length(fieldnames(typeof(solSS))) == 1

test/runtests.jl

@@ -46,6 +46,7 @@ if !is_APPVEYOR && GROUP == "Downstream"
     @time @safetestset "PSOS and Energy Conservation Event Detection" begin include("downstream/psos_and_energy_conservation.jl") end
     @time @safetestset "DE stats" begin include("downstream/destats_tests.jl") end
     @time @safetestset "DEDataArray" begin include("downstream/data_array_regression_tests.jl") end
+    @time @safetestset "Concrete_solve Tests" begin include("downstream/concrete_solve_tests.jl") end
 end
 
 if !is_APPVEYOR && GROUP == "GPU"