Import ControlArgument in flatsys.py, with test

roryyorke · roryyorke · commit 44db1264af4b · 2025-02-02T07:09:58.000+02:00
diff --git a/control/flatsys/flatsys.py b/control/flatsys/flatsys.py
@@ -8,6 +8,7 @@
 import warnings
 from .poly import PolyFamily
 from .systraj import SystemTrajectory
+from ..exception import ControlArgument
 from ..nlsys import NonlinearIOSystem
 from ..timeresp import _check_convert_array
 
diff --git a/control/tests/flatsys_test.py b/control/tests/flatsys_test.py
@@ -618,6 +618,11 @@ def test_point_to_point_errors(self):
                 flat_sys, timepts, x0, u0, xf, uf,
                 constraints=[(None, 0, 0, 0)], basis=fs.PolyFamily(8))
 
+        # too few timepoints
+        with pytest.raises(ct.ControlArgument, match="at least three time points"):
+            fs.point_to_point(
+                flat_sys, timepts[:2], x0, u0, xf, uf, basis=fs.PolyFamily(10), cost=cost_fcn)
+
         # Unsolvable optimization
         constraint = [opt.input_range_constraint(flat_sys, -0.01, 0.01)]
         with pytest.warns(UserWarning, match="unable to solve"):
