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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
import pathlib
from functools import partial
import helpers
import numpy as np
import pytest
import platform
import meshio
test_set = [
helpers.empty_mesh,
helpers.line_mesh,
helpers.tri_mesh_2d,
helpers.tri_mesh,
helpers.triangle6_mesh,
helpers.quad_mesh,
helpers.quad8_mesh,
helpers.tri_quad_mesh,
helpers.tet_mesh,
helpers.tet10_mesh,
helpers.hex_mesh,
helpers.hex20_mesh,
helpers.polygon_mesh,
helpers.pyramid_mesh,
helpers.wedge_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),
# VTK files float data is always stored in big endian
helpers.add_cell_data(helpers.tri_mesh, [("a", (), ">f8")]),
helpers.add_cell_data(helpers.tri_mesh, [("a", (2,), ">f8")]),
helpers.add_cell_data(helpers.tri_mesh, [("a", (3,), ">f8")]),
helpers.add_cell_data(
helpers.add_point_data(helpers.tri_mesh_2d, 2), [("a", (2,), ">f8")]
),
]
@pytest.mark.parametrize("mesh", test_set)
@pytest.mark.parametrize("binary", [True, False])
def test(mesh, binary):
def writer(*args, **kwargs):
return meshio.vtk.write(*args, binary=binary, **kwargs)
helpers.write_read(writer, meshio.vtk.read, mesh, 1.0e-15)
def test_generic_io():
helpers.generic_io("test.vtk")
# With additional, insignificant suffix:
helpers.generic_io("test.0.vtk")
@pytest.mark.parametrize(
"filename, ref_sum, ref_num_cells", [("rbc_001.vtk", 0.00031280518, 996)]
)
@pytest.mark.parametrize("binary", [False, True])
def test_reference_file(filename, ref_sum, ref_num_cells, binary):
this_dir = pathlib.Path(__file__).resolve().parent
filename = this_dir / "meshes" / "vtk" / filename
mesh = meshio.read(filename)
if platform.architecture()[0].startswith('32'):
# tolerance is too tight for i386, so relax a little for 32 bit systems
tol = 3.0e-2
else:
tol = 1.0e-2
s = np.sum(mesh.points)
assert abs(s - ref_sum) < tol * ref_sum
assert mesh.cells[0].type == "triangle"
assert len(mesh.cells[0].data) == ref_num_cells
writer = partial(meshio.vtk.write, binary=binary)
helpers.write_read(writer, meshio.vtk.read, mesh, 1.0e-15)
@pytest.mark.parametrize(
"filename, ref_cells, ref_num_cells, ref_num_pnt",
[
("00_image.vtk", "quad", 81, 100),
("01_image.vtk", "hexahedron", 72, 147),
("02_structured.vtk", "hexahedron", 72, 147),
("03_rectilinear.vtk", "hexahedron", 72, 147),
("04_rectilinear.vtk", "quad", 27, 40),
("05_rectilinear.vtk", "quad", 27, 40),
("06_unstructured.vtk", "hexahedron", 12, 42),
("gh-935.vtk", "triangle", 2, 6),
],
)
def test_structured(filename, ref_cells, ref_num_cells, ref_num_pnt):
this_dir = pathlib.Path(__file__).resolve().parent
filename = this_dir / "meshes" / "vtk" / 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
def test_pathlike():
this_dir = pathlib.Path(__file__).resolve().parent
meshio.read(this_dir / "meshes" / "vtk" / "rbc_001.vtk")
@pytest.mark.parametrize(
"filename, ref_num_points, ref_num_cells", [("06_color_scalars.vtk", 5, 2)]
)
def test_color_scalars(filename, ref_num_points, ref_num_cells):
this_dir = pathlib.Path(__file__).resolve().parent
filename = this_dir / "meshes" / "vtk" / filename
mesh = meshio.read(filename)
assert len(mesh.points) == ref_num_points
assert len(mesh.cells) == ref_num_cells
if __name__ == "__main__":
test(helpers.tri_mesh, binary=True)
|
