forked from nschloe/meshio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_vtk.py
72 lines (55 loc) · 1.96 KB
/
test_vtk.py
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
# -*- coding: utf-8 -*-
#
import pytest
import meshio
import helpers
import legacy_reader
import legacy_writer
vtk = pytest.importorskip('vtk')
test_set = [
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.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, 1),
helpers.add_cell_data(helpers.tri_mesh, 2),
helpers.add_cell_data(helpers.tri_mesh, 3),
]
@pytest.mark.parametrize('mesh', test_set)
@pytest.mark.parametrize('write_binary', [True, False])
def test(mesh, write_binary):
def writer(*args, **kwargs):
return meshio.vtk_io.write(*args, write_binary=write_binary, **kwargs)
helpers.write_read(writer, meshio.vtk_io.read, mesh, 1.0e-15)
return
@pytest.mark.parametrize('mesh', test_set)
@pytest.mark.parametrize('write_binary', [True, False])
def test_legacy_writer(mesh, write_binary):
# test with legacy writer
def lw(*args, **kwargs):
mode = 'vtk-binary' if write_binary else 'vtk-ascii'
return legacy_writer.write(mode, *args, **kwargs)
# The legacy writer only writes with low precision.
helpers.write_read(lw, meshio.vtk_io.read, mesh, 1.0e-11)
return
@pytest.mark.parametrize('mesh', test_set)
@pytest.mark.parametrize('write_binary', [True, False])
def test_legacy_reader(mesh, write_binary):
def writer(*args, **kwargs):
return meshio.vtk_io.write(*args, write_binary=write_binary, **kwargs)
# test with legacy reader
def lr(filename):
mode = 'vtk-binary' if write_binary else 'vtk-ascii'
return legacy_reader.read(mode, filename)
helpers.write_read(writer, lr, mesh, 1.0e-15)
return
if __name__ == '__main__':
test(helpers.tri_mesh, write_binary=True)