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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
# Copyright (C) 2013 Anders Logg
#
# This file is part of DOLFINx (https://www.fenicsproject.org)
#
# SPDX-License-Identifier: LGPL-3.0-or-later
from mpi4py import MPI
import numpy as np
import pytest
import basix
import ufl
from basix.ufl import element
from dolfinx.geometry import squared_distance
from dolfinx.mesh import create_mesh, create_unit_interval
@pytest.mark.skip_in_parallel
def test_distance_interval():
mesh = create_unit_interval(MPI.COMM_SELF, 1)
d = np.array([-1.0, 0.0, 0.0])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(1.0)
d = np.array([0.5, 0.0, 0.0])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(0.0)
@pytest.mark.skip_in_parallel
def test_distance_triangle():
shape, degree = "triangle", 1
domain = basix.create_element(basix.ElementFamily.P, basix.CellType[shape], degree)
x = np.array([[0.0, 0.0, 0.0], [0.0, 1.0, 0.0], [1.0, 1.0, 0.0]], dtype=np.float64)
cells = [[0, 1, 2]]
mesh = create_mesh(MPI.COMM_WORLD, cells, domain, x)
d = np.array([-1.0, -1.0, 0.0])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(2.0)
d = np.array([-1.0, 0.5, 0.0])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(1.0)
d = np.array([0.5, 0.5, 0.0])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(0.0)
@pytest.mark.skip_in_parallel
def test_distance_tetrahedron():
shape = "tetrahedron"
degree = 1
domain = ufl.Mesh(element("Lagrange", shape, degree, shape=(3,), dtype=np.float64))
x = np.array([[0.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 1.0, 1.0], [1, 1.0, 1]], dtype=np.float64)
cells = [[0, 1, 2, 3]]
mesh = create_mesh(MPI.COMM_WORLD, cells, domain, x)
d = np.array([-1.0, -1.0, -1.0])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(3.0)
d = np.array([-1.0, 0.5, 0.5])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(1.0)
d = np.array([0.5, 0.5, 0.5])
assert squared_distance(mesh, mesh.topology.dim, np.array([0]), d) == pytest.approx(0.0)
# @pytest.mark.skip("volume_entities needs fixing")
# @pytest.mark.parametrize(
# 'mesh', [
# create_unit_interval(MPI.COMM_WORLD, 18),
# create_unit_square(MPI.COMM_WORLD, 8, 9, CellType.triangle),
# create_unit_square(MPI.COMM_WORLD, 8, 9, CellType.quadrilateral),
# create_unit_cube(MPI.COMM_WORLD, 8, 9, 5, CellType.tetrahedron)
# ])
# def test_volume_cells(mesh):
# tdim = mesh.topology.dim
# map = mesh.topology.index_map(tdim)
# num_cells = map.size_local
# v = _cpp.mesh.volume_entities(mesh, range(num_cells), mesh.topology.dim)
# assert mesh.comm.allreduce(v.sum(), MPI.SUM) == pytest.approx(1.0, rel=1e-9)
# @pytest.mark.skip("volume_entities needs fixing")
# def test_volume_quadrilateralR2():
# mesh = create_unit_square(MPI.COMM_SELF, 1, 1, CellType.quadrilateral)
# assert _cpp.mesh.volume_entities(mesh, [0], mesh.topology.dim) == 1.0
# @pytest.mark.skip("volume_entities needs fixing")
# @pytest.mark.parametrize(
# 'x',
# [[[0.0, 0.0, 0.0], [0.0, 1.0, 0.0], [1.0, 0.0, 0.0], [1.0, 1.0, 0.0]],
# [[0.0, 0.0, 0.0], [0.0, 0.0, 1.0], [0.0, 1.0, 0.0], [0.0, 1.0, 1.0]]])
# def test_volume_quadrilateralR3(x):
# cells = [[0, 1, 2, 3]]
# domain = ufl.Mesh(element("Lagrange", "quadrilateral", 1, shape=(2,)))
# mesh = create_mesh(MPI.COMM_SELF, cells, domain, x)
# assert _cpp.mesh.volume_entities(mesh, [0], mesh.topology.dim) == 1.0
# @pytest.mark.skip("volume_entities needs fixing")
# @pytest.mark.parametrize(
# 'scaling',
# [1e0, 1e-5, 1e-10, 1e-15, 1e-20, 1e-30, 1e5, 1e10, 1e15, 1e20, 1e30])
# def test_volume_quadrilateral_coplanarity_check_1(scaling):
# with pytest.raises(RuntimeError) as error:
# # Unit square cell scaled down by 'scaling' and the first vertex
# # is distorted so that the vertices are clearly non coplanar
# x = [[scaling, 0.5 * scaling, 0.6 * scaling],
# [0.0, scaling, 0.0],
# [0.0, 0.0, scaling],
# [0.0, scaling, scaling]]
# cells = [[0, 1, 2, 3]]
# domain = ufl.Mesh(element("Lagrange", "quadrilateral", 1, shape=(2,)))
# mesh = create_mesh(MPI.COMM_SELF, cells, domain, x)
# _cpp.mesh.volume_entities(mesh, [0], mesh.topology.dim)
# assert "Not coplanar" in str(error.value)
# Test when |p0-p3| is ~ 1 but |p1-p2| is small
# The cell is degenerate when scale is below 1e-17, it is expected to
# fail the test.
# @pytest.mark.skip("volume_entities needs fixing")
# @pytest.mark.parametrize('scaling', [1e0, 1e-5, 1e-10, 1e-15])
# def test_volume_quadrilateral_coplanarity_check_2(scaling):
# with pytest.raises(RuntimeError) as error:
# # Unit square cell scaled down by 'scaling' and the first vertex
# # is distorted so that the vertices are clearly non coplanar
# x = [[1.0, 0.5, 0.6], [0.0, scaling, 0.0],
# [0.0, 0.0, scaling], [0.0, 1.0, 1.0]]
# cells = [[0, 1, 2, 3]]
# domain = ufl.Mesh(element("Lagrange", "quadrilateral", 1, shape=(2,)))
# mesh = create_mesh(MPI.COMM_SELF, cells, domain, x)
# _cpp.mesh.volume_entities(mesh, [0], mesh.topology.dim)
# assert "Not coplanar" in str(error.value)
|
