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
|
# SPDX-FileCopyrightText: Copyright © DUNE Project contributors, see file AUTHORS.md
# SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception OR LGPL-3.0-or-later
import numpy as np
import dune.grid
import dune.functions as functions
from dune.functions import defaultGlobalBasis, subspaceBasis, Lagrange, Power, RaviartThomas, Composite
import basistest # The general test suite for function space bases
import interpolatetest # The general test suite for interpolation into bases
# Test all currently supported global bases in dimension `dimension`.
# Only the size of the global bases are tested.
def test(dimension):
lowerLeft = [-1] * dimension
upperRight = [1] * dimension
elements = [1] * dimension
grid = dune.grid.structuredGrid(lowerLeft,upperRight,elements)
# Test with first-order Lagrange basis vector field
basis1 = defaultGlobalBasis(grid, Power(Lagrange(order=1), dimension))
for i in range(dimension):
basis1_i = subspaceBasis(basis1, i)
basistest.checkBasis(basis1_i, checkLocalIndexSetCompleteness=False)
interpolatetest.checkConstantInterpolation(basis1_i, 0)
interpolatetest.checkBasisFunctionInterpolation(basis1_i)
# Test with lowest order Taylor-Hood basis
taylorHoodBasis = defaultGlobalBasis(grid, Composite(Power(Lagrange(order=2), dimension), Lagrange(order=1)))
# Check velocity subspace
velocityBasis = subspaceBasis(taylorHoodBasis, 0)
basistest.checkBasis(velocityBasis, checkLocalIndexSetCompleteness=False)
interpolatetest.checkConstantInterpolation(velocityBasis, np.zeros(dimension))
interpolatetest.checkBasisFunctionInterpolation(velocityBasis)
# Check pressure subspace
pressureBasis = subspaceBasis(taylorHoodBasis, 1)
basistest.checkBasis(pressureBasis, checkLocalIndexSetCompleteness=False)
interpolatetest.checkConstantInterpolation(pressureBasis, 0)
interpolatetest.checkBasisFunctionInterpolation(pressureBasis)
# Check velocity component subspaces obtained by passing multiple indices at once
for i in range(dimension):
velocityBasis_i = subspaceBasis(taylorHoodBasis, 0, i)
basistest.checkBasis(velocityBasis_i, checkLocalIndexSetCompleteness=False)
interpolatetest.checkConstantInterpolation(velocityBasis_i, 0)
interpolatetest.checkBasisFunctionInterpolation(velocityBasis_i)
# Check velocity component subspaces obtained via nested subspaceBasis call
for i in range(dimension):
velocityBasis_i = subspaceBasis(velocityBasis, i)
basistest.checkBasis(velocityBasis_i, checkLocalIndexSetCompleteness=False)
interpolatetest.checkConstantInterpolation(velocityBasis_i, 0)
interpolatetest.checkBasisFunctionInterpolation(velocityBasis_i)
# Test with the basis tree used for the mixed Poisson problem
# (This involves a basis of vector-valued functions.)
mixedPoissonBasis = defaultGlobalBasis(grid, Composite(RaviartThomas(order=0),
Lagrange(order=0)))
for i in (0,1):
subBasis = functions.subspaceBasis(mixedPoissonBasis, i)
basistest.checkBasis(subBasis, checkLocalIndexSetCompleteness=False)
interpolatetest.checkBasisFunctionInterpolation(subBasis)
# Run tests for grids of dimension 2
test(2)
|
