def test_mesh_function_assign_2D_facets():
mesh = UnitSquareMesh(MPI.comm_world, 3, 3)
mesh.init(1)
f = MeshFunction("int", mesh, mesh.topology.dim - 1, 25)
for cell in Cells(mesh):
for i, facet in enumerate(FacetRange(cell)):
assert 25 == f[facet]
g = MeshValueCollection("int", mesh, 1)
g.assign(f)
assert mesh.num_facets() == f.size()
assert mesh.num_cells() * 3 == g.size()
for cell in Cells(mesh):
for i, facet in enumerate(FacetRange(cell)):
assert 25 == g.get_value(cell.index(), i)
f2 = MeshFunction("int", mesh, g, 0)
for cell in Cells(mesh):
for i, facet in enumerate(FacetRange(cell)):
assert f2[facet] == g.get_value(cell.index(), i)
