@@ -167,7 +167,7 @@ def invpend_ode(t, x, m=0, l=0, b=0, g=0):
167167ct .phase_plane_plot (
168168invpend_ode , [- 5 ,5 ,2 ,2 ],params = {'stuff' : (1 ,1 ,0.2 ,1 )},
169169plot_streamlines = True )
170-
170+
171171with pytest .raises (ValueError ,match = "gridtype must be 'meshgrid' when using streamplot" ):
172172ct .phase_plane_plot (ct .rss (2 ,1 ,1 ),plot_streamlines = False ,
173173plot_streamplot = True ,gridtype = 'boxgrid' )
@@ -190,7 +190,7 @@ def invpend_ode(t, x, m=0, l=0, b=0, g=0):
190190sys , [- 12 ,12 ,- 10 ,10 ],15 ,gridspec = [2 ,9 ],
191191plot_streamlines = True ,plot_separatrices = False ,
192192suppress_warnings = True )
193-
193+
194194@pytest .mark .usefixtures ('mplcleanup' )
195195def test_phase_plot_zorder ():
196196# some of these tests are a bit akward since the streamlines and separatrices
@@ -211,7 +211,7 @@ def get_zorders(cplt):
211211assert cplt .lines [3 ]== None or isinstance (cplt .lines [3 ],mpl .streamplot .StreamplotSet )
212212streamplot = max (cplt .lines [3 ].lines .get_zorder (),cplt .lines [3 ].arrows .get_zorder ())if cplt .lines [3 ]else None
213213return streamlines ,quiver ,streamplot ,separatrices ,equilpoints
214-
214+
215215def assert_orders (streamlines ,quiver ,streamplot ,separatrices ,equilpoints ):
216216print (streamlines ,quiver ,streamplot ,separatrices ,equilpoints )
217217if streamlines is not None :
@@ -261,8 +261,6 @@ def sys(t, x):
261261# make sure changing the norm at least doesn't throw an error
262262ct .phase_plane_plot (sys ,plot_streamplot = dict (vary_color = True ,norm = mpl .colors .LogNorm ()))
263263
264-
265-
266264
267265@pytest .mark .usefixtures ('mplcleanup' )
268266def test_basic_phase_plots (savefigs = False ):