def test_save_2D_facet_function(tempdir, encoding, data_type):
if invalid_config(encoding):
pytest.skip("XDMF unsupported in current configuration")
dtype_str, dtype = data_type
mesh = UnitSquareMesh(MPI.comm_world, 32, 32)
mf = MeshFunction(dtype_str, mesh, mesh.topology.dim - 1, 0)
mf.rename("facets")
if (MPI.size(mesh.mpi_comm()) == 1):
for facet in Facets(mesh):
mf[facet] = dtype(facet.index())
else:
for facet in Facets(mesh):
mf[facet] = dtype(facet.global_index())
filename = os.path.join(tempdir, "mf_facet_2D_%s.xdmf" % dtype_str)
with XDMFFile(mesh.mpi_comm(), filename, encoding=encoding) as xdmf:
xdmf.write(mf)
with XDMFFile(mesh.mpi_comm(), filename) as xdmf:
read_function = getattr(xdmf, "read_mf_" + dtype_str)
mf_in = read_function(mesh, "facets")
diff = 0
for facet in Facets(mesh):
diff += (mf_in[facet] - mf[facet])
assert diff == 0
def test_save_3D_facet_function(tempdir, encoding, data_type):
dtype_str, dtype = data_type
mesh = UnitCubeMesh(MPI.comm_world, 4, 4, 4)
mf = MeshFunction(dtype_str, mesh, mesh.topology.dim - 1, 0)
mf.rename("facets")
if (MPI.size(mesh.mpi_comm()) == 1):
for facet in Facets(mesh):
mf[facet] = dtype(facet.index())
else:
for facet in Facets(mesh):
mf[facet] = dtype(facet.global_index())
filename = os.path.join(tempdir, "mf_facet_3D_%s.xdmf" % dtype_str)
with XDMFFile(mesh.mpi_comm(), filename, encoding=encoding) as xdmf:
xdmf.write(mf)
with XDMFFile(mesh.mpi_comm(), filename) as xdmf:
read_function = getattr(xdmf, "read_mf_" + dtype_str)
mf_in = read_function(mesh, "facets")
diff = 0
for facet in Facets(mesh):
diff += (mf_in[facet] - mf[facet])
assert diff == 0
def test_append_and_load_mesh_functions(tempdir, encoding, data_type):
if invalid_config(encoding):
pytest.skip("XDMF unsupported in current configuration")
dtype_str, dtype = data_type
meshes = [
UnitSquareMesh(MPI.comm_world, 12, 12),
UnitCubeMesh(MPI.comm_world, 2, 2, 2)
]
for mesh in meshes:
dim = mesh.topology.dim
vf = MeshFunction(dtype_str, mesh, 0, 0)
vf.rename("vertices")
ff = MeshFunction(dtype_str, mesh, mesh.topology.dim - 1, 0)
ff.rename("facets")
cf = MeshFunction(dtype_str, mesh, mesh.topology.dim, 0)
cf.rename("cells")
if (MPI.size(mesh.mpi_comm()) == 1):
for vertex in Vertices(mesh):
vf[vertex] = dtype(vertex.index())
for facet in Facets(mesh):
ff[facet] = dtype(facet.index())
for cell in Cells(mesh):
cf[cell] = dtype(cell.index())
else:
for vertex in Vertices(mesh):
vf[vertex] = dtype(vertex.global_index())
for facet in Facets(mesh):
ff[facet] = dtype(facet.global_index())
for cell in Cells(mesh):
cf[cell] = dtype(cell.global_index())
filename = os.path.join(tempdir, "appended_mf_%dD.xdmf" % dim)
with XDMFFile(mesh.mpi_comm(), filename, encoding=encoding) as xdmf:
xdmf.write(mesh)
xdmf.write(vf)
xdmf.write(ff)
xdmf.write(cf)
with XDMFFile(mesh.mpi_comm(), filename) as xdmf:
read_function = getattr(xdmf, "read_mf_" + dtype_str)
vf_in = read_function(mesh, "vertices")
ff_in = read_function(mesh, "facets")
cf_in = read_function(mesh, "cells")
diff = 0
for vertex in Vertices(mesh):
diff += (vf_in[vertex] - vf[vertex])
for facet in Facets(mesh):
diff += (ff_in[facet] - ff[facet])
for cell in Cells(mesh):
diff += (cf_in[cell] - cf[cell])
assert diff == 0
def test_facet_iterators():
"Iterate over facets"
mesh = UnitCubeMesh(MPI.comm_world, 5, 5, 5)
n = 0
for f in Facets(mesh):
n += 1
assert n == mesh.num_facets()
def test_facet_iterators():
"""Iterate over facets"""
mesh = UnitCubeMesh(MPI.comm_world, 5, 5, 5)
n = 0
for f in Facets(mesh):
n += 1
assert n == mesh.num_entities(mesh.topology.dim - 1)
def test_normal():
"Test that the normal() method is wrapped"
mesh = UnitSquareMesh(MPI.comm_world, 4, 4)
mesh.init(1)
for facet in Facets(mesh):
n = facet.normal()
nx, ny, nz = n[0], n[1], n[2]
assert isinstance(nx, float)
assert isinstance(ny, float)
assert isinstance(nz, float)
def test_ghost_connectivities(mode):
# Ghosted mesh
meshG = UnitSquareMesh(MPI.comm_world, 4, 4, ghost_mode=mode)
meshG.init(1, 2)
# Reference mesh, not ghosted, not parallel
meshR = UnitSquareMesh(MPI.comm_self,
4,
4,
ghost_mode=cpp.mesh.GhostMode.none)
meshR.init(1, 2)
# Create reference mapping from facet midpoint to cell midpoint
reference = {}
for facet in Facets(meshR):
fidx = facet.index()
facet_mp = tuple(facet.midpoint()[:])
reference[facet_mp] = []
for cidx in meshR.topology.connectivity(1, 2)(fidx):
cell = Cell(meshR, cidx)
cell_mp = tuple(cell.midpoint()[:])
reference[facet_mp].append(cell_mp)
# Loop through ghosted mesh and check connectivities
allowable_cell_indices = [
c.index() for c in Cells(meshG, cpp.mesh.MeshRangeType.ALL)
]
for facet in Facets(meshG, cpp.mesh.MeshRangeType.REGULAR):
fidx = facet.index()
facet_mp = tuple(facet.midpoint()[:])
assert facet_mp in reference
for cidx in meshG.topology.connectivity(1, 2)(fidx):
assert cidx in allowable_cell_indices
cell = Cell(meshG, cidx)
cell_mp = tuple(cell.midpoint()[:])
assert cell_mp in reference[facet_mp]
