clear figure in unit tests to avoid slow plotting problem

murrayrm · murrayrm · commit 2702ee7f4e5f · 2021-01-02T08:28:39.000-08:00
diff --git a/control/tests/freqresp_test.py b/control/tests/freqresp_test.py
@@ -304,11 +304,10 @@ def test_initial_phase(TF, initial_phase, default_phase, expected_phase):
     assert(abs(phase[0] - expected_phase) < 0.1)
 
     # Make sure everything works in rad/sec as well
-    # Turn off plotting since that seems to slow things down a lot (?)
     if initial_phase:
+        plt.clf()               # clear previous figure (speeds things up)
         mag, phase, omega = ctrl.bode(
-            TF, initial_phase=initial_phase/180. * math.pi, deg=False,
-            plot=False)
+            TF, initial_phase=initial_phase/180. * math.pi, deg=False)
         assert(abs(phase[0] - expected_phase) < 0.1)
 
 
