Skip to content

Commit bf3c193

Browse files
Update core_tests.jl
1 parent d737b84 commit bf3c193

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

lib/SCCNonlinearSolve/test/core_tests.jl

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -20,38 +20,38 @@ end
2020
sol = solve(prob, NewtonRaphson())
2121

2222
u0 = zeros(2)
23-
cache = zeros(3)
23+
p = zeros(3)
2424

2525
function f1(du, u, (cache, p))
2626
du[1] = cos(u[2]) - u[1]
2727
du[2] = sin(u[1] + u[2]) + u[2]
2828
end
2929
explicitfun1(cache, sols) = nothing
3030
prob1 = NonlinearProblem(
31-
NonlinearFunction{true, SciMLBase.NoSpecialize}(f1), zeros(2), cache)
31+
NonlinearFunction{true, SciMLBase.NoSpecialize}(f1), zeros(2), p)
3232
sol1 = solve(prob1, NewtonRaphson())
3333

34-
function f2(du, u, (cache, p))
34+
function f2(du, u, p)
3535
du[1] = 2u[2] + u[1] + 1.0
3636
du[2] = u[3]^2 + u[2]
3737
du[3] = u[1]^2 + u[3]
3838
end
3939
explicitfun2(cache, sols) = nothing
4040
prob2 = NonlinearProblem(
41-
NonlinearFunction{true, SciMLBase.NoSpecialize}(f2), zeros(3), cache)
41+
NonlinearFunction{true, SciMLBase.NoSpecialize}(f2), zeros(3), p)
4242
sol2 = solve(prob2, NewtonRaphson())
4343

44-
function f3(du, u, (cache, p))
45-
du[1] = cache[1] + 2.0u[1] + 2.5u[2] + 1.5u[3]
46-
du[2] = cache[2] + 4.0u[1] - 1.5u[2] + 1.5u[3]
47-
du[3] = cache[3] + +u[1] - u[2] - u[3]
44+
function f3(du, u, p)
45+
du[1] = p[1] + 2.0u[1] + 2.5u[2] + 1.5u[3]
46+
du[2] = p[2] + 4.0u[1] - 1.5u[2] + 1.5u[3]
47+
du[3] = p[3] + +u[1] - u[2] - u[3]
4848
end
4949
prob3 = NonlinearProblem(
50-
NonlinearFunction{true, SciMLBase.NoSpecialize}(f3), zeros(3), cache)
51-
function explicitfun3(cache, sols)
52-
cache[1] = sols[1][1] + sols[1][2] + sols[2][1] + sols[2][2] + sols[2][3]
53-
cache[2] = sols[1][1] + sols[1][2] + sols[2][1] + 2.0sols[2][2] + sols[2][3]
54-
cache[3] = sols[1][1] + 2.0sols[1][2] + 3.0sols[2][1] + 5.0sols[2][2] +
50+
NonlinearFunction{true, SciMLBase.NoSpecialize}(f3), zeros(3), p)
51+
function explicitfun3(p, sols)
52+
p[1] = sols[1][1] + sols[1][2] + sols[2][1] + sols[2][2] + sols[2][3]
53+
p[2] = sols[1][1] + sols[1][2] + sols[2][1] + 2.0sols[2][2] + sols[2][3]
54+
p[3] = sols[1][1] + 2.0sols[1][2] + 3.0sols[2][1] + 5.0sols[2][2] +
5555
6.0sols[2][3]
5656
end
5757
explicitfun3(cache, [sol1, sol2])

0 commit comments

Comments
 (0)