def run_test_2d(f, main__file__, show=False):
mpl.clf()
slc = "x=-20f:12f, y=0f"
plot_kwargs = dict(title=True, earth=True)
mpl.subplot(141)
mpl.plot(f['pp'], slc, logscale=True, **plot_kwargs)
mpl.plot(np.abs(f['psi']), style='contour', logscale=True, levels=30,
linewidths=0.8, colors='grey', linestyles='solid', cbar=None,
x=(-20, 12))
mpl.subplot(142)
mpl.plot(viscid.magnitude(f['bcc']), slc, logscale=True, **plot_kwargs)
mpl.plot2d_quiver(f['v'][slc], step=5, color='y', pivot='mid', width=0.03,
scale=600)
mpl.subplot(143)
mpl.plot(f['jy'], slc, clim=[-0.005, 0.005], **plot_kwargs)
mpl.streamplot(f['v'][slc], linewidth=0.3)
mpl.subplot(144)
mpl.plot(f['jy'], "x=7f:12f, y=0f, z=0f")
mpl.plt.suptitle("2D File")
mpl.auto_adjust_subplots(subplot_params=dict(top=0.9, wspace=1.3))
mpl.plt.gcf().set_size_inches(10, 4)
mpl.savefig(next_plot_fname(main__file__))
if show:
mpl.show()
def run_test_3d(f, main__file__, show=False):
mpl.clf()
slc = "x=-20f:12f, y=0f"
plot_kwargs = dict(title=True, earth=True)
mpl.subplot(141)
mpl.plot(f['pp'], slc, logscale=True, **plot_kwargs)
mpl.subplot(142)
mpl.plot(viscid.magnitude(f['bcc']), slc, logscale=True, **plot_kwargs)
mpl.plot2d_quiver(f['v'][slc], step=5, color='y', pivot='mid', width=0.03,
scale=600)
mpl.subplot(143)
mpl.plot(f['jy'], slc, clim=(-0.005, 0.005), **plot_kwargs)
mpl.streamplot(f['v'][slc], linewidth=0.3)
mpl.subplot(144)
mpl.plot(f['jy'], "x=7f:12f, y=0f, z=0f")
mpl.plt.suptitle("3D File")
mpl.auto_adjust_subplots(subplot_params=dict(top=0.9, wspace=1.3))
mpl.plt.gcf().set_size_inches(10, 4)
mpl.savefig(next_plot_fname(main__file__))
if show:
mpl.show()