test: update NonlinearSystem tests with new Initial parameters

Author: AayushSabharwal

test/nonlinearsystem.jl

@@ -239,11 +239,13 @@ testdict = Dict([:test => 1])
 
     prob_ = remake(prob, u0 = [1.0, 2.0, 3.0], p = [a => 1.1, b => 1.2, c => 1.3])
     @test prob_.u0 == [1.0, 2.0, 3.0]
-    @test prob_.p == MTKParameters(sys, [a => 1.1, b => 1.2, c => 1.3])
+    initials = unknowns(sys) .=> ones(3)
+    @test prob_.p == MTKParameters(sys, [a => 1.1, b => 1.2, c => 1.3, initials...])
 
     prob_ = remake(prob, u0 = Dict(y => 2.0), p = Dict(a => 2.0))
     @test prob_.u0 == [1.0, 2.0, 1.0]
-    @test prob_.p == MTKParameters(sys, [a => 2.0, b => 1.0, c => 1.0])
+    initials = [x => 1.0, y => 2.0, z => 1.0]
+    @test prob_.p == MTKParameters(sys, [a => 2.0, b => 1.0, c => 1.0, initials...])
 end
 
 @testset "Observed function generation without parameters" begin

test/parameter_dependencies.jl

@@ -326,7 +326,7 @@ end
     eqs = [0 ~ p1 * x * exp(x) + p2]
     @mtkbuild sys = NonlinearSystem(eqs; parameter_dependencies = [p2 => 2p1])
     @test isequal(only(parameters(sys)), p1)
-    @test Set(full_parameters(sys)) == Set([p1, p2, Initial(p2)])
+    @test Set(full_parameters(sys)) == Set([p1, p2, Initial(p2), Initial(x)])
     prob = NonlinearProblem(sys, [x => 1.0])
     @test prob.ps[p1] == 1.0
     @test prob.ps[p2] == 2.0