Add test

Author: YingboMa

src/systems/diffeqs/modelingtoolkitize.jl

@@ -54,14 +54,23 @@ function modelingtoolkitize(prob::DiffEqBase.ODEProblem; kwargs...)
 
     sts = vec(collect(vars))
 
+    _params = params
     params = values(params)
     params = if params isa Number || (params isa Array && ndims(params) == 0)
         [params[1]]
     else
         vec(collect(params))
     end
     default_u0 = Dict(sts .=> vec(collect(prob.u0)))
-    default_p = has_p ? Dict(params .=> vec(collect(prob.p))) : Dict()
+    default_p = if has_p
+        if prob.p isa AbstractDict
+            Dict(v => prob.p[k] for (k, v) in pairs(_params))
+        else
+            Dict(params .=> vec(collect(prob.p)))
+        end
+    else
+        Dict()
+    end
 
     de = ODESystem(eqs, t, sts, params,
                    defaults = merge(default_u0, default_p);
@@ -95,7 +104,7 @@ function define_params(p)
 end
 
 function define_params(p::AbstractDict)
-    Dict(k => toparam(variable(:α, i)) for (i, k) in zip(1:length(p), keys(p)))
+    OrderedDict(k => toparam(variable(:α, i)) for (i, k) in zip(1:length(p), keys(p)))
 end
 
 function define_params(p::Union{SLArray, LArray})

test/modelingtoolkitize.jl

@@ -1,4 +1,4 @@
-using OrdinaryDiffEq, ModelingToolkit, Test
+using OrdinaryDiffEq, ModelingToolkit, DataStructures, Test
 using Optimization, RecursiveArrayTools, OptimizationOptimJL
 
 N = 32
@@ -277,3 +277,15 @@ params = (1, 1)
 prob = ODEProblem(ode_prob, [1 1], (0, 1), params)
 sys = modelingtoolkitize(prob)
 @test nameof.(parameters(sys)) == [:α₁, :α₂]
+
+function ode_prob_dict(du, u, p, t)
+    du[1] = u[1] + p[:a]
+    du[2] = u[2] + p[:b]
+    nothing
+end
+params = OrderedDict(:a => 10, :b => 20)
+u0 = [1, 2.0]
+prob = ODEProblem(ode_prob_dict, u0, (0.0, 1.0), params)
+sys = modelingtoolkitize(prob)
+@test [ModelingToolkit.defaults(sys)[s] for s in states(sys)] == u0
+@test [ModelingToolkit.defaults(sys)[s] for s in parameters(sys)] == [10, 20]