def test_assign_2D_facets():
mesh = UnitSquareMesh(MPI.comm_world, 3, 3)
mesh.create_connectivity(2, 1)
tdim = mesh.topology.dim
num_cell_facets = cpp.mesh.cell_num_entities(mesh.topology.cell_type,
tdim - 1)
ncells = mesh.num_cells()
f = MeshValueCollection("int", mesh, 1)
all_new = True
for c in range(ncells):
value = ncells - c
for i in range(num_cell_facets):
all_new = all_new and f.set_value(c, i, value + i)
g = MeshValueCollection("int", mesh, 1)
g.assign(f)
assert ncells * 3 == f.size()
assert ncells * 3 == g.size()
assert all_new
for c in range(ncells):
value = ncells - c
for i in range(num_cell_facets):
assert value + i == g.get_value(c, i)
def test_assign_2D_vertices():
mesh = UnitSquareMesh(MPI.comm_world, 3, 3)
mesh.create_connectivity(2, 0)
ncells = mesh.num_cells()
num_cell_vertices = cpp.mesh.cell_num_vertices(mesh.cell_type)
f = MeshValueCollection("int", mesh, 0)
all_new = True
for c in range(ncells):
value = ncells - c
for i in range(num_cell_vertices):
all_new = all_new and f.set_value(c, i, value + i)
g = MeshValueCollection("int", mesh, 0)
g.assign(f)
assert ncells * 3 == f.size()
assert ncells * 3 == g.size()
assert all_new
for c in range(ncells):
value = ncells - c
for i in range(num_cell_vertices):
assert value + i == g.get_value(c, i)