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 numpy as np
import meshio
from . import helpers
def is_same_mesh(mesh0, mesh1, atol):
if not np.allclose(mesh0.points, mesh1.points, atol=atol, rtol=0.0):
return False
for cells0, cells1 in zip(mesh0.cells, mesh1.cells):
if cells0.type != cells1.type or not np.allclose(cells0.data, cells1.data):
return False
return True
def test_info(tmp_path):
infile = tmp_path / "out.msh"
meshio.write(infile, helpers.tri_mesh, file_format="gmsh")
meshio._cli.main(["info", str(infile), "--input-format", "gmsh"])
def test_convert(tmp_path):
input_mesh = helpers.tri_mesh
infile = tmp_path / "in.msh"
meshio.write(infile, helpers.tri_mesh, file_format="gmsh")
outfile = tmp_path / "out.msh"
meshio._cli.main(
[
"convert",
str(infile),
str(outfile),
"--input-format",
"gmsh",
"--output-format",
"vtk",
"--sets-to-int-data",
]
)
mesh = meshio.read(outfile, file_format="vtk")
atol = 1.0e-15
assert np.allclose(input_mesh.points, mesh.points, atol=atol, rtol=0.0)
for cells0, cells1 in zip(input_mesh.cells, mesh.cells):
assert cells0.type == cells1.type
assert np.allclose(cells0.data, cells1.data)
def test_compress(tmp_path):
input_mesh = helpers.tri_mesh
infile = tmp_path / "in.vtu"
meshio.write(infile, input_mesh)
meshio._cli.main(["decompress", str(infile)])
mesh = meshio.read(infile)
assert is_same_mesh(input_mesh, mesh, atol=1.0e-15)
meshio._cli.main(["compress", str(infile)])
mesh = meshio.read(infile)
assert is_same_mesh(input_mesh, mesh, atol=1.0e-15)
def test_ascii_binary(tmp_path):
input_mesh = helpers.tri_mesh
infile = tmp_path / "in.vtu"
meshio.write(infile, input_mesh)
meshio._cli.main(["ascii", str(infile)])
mesh = meshio.read(infile)
assert is_same_mesh(input_mesh, mesh, atol=1.0e-12)
meshio._cli.main(["binary", str(infile)])
mesh = meshio.read(infile)
assert is_same_mesh(input_mesh, mesh, atol=1.0e-12)
|