File: test_control_volumes.py

package info (click to toggle)
python-meshplex 0.17.1-5
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 668 kB
  • sloc: python: 3,626; makefile: 13
file content (93 lines) | stat: -rw-r--r-- 2,302 bytes parent folder | download | duplicates (2)
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)