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