Add tests for polynomial line search models

sblauth · Sep 14, 2022 · 0c3c7a4 · 0c3c7a4
1 parent 66eecf2
commit 0c3c7a4
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/test_optimal_control.py b/tests/test_optimal_control.py
@@ -752,3 +752,22 @@ def test_safeguard_gd():
     ocp = cashocs.OptimalControlProblem(F, bcs, J, y, u, p, config)
     ocp.solve("gd", rtol=1e-2, atol=0.0, max_iter=50)
     assert ocp.solver.relative_norm <= ocp.solver.rtol
+
+
+def test_polynomial_stepsize():
+    config = cashocs.load_config(dir_path + "/config_ocp.ini")
+    config.set("LineSearch", "method", "polynomial")
+    u.vector().vec().set(0.0)
+    u.vector().apply("")
+    ocp = cashocs.OptimalControlProblem(F, bcs, J, y, u, p, config)
+    ocp.solve("gd", rtol=1e-2, atol=0.0, max_iter=42)
+    assert ocp.solver.relative_norm <= ocp.solver.rtol
+
+    config = cashocs.load_config(dir_path + "/config_ocp.ini")
+    config.set("LineSearch", "method", "polynomial")
+    config.set("LineSearch", "polynomial_model", "quadratic")
+    u.vector().vec().set(0.0)
+    u.vector().apply("")
+    ocp = cashocs.OptimalControlProblem(F, bcs, J, y, u, p, config)
+    ocp.solve("gd", rtol=1e-2, atol=0.0, max_iter=44)
+    assert ocp.solver.relative_norm <= ocp.solver.rtol