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
|
from utils import FiniteElement, LagrangeElement, MixedElement, SymmetricElement
from ufl import Cell, Mesh
from ufl.functionspace import FunctionSpace
from ufl.pullback import contravariant_piola, covariant_piola
from ufl.sobolevspace import HCurl, HDiv
def test_reference_shapes():
# show_elements()
cell = Cell("triangle")
domain = Mesh(LagrangeElement(cell, 1, (3,)))
V = FiniteElement("N1curl", cell, 1, (2,), covariant_piola, HCurl)
Vspace = FunctionSpace(domain, V)
assert Vspace.value_shape == (3,)
assert V.reference_value_shape == (2,)
U = FiniteElement("Raviart-Thomas", cell, 1, (2,), contravariant_piola, HDiv)
Uspace = FunctionSpace(domain, U)
assert Uspace.value_shape == (3,)
assert U.reference_value_shape == (2,)
W = LagrangeElement(cell, 1)
Wspace = FunctionSpace(domain, W)
assert Wspace.value_shape == ()
assert W.reference_value_shape == ()
Q = LagrangeElement(cell, 1, (3,))
Qspace = FunctionSpace(domain, Q)
assert Qspace.value_shape == (3,)
assert Q.reference_value_shape == (3,)
T = LagrangeElement(cell, 1, (3, 3))
Tspace = FunctionSpace(domain, T)
assert Tspace.value_shape == (3, 3)
assert T.reference_value_shape == (3, 3)
S = SymmetricElement(
{
(0, 0): 0,
(1, 0): 1,
(2, 0): 2,
(0, 1): 1,
(1, 1): 3,
(2, 1): 4,
(0, 2): 2,
(1, 2): 4,
(2, 2): 5,
},
[LagrangeElement(cell, 1) for _ in range(6)],
)
Sspace = FunctionSpace(domain, S)
assert Sspace.value_shape == (3, 3)
assert S.reference_value_shape == (6,)
M = MixedElement([V, U, W])
Mspace = FunctionSpace(domain, M)
assert Mspace.value_shape == (7,)
assert M.reference_value_shape == (5,)
|
