add unit test showing failure in #1087

murrayrm · murrayrm · commit 448fb620ac9b · 2024-12-30T14:15:53.000-08:00
diff --git a/control/tests/timeresp_test.py b/control/tests/timeresp_test.py
@@ -1247,13 +1247,14 @@ def test_to_pandas():
     np.testing.assert_equal(df['x[1]'], resp.states[1])
 
     # Change the time points
-    sys = ct.rss(2, 1, 1)
+    sys = ct.rss(2, 1, 2)
     T = np.linspace(0, timepts[-1]/2, timepts.size * 2)
-    resp = ct.input_output_response(sys, timepts, np.sin(timepts), t_eval=T)
+    resp = ct.input_output_response(
+        sys, timepts, [np.sin(timepts), 0], t_eval=T)
     df = resp.to_pandas()
     np.testing.assert_equal(df['time'], resp.time)
-    np.testing.assert_equal(df['u[0]'], resp.inputs)
-    np.testing.assert_equal(df['y[0]'], resp.outputs)
+    np.testing.assert_equal(df['u[0]'], resp.inputs[0])
+    np.testing.assert_equal(df['y[0]'], resp.outputs[0])
     np.testing.assert_equal(df['x[0]'], resp.states[0])
     np.testing.assert_equal(df['x[1]'], resp.states[1])
 
@@ -1265,6 +1266,17 @@ def test_to_pandas():
     np.testing.assert_equal(df['u[0]'], resp.inputs)
     np.testing.assert_equal(df['y[0]'], resp.inputs * 5)
 
+    # https://github.com/python-control/python-control/issues/1087
+    model = ct.rss(
+        states=['x0', 'x1'], outputs=['y0', 'y1'],
+        inputs=['u0', 'u1'], name='My Model')
+    T = np.linspace(0, 10, 100, endpoint=False)
+    X0 = np.zeros(model.nstates)
+    res = ct.step_response(model, T=T, X0=X0, input=0)
+    df = res.to_pandas()
+    np.testing.assert_equal(df['time'], res.time)
+    np.testing.assert_equal(df['y1'], res.outputs['y1'])
+
 
 @pytest.mark.skipif(pandas_check(), reason="pandas installed")
 def test_no_pandas():
