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 94 95 96
|
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)
ref = np.array([10.0, 10.0 / 3.0, 5.0 / 3.0])
print(mesh.ce_ratios)
assert ref.shape == mesh.ce_ratios.shape
assert np.all(np.abs(mesh.ce_ratios - ref) < 1.0e-14 * np.abs(ref))
def test_tri_simple():
X = np.array(
[
[0.0, 0.0],
[1.0, 0.0],
[1.0, 1.0],
]
)
cells = np.array([[0, 1, 2]])
mesh = meshplex.Mesh(X, cells)
ref = np.array([[0.5], [0.0], [0.5]])
print(mesh.ce_ratios)
assert ref.shape == mesh.ce_ratios.shape
assert np.all(np.abs(mesh.ce_ratios - ref) < 1.0e-14 * np.abs(ref) + 1.0e-14)
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.5, 0.5], [0.0, 0.5], [0.5, 0.0]])
print(mesh.ce_ratios)
assert ref.shape == mesh.ce_ratios.shape
assert np.all(np.abs(mesh.ce_ratios - 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(
[
[
[-1 / 24, -1 / 24],
[1 / 8, 1 / 8],
[1 / 8, 1 / 8],
[0.0, 0.0],
],
[
[-1 / 24, -1 / 24],
[1 / 8, 1 / 8],
[0.0, 0.0],
[1 / 8, 1 / 8],
],
[
[-1 / 24, -1 / 24],
[0.0, 0.0],
[1 / 8, 1 / 8],
[1 / 8, 1 / 8],
],
],
)
print(mesh.ce_ratios)
assert ref.shape == mesh.ce_ratios.shape
assert np.all(np.abs(mesh.ce_ratios - ref) < 1.0e-14 * np.abs(ref) + 1.0e-14)
if __name__ == "__main__":
test_tri_simple()
|