fixed phaseplot test to use system which will always have separatrices

lkies · lkies · commit 4724e5ad4591 · 2025-02-09T23:11:47.000+01:00
diff --git a/control/tests/phaseplot_test.py b/control/tests/phaseplot_test.py
@@ -170,7 +170,8 @@ def invpend_ode(t, x, m=0, l=0, b=0, g=0):
                             plot_streamlines=True)
         
     with pytest.raises(ValueError, match="gridtype must be 'meshgrid' when using streamplot"):
-        ct.phase_plane_plot(ct.rss(2, 1, 1), plot_streamplot=True, gridtype='boxgrid')
+        ct.phase_plane_plot(ct.rss(2, 1, 1), plot_streamlines=False,
+                            plot_streamplot=True, gridtype='boxgrid')
 
     # Warning messages for invalid solutions: nonlinear spring mass system
     sys = ct.nlsys(
@@ -220,7 +221,8 @@ def assert_orders(streamlines, quiver, streamplot, separatrices, equilpoints):
         if streamplot is not None:
             assert streamplot < separatrices < equilpoints
 
-    sys = ct.rss(2, 1, 1)
+    def sys(t, x):
+        return np.array([4*x[1], -np.sin(4*x[0])])
 
     # ensure correct zordering for all three flow types
     res_streamlines = ct.phase_plane_plot(sys, plot_streamlines=dict(color=key_color))
@@ -239,7 +241,9 @@ def assert_orders(streamlines, quiver, streamplot, separatrices, equilpoints):
 
 @pytest.mark.usefixtures('mplcleanup')
 def test_stream_plot_magnitude():
-    sys = ct.rss(2, 1, 1)
+    def sys(t, x):
+        return np.array([4*x[1], -np.sin(4*x[0])])
+
     # plt context with linewidth
     with plt.rc_context({'lines.linewidth': 4}):
         res = ct.phase_plane_plot(sys, plot_streamplot=dict(vary_linewidth=True))
