def test_basic_assembly_constant(mode):
"""Tests assembly with Constant
The following test should be sensitive to order of flattening the
matrix-valued constant.
"""
mesh = create_unit_square(MPI.COMM_WORLD, 5, 5, ghost_mode=mode)
V = FunctionSpace(mesh, ("Lagrange", 1))
u, v = ufl.TrialFunction(V), ufl.TestFunction(V)
c = Constant(mesh, np.array([[1.0, 2.0], [5.0, 3.0]], PETSc.ScalarType))
a = inner(c[1, 0] * u, v) * dx + inner(c[1, 0] * u, v) * ds
L = inner(c[1, 0], v) * dx + inner(c[1, 0], v) * ds
a, L = form(a), form(L)
# Initial assembly
A1 = assemble_matrix(a)
A1.assemble()
b1 = assemble_vector(L)
b1.ghostUpdate(addv=PETSc.InsertMode.ADD, mode=PETSc.ScatterMode.REVERSE)
c.value = [[1.0, 2.0], [3.0, 4.0]]
A2 = assemble_matrix(a)
A2.assemble()
b2 = assemble_vector(L)
b2.ghostUpdate(addv=PETSc.InsertMode.ADD, mode=PETSc.ScatterMode.REVERSE)
assert (A1 * 3.0 - A2 * 5.0).norm() == pytest.approx(0.0)
assert (b1 * 3.0 - b2 * 5.0).norm() == pytest.approx(0.0)
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
c.value += 1.0
assert c.value == 2.0
c.value = 3.0
assert c.value == 3.0
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]