Correct plotting of robust MIMO example

roryyorke · roryyorke · commit f63323c9a3e4 · 2021-01-24T13:56:52.000+02:00
Done by passing `squeeze=True` to `step_response`.
diff --git a/examples/robust_mimo.py b/examples/robust_mimo.py
@@ -54,11 +54,8 @@ def analysis():
     g = plant()
 
     t = np.linspace(0, 10, 101)
-    _, yu1 = step_response(g, t, input=0)
-    _, yu2 = step_response(g, t, input=1)
-
-    yu1 = yu1
-    yu2 = yu2
+    _, yu1 = step_response(g, t, input=0, squeeze=True)
+    _, yu2 = step_response(g, t, input=1, squeeze=True)
 
     # linear system, so scale and sum previous results to get the
     # [1,-1] response
@@ -112,8 +109,8 @@ def synth(wb1, wb2):
 
 def step_opposite(g, t):
     """reponse to step of [-1,1]"""
-    _, yu1 = step_response(g, t, input=0)
-    _, yu2 = step_response(g, t, input=1)
+    _, yu1 = step_response(g, t, input=0, squeeze=True)
+    _, yu2 = step_response(g, t, input=1, squeeze=True)
     return yu1 - yu2
 
 
