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
|
import pathlib
import numpy as np
import pytest
import meshio
from . import helpers
test_set = [
# helpers.empty_mesh,
helpers.tri_mesh_2d,
helpers.tet_mesh,
helpers.hex_mesh,
]
this_dir = pathlib.Path(__file__).resolve().parent
@pytest.mark.parametrize("mesh", test_set)
def test(mesh, tmp_path):
helpers.write_read(tmp_path, meshio.su2.write, meshio.su2.read, mesh, 1.0e-15)
@pytest.mark.parametrize(
"filename, ref_num_cells, ref_num_points,ref_num_unique_tags,sum_tags",
[("square.su2", 16, 9, 4, 20), ("mixgrid.su2", 30, 16, 6, 62)],
)
def test_structured(
filename, ref_num_cells, ref_num_points, ref_num_unique_tags, sum_tags
):
filename = this_dir / "meshes" / "su2" / filename
mesh = meshio.read(filename)
assert sum(len(block.data) for block in mesh.cells) == ref_num_cells
assert len(mesh.points) == ref_num_points
all_tags = np.concatenate([tags for tags in mesh.cell_data["su2:tag"]])
assert sum(all_tags) == sum_tags
all_unique_tags = np.unique(all_tags)
assert len(all_unique_tags) == ref_num_unique_tags + 1
|
