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
|
# Copyright (C) 2019 Chris Richardson
#
# This file is part of DOLFINx (https://www.fenicsproject.org)
#
# SPDX-License-Identifier: LGPL-3.0-or-later
"""Unit tests for the Constant class"""
from mpi4py import MPI
import numpy as np
import pytest
from dolfinx.fem import Constant
from dolfinx.mesh import create_unit_cube
def test_scalar_constant():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
c = Constant(mesh, 1.0)
assert c.value.shape == ()
assert c.value == 1.0 # /NOSONAR
c.value += 1.0
assert c.value == 2.0 # /NOSONAR
c.value = 3.0
assert c.value == 3.0 # /NOSONAR
def test_reshape():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
c = Constant(mesh, 1.0)
with pytest.raises(ValueError):
c.value.resize(100)
def test_wrong_dim():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
c = Constant(mesh, [1.0, 2.0])
assert c.value.shape == (2,)
with pytest.raises(ValueError):
c.value = [1.0, 2.0, 3.0]
def test_vector_constant():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
c0 = Constant(mesh, [1.0, 2.0])
c1 = Constant(mesh, np.array([1.0, 2.0]))
assert c0.value.all() == c1.value.all()
c0.value += 1.0
assert c0.value.all() == np.array([2.0, 3.0]).all()
c0.value -= [1.0, 2.0]
assert c0.value[0] == c0.value[1]
def test_tensor_constant():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
data = [[1.0, 2.0, 1.0], [1.0, 2.0, 1.0], [1.0, 2.0, 1.0]]
c0 = Constant(mesh, data)
assert c0.value.shape == (3, 3)
assert c0.value.all() == np.asarray(data).all()
c0.value *= 2.0
assert c0.value.all() == (2.0 * np.asarray(data)).all()
def test_float_method():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
a = 1.0
c0 = Constant(mesh, a)
assert a == float(c0) # /NOSONAR
def test_complex_method():
mesh = create_unit_cube(MPI.COMM_WORLD, 2, 2, 2)
a = 1.0 + 1.0j
c0 = Constant(mesh, a)
assert a == complex(c0)
|
