fix formatting

Author: isaacsas

src/systems/callbacks.jl

@@ -441,7 +441,7 @@ function compile_affect(affect::FunctionalAffect, sys, dvs, ps; kwargs...)
 end
 
 function generate_timed_callback(cb, sys, dvs, ps; postprocess_affect_expr! = nothing,
-                                                   kwargs...)
+                                 kwargs...)
     cond = condition(cb)
     as = compile_affect(affects(cb), sys, dvs, ps; expression = Val{false},
                         postprocess_affect_expr!, kwargs...)
@@ -455,7 +455,7 @@ function generate_timed_callback(cb, sys, dvs, ps; postprocess_affect_expr! = no
 end
 
 function generate_discrete_callback(cb, sys, dvs, ps; postprocess_affect_expr! = nothing,
-                                                      kwargs...)
+                                    kwargs...)
     if is_timed_condition(cb)
         return generate_timed_callback(cb, sys, dvs, ps; postprocess_affect_expr!,
                                        kwargs...)

src/systems/jumps/jumpsystem.jl

@@ -342,7 +342,7 @@ sol = solve(jprob, SSAStepper())
 ```
 """
 function JumpProcesses.JumpProblem(js::JumpSystem, prob, aggregator; callback = nothing,
-                                                                     kwargs...)
+                                   kwargs...)
     statetoid = Dict(value(state) => i for (i, state) in enumerate(states(js)))
     eqs = equations(js)
     invttype = prob.tspan[1] === nothing ? Float64 : typeof(1 / prob.tspan[2])

test/root_equations.jl

@@ -338,7 +338,8 @@ sys = structural_simplify(model)
 @test isempty(ModelingToolkit.continuous_events(sys))
 
 let
-    function testsol(osys, u0, p, tspan; tstops = Float64[], skipparamtest = false, kwargs...)
+    function testsol(osys, u0, p, tspan; tstops = Float64[], skipparamtest = false,
+                     kwargs...)
         oprob = ODEProblem(osys, u0, tspan, p; kwargs...)
         sol = solve(oprob, Tsit5(); tstops = tstops, abstol = 1e-10, reltol = 1e-10)
         @test isapprox(sol(1.0000000001)[1] - sol(0.999999999)[1], 1.0; rtol = 1e-6)
@@ -401,22 +402,23 @@ let
     testsol(osys6, u0, p, tspan; tstops = [1.0, 2.0])
 
     # mix a continuous event too
-    cond3 = A ~ .1
+    cond3 = A ~ 0.1
     affect3 = [k ~ 0.0]
     cb3 = cond3 => affect3
     @named osys7 = ODESystem(eqs, t, [A], [k, t1, t2], discrete_events = [cb1, cb2‵‵‵],
                              continuous_events = [cb3])
     sol = testsol(osys7, u0, p, (0.0, 10.0); tstops = [1.0, 2.0], skipparamtest = true)
-    @test isapprox(sol(10.0)[1], .1; atol=1e-10, rtol=1e-10)
+    @test isapprox(sol(10.0)[1], 0.1; atol = 1e-10, rtol = 1e-10)
 end
 
 let
-    function testsol(ssys, u0, p, tspan; tstops = Float64[], skipparamtest = false, kwargs...)
+    function testsol(ssys, u0, p, tspan; tstops = Float64[], skipparamtest = false,
+                     kwargs...)
         sprob = SDEProblem(ssys, u0, tspan, p; kwargs...)
         sol = solve(sprob, RI5(); tstops = tstops, abstol = 1e-10, reltol = 1e-10)
         @test isapprox(sol(1.0000000001)[1] - sol(0.999999999)[1], 1.0; rtol = 1e-4)
         !skipparamtest && (@test sprob.p[1] == 1.0)
-        @test isapprox(sol(4.0)[1], 2 * exp(-2.0), atol=1e-4)
+        @test isapprox(sol(4.0)[1], 2 * exp(-2.0), atol = 1e-4)
         sol
     end
 
@@ -432,7 +434,8 @@ let
 
     ∂ₜ = Differential(t)
     eqs = [∂ₜ(A) ~ -k * A]
-    @named ssys = SDESystem(eqs, Equation[], t, [A], [k, t1, t2], discrete_events = [cb1, cb2])
+    @named ssys = SDESystem(eqs, Equation[], t, [A], [k, t1, t2],
+                            discrete_events = [cb1, cb2])
     u0 = [A => 1.0]
     p = [k => 0.0, t1 => 1.0, t2 => 2.0]
     tspan = (0.0, 4.0)
@@ -441,7 +444,8 @@ let
     cond1a = (t == t1)
     affect1a = [A ~ A + 1, B ~ A]
     cb1a = cond1a => affect1a
-    @named ssys1 = SDESystem(eqs, Equation[], t, [A, B], [k, t1, t2], discrete_events = [cb1a, cb2])
+    @named ssys1 = SDESystem(eqs, Equation[], t, [A, B], [k, t1, t2],
+                             discrete_events = [cb1a, cb2])
     u0′ = [A => 1.0, B => 0.0]
     sol = testsol(ssys1, u0′, p, tspan; tstops = [1.0, 2.0], check_length = false)
     @test sol(1.0000001, idxs = 2) == 2.0
@@ -453,7 +457,8 @@ let
     testsol(ssys‵, u0, p, tspan)
 
     # mixing discrete affects
-    @named ssys3 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2], discrete_events = [cb1, cb2‵])
+    @named ssys3 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2],
+                             discrete_events = [cb1, cb2‵])
     testsol(ssys3, u0, p, tspan; tstops = [1.0])
 
     # mixing with a func affect
@@ -462,29 +467,33 @@ let
         nothing
     end
     cb2‵‵ = [2.0] => (affect!, [], [k], nothing)
-    @named ssys4 = SDESystem(eqs, Equation[], t, [A], [k, t1], discrete_events = [cb1, cb2‵‵])
+    @named ssys4 = SDESystem(eqs, Equation[], t, [A], [k, t1],
+                             discrete_events = [cb1, cb2‵‵])
     testsol(ssys4, u0, p, tspan; tstops = [1.0])
 
     # mixing with symbolic condition in the func affect
     cb2‵‵‵ = (t == t2) => (affect!, [], [k], nothing)
-    @named ssys5 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2], discrete_events = [cb1, cb2‵‵‵])
+    @named ssys5 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2],
+                             discrete_events = [cb1, cb2‵‵‵])
     testsol(ssys5, u0, p, tspan; tstops = [1.0, 2.0])
-    @named ssys6 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2], discrete_events = [cb2‵‵‵, cb1])
+    @named ssys6 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2],
+                             discrete_events = [cb2‵‵‵, cb1])
     testsol(ssys6, u0, p, tspan; tstops = [1.0, 2.0])
 
     # mix a continuous event too
-    cond3 = A ~ .1
+    cond3 = A ~ 0.1
     affect3 = [k ~ 0.0]
     cb3 = cond3 => affect3
-    @named ssys7 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2], discrete_events = [cb1, cb2‵‵‵],
+    @named ssys7 = SDESystem(eqs, Equation[], t, [A], [k, t1, t2],
+                             discrete_events = [cb1, cb2‵‵‵],
                              continuous_events = [cb3])
     sol = testsol(ssys7, u0, p, (0.0, 10.0); tstops = [1.0, 2.0], skipparamtest = true)
-    @test isapprox(sol(10.0)[1], .1; atol=1e-10, rtol=1e-10)
+    @test isapprox(sol(10.0)[1], 0.1; atol = 1e-10, rtol = 1e-10)
 end
 
 let rng = rng
     function testsol(jsys, u0, p, tspan; tstops = Float64[], skipparamtest = false,
-                                         N = 40000, kwargs...)
+                     N = 40000, kwargs...)
         dprob = DiscreteProblem(jsys, u0, tspan, p)
         jprob = JumpProblem(jsys, dprob, Direct(); kwargs...)
         sol = solve(jprob, SSAStepper(); tstops = tstops)