def test_contour_callback(): with _cleanup_fname() as prefix: ds = fake_amr_ds(fields = ("density", "temperature")) for ax in 'xyz': p = ProjectionPlot(ds, ax, "density") p.annotate_contour("temperature") assert_fname(p.save(prefix)[0]) p = ProjectionPlot(ds, ax, "density", weight_field="density") p.annotate_contour("temperature") assert_fname(p.save(prefix)[0]) p = SlicePlot(ds, ax, "density") p.annotate_contour("temperature") # BREAKS WITH ndarray assert_fname(p.save(prefix)[0]) # Now we'll check a few additional minor things p = SlicePlot(ds, "x", "density") p.annotate_contour("temperature", ncont=10, factor=8, take_log=False, clim=(0.4, 0.6), plot_args={'linewidths':2.0}, label=True, text_args={'text-size':'x-large'}) p.save(prefix) p = SlicePlot(ds, "x", "density") s2 = ds.slice(0, 0.2) p.annotate_contour("temperature", ncont=10, factor=8, take_log=False, clim=(0.4, 0.6), plot_args={'linewidths':2.0}, label=True, text_args={'text-size':'x-large'}, data_source=s2) p.save(prefix) with _cleanup_fname() as prefix: ds = load(cyl_2d) slc = SlicePlot(ds, "theta", "plasma_beta") slc.annotate_contour("plasma_beta", ncont=2, factor=7., take_log=False, clim=(1.e-1,1.e1), label=True, plot_args={"colors": ("c","w"), "linewidths": 1}, text_args={"fmt": "%1.1f"}) assert_fname(slc.save(prefix)[0]) with _cleanup_fname() as prefix: ds = fake_amr_ds(fields = ("density", "temperature"), geometry="spherical") p = SlicePlot(ds, "r", "density") p.annotate_contour("temperature", ncont=10, factor=8, take_log=False, clim=(0.4, 0.6), plot_args={'linewidths':2.0}, label=True, text_args={'text-size':'x-large'}) assert_raises(YTDataTypeUnsupported, p.save, prefix)
def test_contour_callback(): with _cleanup_fname() as prefix: ds = fake_amr_ds(fields=("density", "temperature"), units=("g/cm**3", "K")) for ax in "xyz": p = ProjectionPlot(ds, ax, ("gas", "density")) p.annotate_contour(("gas", "temperature")) assert_fname(p.save(prefix)[0]) p = ProjectionPlot( ds, ax, ("gas", "density"), weight_field=("gas", "density") ) p.annotate_contour(("gas", "temperature")) assert_fname(p.save(prefix)[0]) p = SlicePlot(ds, ax, ("gas", "density")) p.annotate_contour(("gas", "temperature")) # BREAKS WITH ndarray assert_fname(p.save(prefix)[0]) # Now we'll check a few additional minor things p = SlicePlot(ds, "x", ("gas", "density")) p.annotate_contour( ("gas", "temperature"), levels=10, factor=8, take_log=False, clim=(0.4, 0.6), plot_args={"linewidths": 2.0}, label=True, text_args={"fontsize": "x-large"}, ) p.save(prefix) p = SlicePlot(ds, "x", ("gas", "density")) s2 = ds.slice(0, 0.2) p.annotate_contour( ("gas", "temperature"), levels=10, factor=8, take_log=False, clim=(0.4, 0.6), plot_args={"linewidths": 2.0}, label=True, text_args={"fontsize": "x-large"}, data_source=s2, ) p.save(prefix) with _cleanup_fname() as prefix: ds = load(cyl_2d) slc = SlicePlot(ds, "theta", ("gas", "plasma_beta")) slc.annotate_contour( ("gas", "plasma_beta"), levels=2, factor=7, take_log=False, clim=(1.0e-1, 1.0e1), label=True, plot_args={"colors": ("c", "w"), "linewidths": 1}, text_args={"fmt": "%1.1f"}, ) assert_fname(slc.save(prefix)[0]) check_axis_manipulation(slc, prefix) with _cleanup_fname() as prefix: ds = fake_amr_ds( fields=("density", "temperature"), units=("g/cm**3", "K"), geometry="spherical", ) p = SlicePlot(ds, "r", ("gas", "density")) kwargs = dict( levels=10, factor=8, take_log=False, clim=(0.4, 0.6), plot_args={"linewidths": 2.0}, label=True, text_args={"fontsize": "x-large"}, ) assert_raises( YTDataTypeUnsupported, p.annotate_contour, ("gas", "temperature"), **kwargs )