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
|
import numpy as np
import meshplex
def test_line():
X = np.array([0.0, 0.1, 0.4, 1.0])
cells = np.array([[0, 1], [1, 2], [2, 3]])
mesh = meshplex.Mesh(X, cells)
vol = np.sum(mesh.cell_volumes)
assert np.all(np.abs(np.sum(mesh.control_volumes) - vol) < 1.0e-14 * np.abs(vol))
ref = np.array([0.05, 0.2, 0.45, 0.3])
assert np.all(np.abs(mesh.control_volumes - ref) < 1.0e-14 * np.abs(ref))
def test_tri_degen():
X = np.array(
[
[0.0, 0.0],
[1.0, 0.0],
[0.5, 1.0e-2],
]
)
cells = np.array([[0, 1, 2]])
mesh = meshplex.Mesh(X, cells)
vol = np.sum(mesh.cell_volumes)
assert np.all(np.abs(np.sum(mesh.control_volumes) - vol) < 1.0e-12 * np.abs(vol))
ref = np.array([-1.560625, -1.560625, 3.12625])
print(mesh.control_volumes)
assert np.all(np.abs(mesh.control_volumes - ref) < 1.0e-14 * np.abs(ref))
def test_tri():
X = np.array(
[
[0.0, 0.0],
[1.0, 0.0],
[1.0, 1.0],
[0.0, 1.0],
]
)
cells = np.array([[0, 1, 2], [0, 2, 3]])
mesh = meshplex.Mesh(X, cells)
ref = np.array([0.25, 0.25, 0.25, 0.25])
assert np.all(np.abs(mesh.control_volumes - ref) < 1.0e-14 * np.abs(ref))
def test_tetra_simple():
X = np.array(
[
[0.0, 0.0, 0.0],
[1.0, 0.0, 0.0],
[0.0, 1.0, 0.0],
[0.0, 0.0, 1.0],
]
)
cells = np.array([[0, 1, 2, 3]])
mesh = meshplex.Mesh(X, cells)
ref = np.array([9.0, 1.0, 1.0, 1.0]) / 72
mesh.control_volumes
print()
print(mesh.control_volumes)
print(sum(mesh.control_volumes))
print(ref)
print(sum(ref))
assert np.all(np.abs(mesh.control_volumes - ref) < 1.0e-14 * np.abs(ref) + 1.0e-14)
def test_tetra():
X = np.array(
[
[0.0, 0.0, 0.0],
[1.0, 0.0, 0.0],
[0.0, 1.0, 0.0],
[0.0, 0.0, -1.0],
[0.0, 0.0, 1.0],
]
)
cells = np.array([[0, 1, 2, 3], [0, 1, 2, 4]])
mesh = meshplex.Mesh(X, cells)
ref = np.array([18.0, 2.0, 2.0, 1.0, 1.0]) / 72
print(mesh.control_volumes)
print(ref)
assert np.all(np.abs(mesh.control_volumes - ref) < 1.0e-14 * np.abs(ref) + 1.0e-14)
|