1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
import pathlib
import numpy as np
import pytest
import meshio
from . import helpers
test_set = [
# helpers.empty_mesh,
helpers.line_mesh,
helpers.tri_mesh,
helpers.tri_mesh_one_cell,
helpers.triangle6_mesh,
helpers.quad_mesh,
helpers.quad8_mesh,
helpers.tri_quad_mesh,
helpers.polygon_mesh,
helpers.polygon_mesh_one_cell,
helpers.polygon2_mesh,
helpers.tet_mesh,
helpers.tet10_mesh,
helpers.hex_mesh,
helpers.hex20_mesh,
helpers.pyramid_mesh,
helpers.wedge_mesh,
helpers.polyhedron_mesh,
helpers.lagrange_high_order_mesh,
helpers.add_point_data(helpers.tri_mesh, 1),
helpers.add_point_data(helpers.tri_mesh, 2),
helpers.add_point_data(helpers.tri_mesh, 3),
helpers.add_cell_data(helpers.tri_mesh, [("a", (), np.float64)]),
helpers.add_cell_data(helpers.tri_quad_mesh, [("a", (), np.float64)]),
helpers.add_cell_data(helpers.tri_mesh, [("a", (2,), np.float32)]),
helpers.add_cell_data(helpers.tri_mesh, [("b", (3,), np.float64)]),
helpers.add_cell_data(helpers.polygon_mesh, [("a", (), np.float32)]),
helpers.add_cell_data(helpers.polyhedron_mesh, [("a", (2,), np.float32)]),
]
@pytest.mark.parametrize("mesh", test_set)
@pytest.mark.parametrize(
"data_type", [(False, None), (True, None), (True, "lzma"), (True, "zlib")]
)
def test(mesh, data_type, tmp_path):
binary, compression = data_type
def writer(*args, **kwargs):
return meshio.vtu.write(*args, binary=binary, compression=compression, **kwargs)
# ASCII files are only meant for debugging, VTK stores only 11 digits
# <https://gitlab.kitware.com/vtk/vtk/-/issues/17038#note_264052>
tol = 1.0e-15 if binary else 1.0e-10
helpers.write_read(tmp_path, writer, meshio.vtu.read, mesh, tol)
def test_generic_io(tmp_path):
helpers.generic_io(tmp_path / "test.vtu")
# With additional, insignificant suffix:
helpers.generic_io(tmp_path / "test.0.vtu")
@pytest.mark.parametrize(
"filename, ref_cells, ref_num_cells, ref_num_pnt",
[
("00_raw_binary.vtu", "tetra", 162, 64),
("01_raw_binary_int64.vtu", "tetra", 162, 64),
("02_raw_compressed.vtu", "tetra", 162, 64),
],
)
def test_read_from_file(filename, ref_cells, ref_num_cells, ref_num_pnt):
this_dir = pathlib.Path(__file__).resolve().parent
filename = this_dir / "meshes" / "vtu" / filename
mesh = meshio.read(filename)
assert len(mesh.cells) == 1
assert ref_cells == mesh.cells[0].type
assert len(mesh.cells[0].data) == ref_num_cells
assert len(mesh.points) == ref_num_pnt
|
