def test_vertex_iterators():
"Iterate over vertices"
mesh = UnitCubeMesh(MPI.comm_world, 5, 5, 5)
for i in range(4):
mesh.init(0, i)
# Test connectivity
cons = [(i, mesh.topology.connectivity(0, i)) for i in range(4)]
# Test writability
for i, con in cons:
def assign(con, i):
con(i)[0] = 1
with pytest.raises(Exception):
assign(con, i)
n = 0
for i, v in enumerate(Vertices(mesh)):
n += 1
for j, con in cons:
assert numpy.all(con(i) == v.entities(j))
assert n == mesh.num_vertices()
def test_UnitCubeMeshLocal():
"""Create mesh of unit cube."""
mesh = UnitCubeMesh(MPI.comm_self, 5, 7, 9)
assert mesh.num_vertices() == 480
assert mesh.num_cells() == 1890
assert mesh.geometry.dim == 3
