def test_mesh_function_assign_2D_facets():
mesh = UnitSquareMesh(MPI.comm_world, 3, 3)
mesh.create_entities(1)
tdim = mesh.topology.dim
num_cell_facets = cpp.mesh.cell_num_entities(mesh.topology.cell_type,
tdim - 1)
f = MeshFunction("int", mesh, tdim - 1, 25)
connectivity = mesh.topology.connectivity(tdim, tdim - 1)
for c in range(mesh.num_cells()):
facets = connectivity.links(c)
for i in range(num_cell_facets):
assert 25 == f.values[facets[i]]
g = MeshValueCollection("int", mesh, 1)
g.assign(f)
assert mesh.num_entities(tdim - 1) == len(f.values)
assert mesh.num_cells() * 3 == g.size()
for c in range(mesh.num_cells()):
for i in range(num_cell_facets):
assert 25 == g.get_value(c, i)
f2 = MeshFunction("int", mesh, g, 0)
connectivity = mesh.topology.connectivity(tdim, tdim - 1)
for c in range(mesh.num_cells()):
facets = connectivity.links(c)
for i in range(num_cell_facets):
assert f2.values[facets[i]] == g.get_value(c, i)
def test_RefineUnitSquareMesh():
"""Refine mesh of unit square."""
mesh = UnitSquareMesh(MPI.comm_world, 5, 7)
mesh.create_entities(1)
mesh = refine(mesh, False)
assert mesh.num_entities_global(0) == 165
assert mesh.num_entities_global(2) == 280
def test_mesh_function_assign_2D_vertices():
mesh = UnitSquareMesh(MPI.comm_world, 3, 3)
mesh.create_entities(0)
f = MeshFunction("int", mesh, 0, 25)
g = MeshValueCollection("int", mesh, 0)
g.assign(f)
assert mesh.num_entities(0) == len(f.values)
assert mesh.num_cells() * 3 == g.size()
f2 = MeshFunction("int", mesh, g, 0)
num_cell_vertices = cpp.mesh.cell_num_vertices(mesh.topology.cell_type)
tdim = mesh.topology.dim
connectivity = mesh.topology.connectivity(tdim, 0)
for c in range(mesh.num_cells()):
vertices = connectivity.links(c)
for i in range(num_cell_vertices):
assert 25 == g.get_value(c, i)
assert f2.values[vertices[i]] == g.get_value(c, i)
