Generalized tests to apply to all time-dependent systems (except PDE).

Author: lamorton

src/systems/control/controlsystem.jl

@@ -72,6 +72,11 @@ struct ControlSystem <: AbstractControlSystem
     parameters are not supplied in `ODEProblem`.
     """
     defaults::Dict
+    function ControlSystem(loss, deqs, iv, dvs, controls,ps, observed, name, systems, defaults)
+        check_dependence(dvs,iv)
+        check_parameters(ps,iv)
+        new(loss, deqs, iv, dvs, controls,ps, observed, name, systems, defaults)
+    end
 end
 
 function ControlSystem(loss, deqs::AbstractVector{<:Equation}, iv, dvs, controls, ps;

src/systems/diffeqs/odesystem.jl

@@ -80,16 +80,6 @@ struct ODESystem <: AbstractODESystem
         new(deqs, iv, dvs, ps, observed, tgrad, jac, Wfact, Wfact_t, name, systems, defaults, structure, connection_type)
     end
 end
-function check_parameters(ps,iv)
-    for p in ps
-        isequal(iv,p) && throw(ArgumentError("Independent variable $iv not allowed in parameters."))
-    end
-end
-function check_dependence(dvs,iv)
-    for dv in dvs
-        isequal(iv, iv_from_nested_derivative(dv)) || throw(ArgumentError("Variable $dv is not a function of independent variable $iv."))
-    end
-end 
 
 function ODESystem(
                    deqs::AbstractVector{<:Equation}, iv, dvs, ps;

src/systems/diffeqs/sdesystem.jl

@@ -75,6 +75,12 @@ struct SDESystem <: AbstractODESystem
     type: type of the system
     """
     connection_type::Any
+
+    function SDESystem(deqs, neqs, iv, dvs, ps, observed, tgrad, jac, Wfact, Wfact_t, name, systems, defaults, connection_type)
+        check_dependence(dvs,iv)
+        check_parameters(ps,iv)
+        new(deqs, neqs, iv, dvs, ps, observed, tgrad, jac, Wfact, Wfact_t, name, systems, defaults, connection_type)
+    end
 end
 
 function SDESystem(deqs::AbstractVector{<:Equation}, neqs, iv, dvs, ps;

src/systems/discrete_system/discrete_system.jl

@@ -49,6 +49,11 @@ struct DiscreteSystem <: AbstractSystem
     in `DiscreteSystem`.
     """
     default_p::Dict
+    function DiscreteSystem(discreteEqs, iv, dvs, ps, observed, name, systems, default_u0, default_p)
+        check_dependence(dvs,iv)
+        check_parameters(ps,iv)
+        new(discreteEqs, iv, dvs, ps, observed, name, systems, default_u0, default_p)
+    end
 end
 
 """

src/systems/jumps/jumpsystem.jl

@@ -51,6 +51,11 @@ struct JumpSystem{U <: ArrayPartition} <: AbstractSystem
     type: type of the system
     """
     connection_type::Any
+    function JumpSystem{U}(ap::U, iv, states, ps, observed, name, systems, defaults, connection_type) where U <: ArrayPartition
+        check_dependence(states,iv)
+        check_parameters(ps,iv)
+        new{U}(ap, iv, states, ps, observed, name, systems, defaults, connection_type)
+    end
 end
 
 function JumpSystem(eqs, iv, states, ps;

src/systems/reaction/reactionsystem.jl

@@ -151,7 +151,12 @@ struct ReactionSystem <: AbstractSystem
     systems::Vector
 
     function ReactionSystem(eqs, iv, states, ps, observed, name, systems)
-        new(eqs, value(iv), value.(states), value.(ps), observed, name, systems)
+        iv′ = value(iv)
+        states′ = value.(states)
+        ps′ = value.(ps)
+        check_dependence(states′,iv′)
+        check_parameters(ps′,iv′)
+        new(eqs, iv′, states′, ps′, observed, name, systems)
     end
 end
 

src/utils.jl

@@ -104,3 +104,14 @@ function _readable_code(ex)
     expr
 end
 readable_code(expr) = JuliaFormatter.format_text(string(Base.remove_linenums!(_readable_code(expr))))
+
+function check_parameters(ps,iv)
+    for p in ps
+        isequal(iv,p) && throw(ArgumentError("Independent variable $iv not allowed in parameters."))
+    end
+end
+function check_dependence(dvs,iv)
+    for dv in dvs
+        isequal(iv, iv_from_nested_derivative(dv)) || throw(ArgumentError("Variable $dv is not a function of independent variable $iv."))
+    end
+end 

test/jumpsystem.jl

@@ -3,7 +3,7 @@ MT = ModelingToolkit
 
 # basic MT SIR model with tweaks
 @parameters β γ t
-@variables S I R
+@variables S(t) I(t) R(t)
 rate₁   = β*S*I
 affect₁ = [S ~ S - 1, I ~ I + 1]
 rate₂   = γ*I+t