def test_ghost_connectivities(mode):
# Ghosted mesh
meshG = UnitSquareMesh(MPI.comm_world, 4, 4, ghost_mode=mode)
meshG.create_connectivity(1, 2)
# Reference mesh, not ghosted, not parallel
meshR = UnitSquareMesh(MPI.comm_self, 4, 4, ghost_mode=cpp.mesh.GhostMode.none)
meshR.create_connectivity(1, 2)
tdim = meshR.topology.dim
# Create reference mapping from facet midpoint to cell midpoint
reference = {}
facet_mp = cpp.mesh.midpoints(meshR, tdim - 1, range(meshR.num_entities(tdim - 1)))
cell_mp = cpp.mesh.midpoints(meshR, tdim, range(meshR.num_entities(tdim)))
reference = dict.fromkeys([tuple(row) for row in facet_mp], [])
for i in range(meshR.num_entities(tdim - 1)):
for cidx in meshR.topology.connectivity(1, 2).connections(i):
reference[tuple(facet_mp[i])].append(cell_mp[cidx].tolist())
# Loop through ghosted mesh and check connectivities
tdim = meshG.topology.dim
num_facets = meshG.num_entities(tdim - 1) - meshG.topology.ghost_offset(tdim - 1)
allowable_cell_indices = range(meshG.num_cells())
facet_mp = cpp.mesh.midpoints(meshG, tdim - 1, range(meshG.num_entities(tdim - 1)))
cell_mp = cpp.mesh.midpoints(meshG, tdim, range(meshG.num_entities(tdim)))
for i in range(num_facets):
assert tuple(facet_mp[i]) in reference
for cidx in meshG.topology.connectivity(1, 2).connections(i):
assert cidx in allowable_cell_indices
assert cell_mp[cidx].tolist() in reference[tuple(facet_mp[i])]
def test_issue_568():
mesh = UnitSquareMesh(MPI.comm_self, 4, 4)
cell = Cell(mesh, 0)
# This no longer fails because serial mesh now is building facets, using
# same pipeline as parallel.
# Should throw an error, not just segfault (only works in DEBUG mode!)
with pytest.raises(RuntimeError):
cell.facet_area(0)
# Should work after initializing the connectivity
mesh.create_connectivity(2, 1)
cell.facet_area(0)
def test_assign_2D_vertices():
mesh = UnitSquareMesh(MPI.comm_world, 3, 3)
mesh.create_connectivity(2, 0)
ncells = mesh.num_cells()
f = MeshValueCollection("int", mesh, 0)
all_new = True
for cell in Cells(mesh):
value = ncells - cell.index()
for i, vert in enumerate(VertexRange(cell)):
all_new = all_new and f.set_value(cell.index(), 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 cell in Cells(mesh):
value = ncells - cell.index()
for i, vert in enumerate(VertexRange(cell)):
assert value + i == g.get_value(cell.index(), 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)
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.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)
