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
|
# Copyright (C) 2021 Matthew W. Scroggs and Jack Hale
#
# 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 dolfinx
import ufl
from basix.ufl import element, mixed_element
from dolfinx import default_real_type
from dolfinx.fem import form, functionspace
from dolfinx.mesh import CellType, GhostMode, create_unit_cube, create_unit_square
@pytest.mark.skip_in_parallel
@pytest.mark.parametrize("cell", [ufl.triangle, ufl.tetrahedron])
@pytest.mark.parametrize("degree", [1, 2])
@pytest.mark.parametrize("rank, family", [(0, "Lagrange"), (1, "Lagrange"), (1, "N1curl")])
def test_mixed_element(rank, family, cell, degree):
if cell == ufl.triangle:
mesh = create_unit_square(
MPI.COMM_WORLD, 1, 1, CellType.triangle, ghost_mode=GhostMode.shared_facet
)
else:
mesh = create_unit_cube(
MPI.COMM_WORLD, 1, 1, 1, CellType.tetrahedron, ghost_mode=GhostMode.shared_facet
)
shape = (mesh.geometry.dim,) * rank
norms = []
U_el = element(family, cell.cellname(), degree, shape=shape, dtype=default_real_type)
for i in range(3):
U = functionspace(mesh, U_el)
u = ufl.TrialFunction(U)
v = ufl.TestFunction(U)
a = form(ufl.inner(u, v) * ufl.dx)
A = dolfinx.fem.assemble_matrix(a)
A.scatter_reverse()
norms.append(A.squared_norm())
U_el = mixed_element([U_el])
for i in norms[1:]:
assert np.isclose(norms[0], i)
@pytest.mark.skip_in_parallel
def test_vector_element():
# Function space containing a scalar should work
mesh = create_unit_square(
MPI.COMM_WORLD, 1, 1, CellType.triangle, ghost_mode=GhostMode.shared_facet
)
gdim = mesh.geometry.dim
U = functionspace(mesh, ("P", 2, (gdim,)))
u, v = ufl.TrialFunction(U), ufl.TestFunction(U)
a = form(ufl.inner(u, v) * ufl.dx)
A = dolfinx.fem.assemble_matrix(a)
A.scatter_reverse()
with pytest.raises(ValueError):
# Function space containing a vector should throw an error rather
# than segfaulting
gdim = mesh.geometry.dim
U = functionspace(mesh, ("RT", 2, (gdim + 1,)))
u, v = ufl.TrialFunction(U), ufl.TestFunction(U)
a = form(ufl.inner(u, v) * ufl.dx)
A = dolfinx.fem.assemble_matrix(a)
A.scatter_reverse()
@pytest.mark.skip_in_parallel
@pytest.mark.parametrize("d1", range(1, 4))
@pytest.mark.parametrize("d2", range(1, 4))
def test_element_product(d1, d2):
mesh = create_unit_square(MPI.COMM_WORLD, 2, 2)
P3 = element(
"Lagrange", mesh.basix_cell(), d1, shape=(mesh.geometry.dim,), dtype=default_real_type
)
P1 = element("Lagrange", mesh.basix_cell(), d2, dtype=default_real_type)
TH = mixed_element([P3, P1])
W = functionspace(mesh, TH)
u = ufl.TrialFunction(W)
v = ufl.TestFunction(W)
a = form(ufl.inner(u[0], v[0]) * ufl.dx)
A = dolfinx.fem.assemble_matrix(a)
A.scatter_reverse()
W = functionspace(mesh, P3)
u = ufl.TrialFunction(W)
v = ufl.TestFunction(W)
a = form(ufl.inner(u[0], v[0]) * ufl.dx)
B = dolfinx.fem.assemble_matrix(a)
B.scatter_reverse()
assert np.isclose(A.squared_norm(), B.squared_norm())
|
