fixup! Test if labels are transferred to the response and we can still relabel them

joaoantoniocardoso · joaoantoniocardoso · commit e6ab9fec20b9 · 2023-05-27T14:30:03.000-03:00
diff --git a/control/tests/trdata_test.py b/control/tests/trdata_test.py
@@ -237,10 +237,15 @@ def test_trdata_labels():
         response.input_labels, ["u[%d]" % i for i in range(sys.ninputs)])
 
     # Make sure the selected input and output are both correctly transferred to the response
-    for n,_ in enumerate(U):
-        step_response = ct.step_response(sys, T, input=n, output=n)
-        assert step_response.input_labels == [sys.input_labels[n]]
-        assert step_response.output_labels == [sys.output_labels[n]]
+    for nu in range(sys.ninputs):
+        for ny in range(sys.noutputs):
+            step_response = ct.step_response(sys, T, input=nu, output=ny)
+            assert step_response.input_labels == [sys.input_labels[nu]]
+            assert step_response.output_labels == [sys.output_labels[ny]]
+
+            init_response = ct.initial_response(sys, T, input=nu, output=ny)
+            assert init_response.input_labels == None
+            assert init_response.output_labels == [sys.output_labels[ny]]
 
 
 def test_trdata_multitrace():
