Fix impuse response, input data only contains one inpute[0]==1 in each trace

KybernetikJo · KybernetikJo · commit a03ed38c1dbf · 2024-07-03T20:42:18.000+02:00
diff --git a/control/timeresp.py b/control/timeresp.py
@@ -1960,7 +1960,7 @@ def impulse_response(sys, T=None, input=None, output=None, T_num=None,
         yout[:, inpidx, :] = response.y if output is None \
             else response.y[output]
         xout[:, inpidx, :] = response.x
-        uout[:, inpidx, :] = U[i]
+        uout[:, inpidx, :] = U if input is None else U[i]
 
     # Figure out if the system is SISO or not
     issiso = sys.issiso() or (input is not None and output is not None)
