add unit tests for consistent systems repr() processing

murrayrm · murrayrm · commit b0397e071080 · 2025-01-12T17:27:13.000-08:00
diff --git a/control/tests/iosys_test.py b/control/tests/iosys_test.py
@@ -2339,3 +2339,54 @@ def test_signal_prefixing(fcn):
     assert sys.output_labels == ['yy[0]']
     if sys.nstates:
         assert sys.state_labels == ['xx[0]', 'xx[1]']
+
+@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
