fix: infer transposed input length

marko1olo · marko1olo · commit bdd3e0d65068 · 2026-06-08T00:35:18.000+04:00
diff --git a/control/tests/timeresp_test.py b/control/tests/timeresp_test.py
@@ -1245,6 +1245,17 @@ def test_response_transpose(
         assert y.shape == ysh_no
         assert x.shape == (T.size, sys.nstates)
 
+    def test_forced_response_transpose_without_time_vector(self):
+        sys = ct.ss(
+            [[0.5]], [[1.0, -0.25]], [[1.0]], [[0.0, 0.0]], dt=True)
+        U = np.column_stack((np.arange(5.0), np.arange(5.0) + 10))
+
+        response = ct.forced_response(sys, inputs=U, transpose=True)
+
+        np.testing.assert_allclose(response.time, np.arange(U.shape[0]))
+        np.testing.assert_allclose(response.inputs, U)
+        assert response.outputs.shape == (U.shape[0], sys.noutputs)
+
 
 @pytest.mark.pandas
 def test_to_pandas():
diff --git a/control/timeresp.py b/control/timeresp.py
@@ -1114,6 +1114,8 @@ def forced_response(
             # Set T to equally spaced samples with same length as U
             if U.ndim == 1:
                 n_steps = U.shape[0]
+            elif transpose:
+                n_steps = U.shape[0]
             else:
                 n_steps = U.shape[1]
             dt = 1. if sys.dt in [True, None] else sys.dt
