Skip to content

Commit 827e04a

Browse files
committed
bench: OrthogonalCollocation unit tests
1 parent 78e5f5f commit 827e04a

1 file changed

Lines changed: 23 additions & 0 deletions

File tree

benchmark/3_bench_predictive_control.jl

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,17 @@ nmpc_nonlin_tc = NonLinMPC(
6868
nonlinmodel_c, transcription=TrapezoidalCollocation(),
6969
Mwt=[1], Nwt=[0.1], Lwt=[0.1], Hp=10
7070
)
71+
JuMP.set_attribute(nmpc_nonlin_tc.optim, "tol", 1e-7)
72+
nmpc_nonlin_oc = NonLinMPC(
73+
nonlinmodel_c, transcription=OrthogonalCollocation(),
74+
Mwt=[1], Nwt=[0.1], Lwt=[0.1], Hp=10
75+
)
76+
JuMP.set_attribute(nmpc_nonlin_oc.optim, "tol", 1e-7)
77+
nmpc_nonlin_oc_hess = NonLinMPC(
78+
nonlinmodel_c, transcription=OrthogonalCollocation(), hessian=true,
79+
Mwt=[1], Nwt=[0.1], Lwt=[0.1], Hp=10
80+
)
81+
JuMP.set_attribute(nmpc_nonlin_oc_hess.optim, "tol", 1e-7)
7182

7283
samples, evals, seconds = 10000, 1, 60
7384
UNIT_MPC["NonLinMPC"]["moveinput!"]["LinModel"]["SingleShooting"] =
@@ -118,6 +129,18 @@ UNIT_MPC["NonLinMPC"]["moveinput!"]["NonLinModel"]["TrapezoidalCollocation"] =
118129
setup=preparestate!($nmpc_nonlin_tc, $y_c, $d_c),
119130
samples=samples, evals=evals, seconds=seconds
120131
)
132+
UNIT_MPC["NonLinMPC"]["moveinput!"]["NonLinModel"]["OrthogonalCollocation"] =
133+
@benchmarkable(
134+
moveinput!($nmpc_nonlin_oc, $y_c, $d_c),
135+
setup=preparestate!($nmpc_nonlin_oc, $y_c, $d_c),
136+
samples=samples, evals=evals, seconds=seconds
137+
)
138+
UNIT_MPC["NonLinMPC"]["moveinput!"]["NonLinModel"]["OrthogonalCollocationHessian"] =
139+
@benchmarkable(
140+
moveinput!($nmpc_nonlin_oc_hess, $y_c, $d_c),
141+
setup=preparestate!($nmpc_nonlin_oc_hess, $y_c, $d_c),
142+
samples=samples, evals=evals, seconds=seconds
143+
)
121144
UNIT_MPC["NonLinMPC"]["getinfo!"]["NonLinModel"] =
122145
@benchmarkable(
123146
getinfo($nmpc_nonlin_ss),

0 commit comments

Comments
 (0)