def test_mesh_slices():
# This test is temporarily disabled because it is seg faulty,
# see #1394
return
# Perform I/O in safe place instead of yt main dir
tmpdir = tempfile.mkdtemp()
curdir = os.getcwd()
os.chdir(tmpdir)
np.random.seed(0x4d3d3d3)
# tetrahedral ds
ds = fake_tetrahedral_ds()
for field in ds.field_list:
for idir in [0, 1, 2]:
sl = yt.SlicePlot(ds, idir, field)
sl.annotate_mesh_lines()
sl.save()
# hexahedral ds
ds = fake_hexahedral_ds()
for field in ds.field_list:
for idir in [0, 1, 2]:
sl = yt.SlicePlot(ds, idir, field)
sl.annotate_mesh_lines()
sl.save()
os.chdir(curdir)
# clean up
shutil.rmtree(tmpdir)
def test_mesh_slices_tetrahedral():
ds = fake_tetrahedral_ds()
for field in ds.field_list:
for idir in [0, 1, 2]:
prefix = "%s_%s_%s" % (field[0], field[1], idir)
yield compare(ds,
field,
idir,
test_prefix=prefix,
test_name="mesh_slices_tetrahedral",
annotate=True)
def test_mesh_lines_callback():
with _cleanup_fname() as prefix:
ds = fake_hexahedral_ds()
for field in ds.field_list:
sl = SlicePlot(ds, 1, field)
sl.annotate_mesh_lines(plot_args={'color': 'black'})
assert_fname(sl.save(prefix)[0])
ds = fake_tetrahedral_ds()
for field in ds.field_list:
sl = SlicePlot(ds, 1, field)
sl.annotate_mesh_lines(plot_args={'color': 'black'})
assert_fname(sl.save(prefix)[0])
def test_mesh_lines_callback():
with _cleanup_fname() as prefix:
ds = fake_hexahedral_ds()
for field in ds.field_list:
sl = SlicePlot(ds, 1, field)
sl.annotate_mesh_lines(color="black")
assert_fname(sl.save(prefix)[0])
ds = fake_tetrahedral_ds()
for field in ds.field_list:
sl = SlicePlot(ds, 1, field)
sl.annotate_mesh_lines(color="black")
assert_fname(sl.save(prefix)[0])
check_axis_manipulation(sl, prefix) # only test the final field
def test_mesh_slices_tetrahedral():
ds = fake_tetrahedral_ds()
mesh = ds.index.meshes[0]
ad = ds.all_data()
for field in ds.field_list:
for idir in [0, 1, 2]:
prefix = "%s_%s_%s" % (field[0], field[1], idir)
yield compare(ds,
field,
idir,
test_prefix=prefix,
test_name="mesh_slices_tetrahedral",
annotate=True)
sl_obj = ds.slice(idir, ds.domain_center[idir])
assert sl_obj[field].shape[0] == mesh.count(sl_obj.selector)
assert sl_obj[field].shape[0] < ad[field].shape[0]
def surface_mesh_render():
images = []
ds = fake_tetrahedral_ds()
for field in ds.field_list:
sc = Scene()
sc.add_source(MeshSource(ds, field))
sc.add_camera()
im = sc.render()
images.append(im)
ds = fake_hexahedral_ds()
for field in ds.field_list:
sc = Scene()
sc.add_source(MeshSource(ds, field))
sc.add_camera()
im = sc.render()
images.append(im)
return images
