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)