add unit tests for consistent systems repr() processing

Author: murrayrm

control/tests/iosys_test.py

@@ -17,7 +17,7 @@
 import scipy
 
 import control as ct
-
+from control.tests.conftest import slycotonly
 
 class TestIOSys:
 
@@ -2284,3 +2284,54 @@ def test_signal_indexing():
     with pytest.raises(IndexError, match=r"signal name\(s\) not valid"):
         resp.outputs['y[0]', 'u[0]']
 
+
+@slycotonly
+@pytest.mark.parametrize("fcn, spec, expected, missing", [
+    (ct.ss, {}, "\nstates=4, outputs=3, inputs=2", r"dt|name"),
+    (ct.tf, {}, "\noutputs=3, inputs=2", r"dt|name|states"),
+    (ct.frd, {}, "\noutputs=3, inputs=2", r"dt|states|name"),
+    (ct.ss, {'dt': 0.1}, ".*\ndt=0.1,\nstates=4, outputs=3, inputs=2", r"name"),
+    (ct.tf, {'dt': 0.1}, ".*\ndt=0.1,\noutputs=3, inputs=2", r"name|states"),
+    (ct.frd, {'dt': 0.1},
+     ".*\ndt=0.1,\noutputs=3, inputs=2", r"name|states"),
+    (ct.ss, {'dt': True}, "\ndt=True,\nstates=4, outputs=3, inputs=2", r"name"),
+    (ct.ss, {'dt': None}, "\ndt=None,\nstates=4, outputs=3, inputs=2", r"name"),
+    (ct.ss, {'dt': 0}, "\nstates=4, outputs=3, inputs=2", r"dt|name"),
+    (ct.ss, {'name': 'mysys'}, "\nname='mysys',", r"dt"),
+    (ct.tf, {'name': 'mysys'}, "\nname='mysys',", r"dt|states"),
+    (ct.frd, {'name': 'mysys'}, "\nname='mysys',", r"dt|states"),
+    (ct.ss, {'inputs': ['u1']},
+     r"[\n]states=4, outputs=3, inputs=\['u1'\]", r"dt|name"),
+    (ct.tf, {'inputs': ['u1']},
+     r"[\n]outputs=3, inputs=\['u1'\]", r"dt|name"),
+    (ct.frd, {'inputs': ['u1'], 'name': 'sampled'},
+     r"[\n]name='sampled', outputs=3, inputs=\['u1'\]", r"dt"),
+    (ct.ss, {'outputs': ['y1']},
+     r"[\n]states=4, outputs=\['y1'\], inputs=2", r"dt|name"),
+    (ct.ss, {'name': 'mysys', 'inputs': ['u1']},
+     r"[\n]name='mysys', states=4, outputs=3, inputs=\['u1'\]", r"dt"),
+    (ct.ss, {'name': 'mysys', 'states': [
+        'long_state_1', 'long_state_2', 'long_state_3']},
+     r"[\n]name='.*', states=\[.*\],[\n]outputs=3, inputs=2\)", r"dt"),
+])
+def test_system_repr(fcn, spec, expected, missing):
+    spec['outputs'] = spec.get('outputs', 3)
+    spec['inputs'] = spec.get('inputs', 2)
+    if fcn is ct.ss:
+        spec['states'] = spec.get('states', 4)
+
+    sys = ct.rss(**spec)
+    match fcn:
+        case ct.frd:
+            omega = np.logspace(-1, 1)
+            sys = fcn(sys, omega, name=spec.get('name'))
+        case ct.tf:
+            sys = fcn(sys, name=spec.get('name'))
+
+    assert sys.shape == (sys.noutputs, sys.ninputs)
+
+    out = repr(sys)
+    assert re.search(expected, out) != None
+
+    if missing is not None:
+        assert re.search(missing, out) is None