update unit test per @slivinsgston review

murrayrm · murrayrm · commit 7707ea8a665b · 2025-01-02T08:23:01.000-08:00
diff --git a/control/tests/timeresp_test.py b/control/tests/timeresp_test.py
@@ -1278,6 +1278,8 @@ def test_to_pandas():
     df = res.to_pandas()
     np.testing.assert_equal(
         df[df['trace'] == 'From u0']['time'], res.time)
+    np.testing.assert_equal(
+        df[df['trace'] == 'From u0']['u0'], res.inputs['u0', 0])
     np.testing.assert_equal(
         df[df['trace'] == 'From u0']['y1'], res.outputs['y1', 0])
 
@@ -1287,7 +1289,9 @@ def test_to_pandas():
         np.testing.assert_equal(
             df[df['trace'] == label]['time'], res.time)
         np.testing.assert_equal(
-            df[df['trace'] == label]['u0'], res.inputs['u0', i])
+            df[df['trace'] == label]['u1'], res.inputs['u1', i])
+        np.testing.assert_equal(
+            df[df['trace'] == label]['y0'], res.outputs['y0', i])
 
 
 @pytest.mark.skipif(pandas_check(), reason="pandas installed")
