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
|
import pathlib
import meshplex
this_dir = pathlib.Path(__file__).resolve().parent
def test_get_edges():
mesh = meshplex.read(this_dir / "meshes" / "pacman.vtu")
mesh.create_facets()
edge_mask = mesh.get_edge_mask()
edge_points = mesh.edges["points"][edge_mask]
assert len(edge_points) == 2372
def test_mark_subdomain2d():
mesh = meshplex.read(this_dir / "meshes" / "pacman.vtu")
class Subdomain1:
is_boundary_only = True
# pylint: disable=no-self-use
def is_inside(self, x):
return x[0] < 0.0
class Subdomain2:
is_boundary_only = False
# pylint: disable=no-self-use
def is_inside(self, x):
return x[0] > 0.0
sd1 = Subdomain1()
vertex_mask = mesh.get_vertex_mask(sd1)
assert vertex_mask.sum() == 45
face_mask = mesh.get_face_mask(sd1)
assert face_mask.sum() == 44
cell_mask = mesh.get_cell_mask(sd1)
assert cell_mask.sum() == 0
sd2 = Subdomain2()
vertex_mask = mesh.get_vertex_mask(sd2)
assert vertex_mask.sum() == 395
face_mask = mesh.get_face_mask(sd2)
assert face_mask.sum() == 2148
cell_mask = mesh.get_cell_mask(sd2)
assert cell_mask.sum() == 706
def test_mark_subdomain3d():
mesh = meshplex.read(this_dir / "meshes" / "tetrahedron.vtk")
class Subdomain1:
is_boundary_only = True
# pylint: disable=no-self-use
def is_inside(self, x):
return x[0] < 0.5
class Subdomain2:
is_boundary_only = False
# pylint: disable=no-self-use
def is_inside(self, x):
return x[0] > 0.5
sd1 = Subdomain1()
vertex_mask = mesh.get_vertex_mask(sd1)
assert vertex_mask.sum() == 16
face_mask = mesh.get_face_mask(sd1)
assert face_mask.sum() == 20
cell_mask = mesh.get_cell_mask(sd1)
assert cell_mask.sum() == 0
sd2 = Subdomain2()
vertex_mask = mesh.get_vertex_mask(sd2)
assert vertex_mask.sum() == 10
face_mask = mesh.get_face_mask(sd2)
assert face_mask.sum() == 25
cell_mask = mesh.get_cell_mask(sd2)
assert cell_mask.sum() == 5
|
