Another test

Author: YingboMa

src/systems/diffeqs/sdesystem.jl

@@ -558,7 +558,7 @@ function DiffEqBase.SDEProblem{iip}(sys::SDESystem, u0map = [], tspan = get_tspa
     sparsenoise === nothing && (sparsenoise = get(kwargs, :sparse, false))
 
     noiseeqs = get_noiseeqs(sys)
-    if noiseeqs isa AbstractVector
+    if noiseeqs isa AbstractVector || size(noiseeqs, 2) == 1
         noise_rate_prototype = nothing
     elseif sparsenoise
         I, J, V = findnz(SparseArrays.sparse(noiseeqs))

src/systems/systems.jl

@@ -77,11 +77,12 @@ function structural_simplify(sys::AbstractSystem, io = nothing; simplify = false
             _iszero(dvar) && break
             g_row = get(dvar2eq, dvar, 0)
             iszero(g_row) && error("$dvar isn't handled.")
+            g_row > size(g, 1) && continue
             @views copyto!(sorted_g_rows[i, :], g[g_row, :])
         end
 
-        return SDESystem(equations(ode_sys), sorted_g_rows, get_iv(ode_sys),
-                         states(ode_sys), parameters(ode_sys);
+        return SDESystem(full_equations(ode_sys), sorted_g_rows,
+                         get_iv(ode_sys), states(ode_sys), parameters(ode_sys);
                          name = nameof(ode_sys))
     end
 end

test/sdesystem.jl

@@ -614,3 +614,52 @@ sys2 = SDESystem(drift_eqs, diffusion_eqs, t, sts, ps, name = :sys1)
 prob = SDEProblem(sys1, sts .=> [1.0, 0.0, 0.0],
                   (0.0, 100.0), ps .=> (10.0, 26.0))
 @test_nowarn solve(prob, LambaEulerHeun(), seed = 1)
+
+@variables t
+D = Differential(t)
+sts = @variables S(t) E(t) I(t) R(t) D1(t) D2(t) β(t) logβ(t)
+ps = @parameters σ ρ γ λ N
+@brownian α
+eqs = Equation[
+               β ~ logβ #exp(logβ)
+               D(logβ) ~ sqrt(0.2) * α
+               D(S) ~ -β * S * I / N
+               E ~ N - (S + I + R + D1 + D2)
+               D(I) ~ σ * E - γ * I
+               D(R) ~ (1 - ρ) * γ * I
+               D(D1) ~ ρ * γ * I - λ * D1
+               D(D2) ~ λ * D1]
+@named mecha_bayes = System(eqs, t)
+mecha_bayes = structural_simplify(mecha_bayes)
+dE = 0.4
+dI = 2.0
+Rh = 3.0
+NN = 1
+using Random
+Random.seed!(1)
+
+E0 = rand(Uniform(0, 0.02NN))
+I0 = rand(Uniform(0, 0.02NN))
+R0 = rand(Uniform(0, 0.02NN))
+D10 = rand(Uniform(0, 0.02NN))
+D20 = rand(Uniform(0, 0.02NN))
+S0 = 1 - (E0 + I0 + R0 + D10 + D20)
+u0 = [
+      S =>  S0
+      E =>  E0
+      I =>  I0
+      R =>  R0
+      D1 => D10
+      D2 => D20
+      β => 0.0
+      logβ => log(rand(Gamma(1, dI / Rh)))
+     ]
+
+p = [N => NN,
+     σ => rand(Gamma(100, 100dE)),
+     ρ => rand(Beta(1, 99)),
+     γ => rand(Gamma(100, 100dI)),
+     λ => rand(Gamma(10, 10*25))]
+prob = SDEProblem(mecha_bayes, u0,
+                  (0.0, 10.0), p)
+@test_nowarn sol = solve(prob, SRIW1(), reltol=1e-3, abstol=1e-3, seed = 1);