clear figure in unit tests to avoid slow plotting problem

python-control · Jan 2, 2021 · 2702ee7 · 2702ee7
1 parent fe9928b
commit 2702ee7
Showing 1 changed file with 2 additions and 3 deletions.
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)