From dbe5c700b634a0b47b6d95fc0c70dbd676f5059d Mon Sep 17 00:00:00 2001
From: Christopher Rackauckas <accounts@chrisrackauckas.com>
Date: Mon, 26 Feb 2024 10:19:44 -0600
Subject: [PATCH] format

---
 src/systems/abstractsystem.jl             |  3 ++-
 src/systems/diffeqs/abstractodesystem.jl  | 20 +++++++-------
 src/systems/nonlinear/initializesystem.jl |  4 +--
 test/initializationsystem.jl              | 33 ++++++++++++-----------
 4 files changed, 33 insertions(+), 27 deletions(-)

diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
index 2daf445f24..b317eed8e1 100644
--- a/src/systems/abstractsystem.jl
+++ b/src/systems/abstractsystem.jl
@@ -2097,7 +2097,8 @@ end
 
 returns a `Vector{Pair}` of variables set to `default` which are missing from `get_defaults(sys)`.  The `default` argument can be a single value or vector to set the missing defaults respectively.
 """
-function missing_variable_defaults(sys::AbstractSystem, default = 0.0; subset = unknowns(sys))
+function missing_variable_defaults(
+        sys::AbstractSystem, default = 0.0; subset = unknowns(sys))
     varmap = get_defaults(sys)
     varmap = Dict(Symbolics.diff2term(value(k)) => value(varmap[k]) for k in keys(varmap))
     missingvars = setdiff(subset, keys(varmap))
diff --git a/src/systems/diffeqs/abstractodesystem.jl b/src/systems/diffeqs/abstractodesystem.jl
index 1629f59e94..44dec64159 100644
--- a/src/systems/diffeqs/abstractodesystem.jl
+++ b/src/systems/diffeqs/abstractodesystem.jl
@@ -325,8 +325,8 @@ function DiffEqBase.ODEFunction{iip, specialize}(sys::AbstractODESystem,
         checkbounds = false,
         sparsity = false,
         analytic = nothing,
-        split_idxs = nothing,  
-        initializeprob =  nothing,
+        split_idxs = nothing,
+        initializeprob = nothing,
         initializeprobmap = nothing,
         kwargs...) where {iip, specialize}
     if !iscomplete(sys)
@@ -540,7 +540,7 @@ function DiffEqBase.DAEFunction{iip}(sys::AbstractODESystem, dvs = unknowns(sys)
         sparse = false, simplify = false,
         eval_module = @__MODULE__,
         checkbounds = false,
-        initializeprob =  nothing,
+        initializeprob = nothing,
         initializeprobmap = nothing,
         kwargs...) where {iip}
     if !iscomplete(sys)
@@ -861,12 +861,13 @@ function process_DEProblem(constructor, sys::AbstractODESystem, u0map, parammap;
     # since they will be checked in the initialization problem's construction
     # TODO: make check for if a DAE cheaper than calculating the mass matrix a second time!
     if implicit_dae || calculate_massmatrix(sys) !== I
-        initializeprob =  ModelingToolkit.InitializationProblem(sys, u0map, parammap; guesses, warn_initialize_determined)
+        initializeprob = ModelingToolkit.InitializationProblem(
+            sys, u0map, parammap; guesses, warn_initialize_determined)
         initializeprobmap = getu(initializeprob, unknowns(sys))
-        zerovars = setdiff(unknowns(sys),defaults(sys)) .=> 0.0
-        trueinit =  identity.([zerovars;u0map])
+        zerovars = setdiff(unknowns(sys), defaults(sys)) .=> 0.0
+        trueinit = identity.([zerovars; u0map])
     else
-        initializeprob =  nothing
+        initializeprob = nothing
         initializeprobmap = nothing
         trueinit = u0map
     end
@@ -907,7 +908,7 @@ function process_DEProblem(constructor, sys::AbstractODESystem, u0map, parammap;
         checkbounds = checkbounds, p = p,
         linenumbers = linenumbers, parallel = parallel, simplify = simplify,
         sparse = sparse, eval_expression = eval_expression,
-        initializeprob = initializeprob, 
+        initializeprob = initializeprob,
         initializeprobmap = initializeprobmap,
         kwargs...)
     implicit_dae ? (f, du0, u0, p) : (f, u0, p)
@@ -1523,7 +1524,8 @@ function InitializationProblem{iip, specialize}(sys::AbstractODESystem, u0map =
     if isempty(u0map)
         isys = get_initializesystem(sys)
     else
-        isys = structural_simplify(generate_initializesystem(sys; u0map); fully_determined = false)
+        isys = structural_simplify(
+            generate_initializesystem(sys; u0map); fully_determined = false)
     end
 
     neqs = length(equations(isys))
diff --git a/src/systems/nonlinear/initializesystem.jl b/src/systems/nonlinear/initializesystem.jl
index 8b7ce8c9e0..51e9c49480 100644
--- a/src/systems/nonlinear/initializesystem.jl
+++ b/src/systems/nonlinear/initializesystem.jl
@@ -3,7 +3,7 @@ $(TYPEDSIGNATURES)
 
 Generate `NonlinearSystem` which initializes an ODE problem from specified initial conditions of an `ODESystem`.
 """
-function generate_initializesystem(sys::ODESystem; 
+function generate_initializesystem(sys::ODESystem;
         u0map = Dict(),
         name = nameof(sys),
         guesses = Dict(), check_defguess = false, kwargs...)
@@ -15,7 +15,7 @@ function generate_initializesystem(sys::ODESystem;
     # Start the equations list with algebraic equations
     eqs_ics = eqs[idxs_alge]
     u0 = Vector{Pair}(undef, 0)
-    defs = merge(defaults(sys),todict(u0map))
+    defs = merge(defaults(sys), todict(u0map))
 
     full_states = [sts; getfield.((observed(sys)), :lhs)]
 
diff --git a/test/initializationsystem.jl b/test/initializationsystem.jl
index 147900310d..2e6fc8c95c 100644
--- a/test/initializationsystem.jl
+++ b/test/initializationsystem.jl
@@ -3,41 +3,44 @@ using ModelingToolkit: t_nounits as t, D_nounits as D
 
 @parameters g
 @variables x(t) y(t) [state_priority = 10] λ(t)
-eqs = [
-       D(D(x)) ~ λ * x
+eqs = [D(D(x)) ~ λ * x
        D(D(y)) ~ λ * y - g
-       x^2 + y^2 ~ 1
-      ]
-@mtkbuild pend = ODESystem(eqs,t)
+       x^2 + y^2 ~ 1]
+@mtkbuild pend = ODESystem(eqs, t)
 
-initprob = ModelingToolkit.InitializationProblem(pend, [], [g => 1]; guesses = [ModelingToolkit.missing_variable_defaults(pend); x => 1; y => 0.2])
-conditions = getfield.(equations(initprob.f.sys),:rhs)
+initprob = ModelingToolkit.InitializationProblem(pend, [], [g => 1];
+    guesses = [ModelingToolkit.missing_variable_defaults(pend); x => 1; y => 0.2])
+conditions = getfield.(equations(initprob.f.sys), :rhs)
 
 @test initprob isa NonlinearLeastSquaresProblem
 sol = solve(initprob)
 @test SciMLBase.successful_retcode(sol)
 @test maximum(abs.(sol[conditions])) < 1e-14
 
-initprob = ModelingToolkit.InitializationProblem(pend, [x => 1, y => 0], [g => 1]; guesses = ModelingToolkit.missing_variable_defaults(pend))
+initprob = ModelingToolkit.InitializationProblem(pend, [x => 1, y => 0], [g => 1];
+    guesses = ModelingToolkit.missing_variable_defaults(pend))
 @test initprob isa NonlinearProblem
 sol = solve(initprob)
 @test SciMLBase.successful_retcode(sol)
-@test sol.u == [1.0,0.0,0.0,0.0]
+@test sol.u == [1.0, 0.0, 0.0, 0.0]
 @test maximum(abs.(sol[conditions])) < 1e-14
 
-initprob = ModelingToolkit.InitializationProblem(pend, [], [g => 1]; guesses = ModelingToolkit.missing_variable_defaults(pend))
+initprob = ModelingToolkit.InitializationProblem(
+    pend, [], [g => 1]; guesses = ModelingToolkit.missing_variable_defaults(pend))
 @test initprob isa NonlinearLeastSquaresProblem
 sol = solve(initprob)
 @test !SciMLBase.successful_retcode(sol)
 
-prob = ODEProblem(pend, [x => 1, y => 0], (0.0, 1.5), [g => 1], guesses = ModelingToolkit.missing_variable_defaults(pend))
+prob = ODEProblem(pend, [x => 1, y => 0], (0.0, 1.5), [g => 1],
+    guesses = ModelingToolkit.missing_variable_defaults(pend))
 prob.f.initializeprob isa NonlinearProblem
 sol = solve(prob.f.initializeprob)
 @test maximum(abs.(sol[conditions])) < 1e-14
 sol = solve(prob, Rodas5P())
 @test maximum(abs.(sol[conditions][1])) < 1e-14
 
-prob = ODEProblem(pend, [x => 1], (0.0, 1.5), [g => 1], guesses = ModelingToolkit.missing_variable_defaults(pend))
+prob = ODEProblem(pend, [x => 1], (0.0, 1.5), [g => 1],
+    guesses = ModelingToolkit.missing_variable_defaults(pend))
 prob.f.initializeprob isa NonlinearLeastSquaresProblem
 sol = solve(prob.f.initializeprob)
 @test maximum(abs.(sol[conditions])) < 1e-14
@@ -214,7 +217,7 @@ end
 
 @mtkbuild sys = System()
 initprob = ModelingToolkit.InitializationProblem(sys)
-conditions = getfield.(equations(initprob.f.sys),:rhs)
+conditions = getfield.(equations(initprob.f.sys), :rhs)
 
 @test initprob isa NonlinearLeastSquaresProblem
 @test length(initprob.u0) == 2
@@ -229,7 +232,7 @@ sol = solve(prob, Rodas5P(), initializealg = BrownFullBasicInit())
 @test sol.retcode == SciMLBase.ReturnCode.Unstable
 @test maximum(abs.(initsol[conditions][1])) < 1e-14
 
-prob = ODEProblem(sys, [], (0, 0.1), check=false)
+prob = ODEProblem(sys, [], (0, 0.1), check = false)
 sol = solve(prob, Rodas5P())
 # If initialized incorrectly, then it would be InitialFailure
 @test sol.retcode == SciMLBase.ReturnCode.Unstable
@@ -328,4 +331,4 @@ p = [σ => 28.0,
     β => 8 / 3]
 
 tspan = (0.0, 100.0)
-@test_throws ArgumentError prob = ODEProblem(sys, u0, tspan, p, jac = true)
\ No newline at end of file
+@test_throws ArgumentError prob=ODEProblem(sys, u0, tspan, p, jac = true)