def test_line_integral_convolution_callback(): with _cleanup_fname() as prefix: ds = fake_amr_ds(fields = ("density", "velocity_x", "velocity_y", "velocity_z")) for ax in 'xyz': p = ProjectionPlot(ds, ax, "density") p.annotate_line_integral_convolution("velocity_x", "velocity_y") assert_fname(p.save(prefix)[0]) p = ProjectionPlot(ds, ax, "density", weight_field="density") p.annotate_line_integral_convolution("velocity_x", "velocity_y") assert_fname(p.save(prefix)[0]) p = SlicePlot(ds, ax, "density") p.annotate_line_integral_convolution("velocity_x", "velocity_y") assert_fname(p.save(prefix)[0]) # Now we'll check a few additional minor things p = SlicePlot(ds, "x", "density") p.annotate_line_integral_convolution("velocity_x", "velocity_y", kernellen=100., lim=(0.4,0.7), cmap=ytcfg.get("yt", "default_colormap"), alpha=0.9, const_alpha=True) p.save(prefix) with _cleanup_fname() as prefix: ds = load(cyl_2d) slc = SlicePlot(ds, "theta", "density") slc.annotate_line_integral_convolution("magnetic_field_r", "magnetic_field_z") assert_fname(slc.save(prefix)[0]) with _cleanup_fname() as prefix: ds = fake_amr_ds(fields = ("density", "velocity_r", "velocity_theta", "velocity_phi"), geometry="spherical") p = SlicePlot(ds, "r", "density") p.annotate_line_integral_convolution("velocity_theta", "velocity_phi") assert_raises(YTDataTypeUnsupported, p.save, prefix)
def test_line_integral_convolution_callback(): with _cleanup_fname() as prefix: ds = fake_amr_ds( fields=("density", "velocity_x", "velocity_y", "velocity_z"), units=("g/cm**3", "cm/s", "cm/s", "cm/s"), ) for ax in "xyz": p = ProjectionPlot(ds, ax, ("gas", "density")) p.annotate_line_integral_convolution( ("gas", "velocity_x"), ("gas", "velocity_y") ) assert_fname(p.save(prefix)[0]) p = ProjectionPlot( ds, ax, ("gas", "density"), weight_field=("gas", "density") ) p.annotate_line_integral_convolution( ("gas", "velocity_x"), ("gas", "velocity_y") ) assert_fname(p.save(prefix)[0]) p = SlicePlot(ds, ax, ("gas", "density")) p.annotate_line_integral_convolution( ("gas", "velocity_x"), ("gas", "velocity_y") ) assert_fname(p.save(prefix)[0]) # Now we'll check a few additional minor things p = SlicePlot(ds, "x", ("gas", "density")) p.annotate_line_integral_convolution( ("gas", "velocity_x"), ("gas", "velocity_y"), kernellen=100.0, lim=(0.4, 0.7), cmap=ytcfg.get("yt", "default_colormap"), alpha=0.9, const_alpha=True, ) p.save(prefix) with _cleanup_fname() as prefix: ds = load(cyl_2d) slc = SlicePlot(ds, "theta", ("gas", "magnetic_field_strength")) slc.annotate_line_integral_convolution( ("gas", "magnetic_field_r"), ("gas", "magnetic_field_z") ) assert_fname(slc.save(prefix)[0]) with _cleanup_fname() as prefix: ds = load(cyl_3d) slc = SlicePlot(ds, "r", ("gas", "magnetic_field_strength")) slc.annotate_line_integral_convolution( ("gas", "magnetic_field_theta"), ("gas", "magnetic_field_z") ) assert_fname(slc.save(prefix)[0]) slc = SlicePlot(ds, "z", ("gas", "magnetic_field_strength")) slc.annotate_line_integral_convolution( ("gas", "magnetic_field_cartesian_x"), ("gas", "magnetic_field_cartesian_y") ) assert_fname(slc.save(prefix)[0]) slc = SlicePlot(ds, "theta", ("gas", "magnetic_field_strength")) slc.annotate_line_integral_convolution( ("gas", "magnetic_field_r"), ("gas", "magnetic_field_z") ) assert_fname(slc.save(prefix)[0]) check_axis_manipulation(slc, prefix) with _cleanup_fname() as prefix: ds = fake_amr_ds( fields=("density", "velocity_r", "velocity_theta", "velocity_phi"), units=("g/cm**3", "cm/s", "cm/s", "cm/s"), geometry="spherical", ) p = SlicePlot(ds, "r", ("gas", "density")) assert_raises( YTDataTypeUnsupported, p.annotate_line_integral_convolution, ("gas", "velocity_theta"), ("gas", "velocity_phi"), )