add some tests

Author: TorkelE

test/upstream/mtk_structure_indexing.jl

@@ -339,17 +339,23 @@ let
 
     # Checks for both ODE and SDE problems.
     for (XProblem, solver) in zip([ODEProblem, SDEProblem], [Tsit5(), ImplicitEM()])
-        # Create the problems which we wish to check..
+        # Create the problem which we wish to check.
         u0 = [X1 => 1.0, X2 => 2.0, Y1 => 3.0, Y2 => 4.0, W => 6.0]
         ps = [k1 => 0.1, k2 => 0.2, V0 => 3.0]
         prob1 = XProblem(rs, u0, 0.001, ps; remove_conserved = true)
+        Γ = prob1.f.sys.Γ
+        
+        # Creates various `remake` version of the problem.
         prob2 = remake(prob1, u0 = [X1 => 10.0])
         prob3 = remake(prob2, u0 = [X2 => 20.0])
         prob4 = remake(prob1, u0 = [X2 => 20.0, Y1 => 30.0])
         prob5 = remake(prob1, u0 = [X1 => 10.0, X2 => 20.0])
         prob6 = remake(prob1, u0 = [Y2 => 40.0], p = [k1 => 0.4])
         prob7 = remake(prob1, u0 = [X1 => 10.0, X2 => 20.0], p = [V0 => 50.0])
         prob8 = remake(prob1, u0 = [W => 60.0])
+        prob9 = remake(prob2; p = [Γ => [10.0, 20.0]])        
+        prob10 = remake(prob1; u0 = [Y1 => 20.0], p = [Γ => [20.0, 30.0], k1 => 0.4])
+        prob11 = remake(prob10, u0 = [X1 => 10.0], p = [k2 => 0.5])
 
         # Creates a testing function.
         function test_vals(prob, us_correct::Dict, ps_correct::Dict)
@@ -393,6 +399,18 @@ let
         test_vals(prob8,
             Dict(X1 => 1.0, X2 => 2.0, Y1 => 3.0, Y2 => 4.0, V => 3.0, W => 60.0),
             Dict(k1 => 0.1, k2 => 0.2, V0 => 3.0, v => 3.0, w => 60.0, Γ[1] => 3.0, Γ[2] => 7.0))
+        test_vals(prob8,
+            Dict(X1 => 1.0, X2 => 2.0, Y1 => 3.0, Y2 => 4.0, V => 3.0, W => 60.0),
+            Dict(k1 => 0.1, k2 => 0.2, V0 => 3.0, v => 3.0, w => 60.0, Γ[1] => 3.0, Γ[2] => 7.0))
+        test_vals(prob9,
+            Dict(X1 => 10.0, X2 => 0.0, Y1 => 3.0, Y2 => 17.0, V => 3.0, W => 6.0),
+            Dict(k1 => 0.1, k2 => 0.2, V0 => 3.0, v => 3.0, w => 6.0, Γ[1] => 10.0, Γ[2] => 20.0))
+        test_vals(prob10,
+            Dict(X1 => 1.0, X2 => 19.0, Y1 => 20.0, Y2 => 10.0, V => 3.0, W => 6.0),
+            Dict(k1 => 0.4, k2 => 0.2, V0 => 3.0, v => 3.0, w => 6.0, Γ[1] => 20.0, Γ[2] => 30.0))
+        test_vals(prob11,
+            Dict(X1 => 10.0, X2 => 10.0, Y1 => 20.0, Y2 => 10.0, V => 3.0, W => 6.0),
+            Dict(k1 => 0.4, k2 => 0.5, V0 => 3.0, v => 3.0, w => 6.0, Γ[1] => 20.0, Γ[2] => 30.0))
     end
 end