def test_particle_plot_wf(self):
test_ds = fake_particle_ds()
for dim in range(3):
for weight_field in WEIGHT_FIELDS:
pplot_wf = ParticleProjectionPlot(
test_ds, dim, "particle_mass", weight_field=weight_field)
pplot_wf.save()
def test_particle_plot_c(self):
test_ds = fake_particle_ds()
for center in CENTER_SPECS:
for dim in range(3):
pplot_c = ParticleProjectionPlot(
test_ds, dim, "particle_mass", center=center)
pplot_c.save()
def test_particle_plot_ds(self):
test_ds = fake_particle_ds()
ds_region = test_ds.region([0.5] * 3, [0.4] * 3, [0.6] * 3)
for dim in range(3):
pplot_ds = ParticleProjectionPlot(
test_ds, dim, "particle_mass", data_source=ds_region)
pplot_ds.save()
def test_particle_plot_c(self):
test_ds = fake_particle_ds()
for center in CENTER_SPECS:
for dim in range(3):
pplot_c = ParticleProjectionPlot(test_ds,
dim, ("all", "particle_mass"),
center=center)
with mock.patch(
"yt.visualization._mpl_imports.FigureCanvasAgg.print_figure"
):
pplot_c.save()
def test_particle_plot_ds(self):
test_ds = fake_particle_ds()
ds_region = test_ds.region([0.5] * 3, [0.4] * 3, [0.6] * 3)
for dim in range(3):
pplot_ds = ParticleProjectionPlot(test_ds,
dim, ("all", "particle_mass"),
data_source=ds_region)
with mock.patch(
"yt.visualization._mpl_imports.FigureCanvasAgg.print_figure"
):
pplot_ds.save()
def test_particle_plot_wf(self):
test_ds = fake_particle_ds()
for dim in range(3):
for weight_field in WEIGHT_FIELDS:
pplot_wf = ParticleProjectionPlot(
test_ds, dim, "particle_mass", weight_field=weight_field
)
with mock.patch(
"yt.visualization._mpl_imports.FigureCanvasAgg.print_figure"
):
pplot_wf.save()
def test_particle_plot(self):
test_ds = fake_particle_ds()
for dim in range(3):
pplot = ParticleProjectionPlot(test_ds, dim, "particle_mass")
with mock.patch(
"yt.visualization._mpl_imports.FigureCanvasAgg.print_figure"
), mock.patch(
"yt.visualization._mpl_imports.FigureCanvasPdf.print_figure"
), mock.patch(
"yt.visualization._mpl_imports.FigureCanvasPS.print_figure"
):
for fname in TEST_FLNMS:
pplot.save(fname)[0]
def setUpClass(cls):
test_ds = fake_particle_ds()
ds_region = test_ds.region([0.5] * 3, [0.4] * 3, [0.6] * 3)
pplots = []
pplots_ds = []
pplots_c = []
pplots_wf = []
pplots_w = {}
pplots_m = []
for dim in range(3):
pplots.append(ParticleProjectionPlot(test_ds, dim,
"particle_mass"))
pplots_ds.append(
ParticleProjectionPlot(test_ds,
dim,
"particle_mass",
data_source=ds_region))
for center in CENTER_SPECS:
pplots_c.append(
ParticleProjectionPlot(test_ds,
dim,
"particle_mass",
center=center))
for width in WIDTH_SPECS:
pplots_w[width] = ParticleProjectionPlot(test_ds,
dim,
'particle_mass',
width=width)
for wf in WEIGHT_FIELDS:
pplots_wf.append(
ParticleProjectionPlot(test_ds,
dim,
"particle_mass",
weight_field=wf))
cls.pplots = pplots
cls.pplots_ds = pplots_ds
cls.pplots_c = pplots_c
cls.pplots_wf = pplots_wf
cls.pplots_w = pplots_w
cls.pplots_m = pplots_m
def test_creation_with_width(self):
test_ds = fake_particle_ds()
for width, (xlim, ylim, pwidth, _aun) in WIDTH_SPECS.items():
plot = ParticleProjectionPlot(test_ds, 0, "particle_mass", width=width)
xlim = [plot.ds.quan(el[0], el[1]) for el in xlim]
ylim = [plot.ds.quan(el[0], el[1]) for el in ylim]
pwidth = [plot.ds.quan(el[0], el[1]) for el in pwidth]
[assert_array_almost_equal(px, x, 14) for px, x in zip(plot.xlim, xlim)]
[assert_array_almost_equal(py, y, 14) for py, y in zip(plot.ylim, ylim)]
[assert_array_almost_equal(pw, w, 14) for pw, w in zip(plot.width, pwidth)]
def test_particle_plot(self):
test_ds = fake_particle_ds()
particle_projs = []
for dim in range(3):
particle_projs += [
ParticleProjectionPlot(test_ds, dim, ("all", "particle_mass")),
ParticleProjectionPlot(test_ds,
dim, ("all", "particle_mass"),
deposition="cic"),
ParticleProjectionPlot(test_ds,
dim, ("all", "particle_mass"),
density=True),
]
particle_projs[0]._repr_html_()
with mock.patch(
"yt.visualization._mpl_imports.FigureCanvasAgg.print_figure"
), mock.patch(
"yt.visualization._mpl_imports.FigureCanvasPdf.print_figure"
), mock.patch(
"yt.visualization._mpl_imports.FigureCanvasPS.print_figure"):
for p in particle_projs:
for fname in TEST_FLNMS:
p.save(fname)[0]
def test_particle_plot(self):
test_ds = fake_particle_ds()
for dim in range(3):
pplot = ParticleProjectionPlot(test_ds, dim, "particle_mass")
for fname in TEST_FLNMS:
assert_fname(pplot.save(fname)[0])
