more tests

Author: isaacsas

test/dsl/dsl_advanced_model_construction.jl

@@ -360,3 +360,25 @@ let
     @named twostate = ReactionSystem([rx1, rx2], t)
     @test twostate == rn
 end
+
+############## tests related to hybrid systems ###################
+let
+    t = default_t()
+    D = default_time_deriv()
+    @parameters λ k
+    @variables V(t)
+    @species A(t)
+    rx = Reaction(k*V, [], [A])
+    eq = D(V) ~ λ*V
+    cevents = [[V ~ 2.0] => [V ~ V/2, A ~ A/2]]
+    @named hybrid = ReactionSystem([rx, eq], t; continuous_events = cevents)
+    rn = @reaction_network hybrid begin
+        @parameters λ
+        k*V, 0 --> A
+        @equations D(V) ~ λ*V
+        @continuous_events begin
+            [V ~ 2.0] => [V ~ V/2, A ~ A/2]
+        end
+    end        
+    @test hybrid == rn
+end

test/reactionsystem_core/reactionsystem.jl

@@ -1002,37 +1002,28 @@ let
     @test size(nps.incidencemat) == (3,3)
 end
 
-# test JumpInputs function auto problem selection
+########## tests related to hybrid systems ##########
 let
-    rn = @reaction_network begin
-        k*(1 + sin(t)), 0 --> A
-    end
-    jinput = JumpInputs(rn, [:A => 0], (0.0, 10.0), [:k => .5])
-    @test jinput.prob isa ODEProblem
-    jprob = JumpProblem(jinput; rng)
-    sol = solve(jprob, Tsit5())
-    @test sol(10.0; idxs = :A) > 0
-
-    rn = @reaction_network begin
-        k, 0 --> A
-    end
-    jinput = JumpInputs(rn, [:A => 0], (0.0, 10.0), [:k => .5])
-    @test jinput.prob isa DiscreteProblem
-    jprob = JumpProblem(jinput; rng)
-    sol = solve(jprob)
-    @test sol(10.0; idxs = :A) > 0
-
-    rn = @reaction_network begin
-        @parameters λ
-        k*V, 0 --> A
-        @equations D(V) ~ λ*V
-        @continuous_events begin
-            [V ~ 2.0] => [V ~ V/2, A ~ rand()*A]
-        end
-    end        
-    jinput = JumpInputs(rn, [:A => 0, :V => 1.0], (0.0, 10.0), [:k => 1.0, :λ => .2])
+    t = default_t()
+    D = default_time_deriv()
+    @parameters λ k
+    @variables V(t)
+    @species A(t) B(t) C(t)
+    rxs = [Reaction(k*V, [], [A]), Reaction(λ*A, [B], nothing),
+        Reaction(k, [A, B], nothing), Reaction(λ, [C], [A])]
+    eqs = [D(V) ~ λ*V*C]
+    cevents = [[V ~ 2.0] => [V ~ V/2, A ~ A/2]]
+    @named rs = ReactionSystem(vcat(rxs, eqs), t; continuous_events = cevents)
+    rs = complete(rs)
+    jinput = JumpInputs(rn, [:A => 0, :V => 1.0], (0.0, 10.0), [:k => 1.0, :λ => .4])
     @test jinput.prob isa ODEProblem
-    jprob = JumpProblem(jinput; rng)
-    sol = solve(jprob, Tsit5()) 
-
-end
+    sys = jinput.sys
+    @test sys isa JumpSystem
+    @test MT.has_equations(sys)
+    eqs = MT.equations(sys)
+    @test isempty(eqs.x[1]) 
+    @test isempty(eqs.x[2]) 
+    @test length(eqs.x[3]) == 1
+    @test length(eqs.x[4]) == 2  # dA/dt and dV/dt
+    @test length(continuous_events(sys)) == 1
+end

test/simulation_and_solving/hybrid_models.jl

@@ -0,0 +1,43 @@
+# Fetch packages.
+using Catalyst, JumpProcesses, OrdinaryDiffEqTsit5, Statistics, Test
+
+# Sets stable rng number.
+using StableRNGs
+rng = StableRNG(12345)
+seed = rand(rng, 1:100)
+
+# test JumpInputs function auto problem selection
+let
+    rn = @reaction_network begin
+        k*(1 + sin(t)), 0 --> A
+    end
+    jinput = JumpInputs(rn, [:A => 0], (0.0, 10.0), [:k => .5])
+    @test jinput.prob isa ODEProblem
+    jprob = JumpProblem(jinput; rng)
+    sol = solve(jprob, Tsit5())
+    @test sol(10.0; idxs = :A) > 0
+
+    rn = @reaction_network begin
+        k, 0 --> A
+    end
+    jinput = JumpInputs(rn, [:A => 0], (0.0, 10.0), [:k => .5])
+    @test jinput.prob isa DiscreteProblem
+    jprob = JumpProblem(jinput; rng)
+    sol = solve(jprob)
+    @test sol(10.0; idxs = :A) > 0
+
+    rn = @reaction_network begin
+        @parameters λ
+        k*V, 0 --> A
+        @equations D(V) ~ λ*V
+        @continuous_events begin
+            [V ~ 2.0] => [V ~ V/2, A ~ A/2]
+        end
+    end        
+    jinput = JumpInputs(rn, [:A => 0, :V => 1.0], (0.0, 10.0), [:k => 1.0, :λ => .4])
+    @test jinput.prob isa ODEProblem
+    jprob = JumpProblem(jinput; rng)
+    sol = solve(jprob, Tsit5()) 
+end
+
+