def test_save_1d_mesh(tempfile, file_options):
mesh = UnitIntervalMesh(MPI.COMM_WORLD, 32)
VTKFile(tempfile + "mesh.pvd").write(mesh)
f = VTKFile(tempfile + "mesh.pvd")
f.write(mesh, 0.)
f.write(mesh, 1.)
for file_option in file_options:
VTKFile(tempfile + "mesh.pvd", file_option).write(mesh)
def test_save_3d_mesh(tempfile, file_options):
mesh = UnitCubeMesh(MPI.COMM_WORLD, 8, 8, 8)
VTKFile(tempfile + "mesh.pvd").write(mesh)
f = VTKFile(tempfile + "mesh.pvd")
f.write(mesh, 0.)
f.write(mesh, 1.)
for file_option in file_options:
VTKFile(tempfile + "mesh.pvd", file_option).write(mesh)
def test_save_2d_mesh(tempfile, file_options):
mesh = UnitSquareMesh(MPI.comm_world, 32, 32)
VTKFile(tempfile + "mesh.pvd").write(mesh)
f = VTKFile(tempfile + "mesh.pvd")
f.write(mesh, 0.)
f.write(mesh, 1.)
for file_option in file_options:
VTKFile(tempfile + "mesh.pvd", file_option).write(mesh)
def test_save_2d_scalar(tempfile, file_options):
mesh = UnitSquareMesh(MPI.COMM_WORLD, 16, 16)
u = Function(FunctionSpace(mesh, ("Lagrange", 2)))
u.vector.set(1.0)
VTKFile(tempfile + "u.pvd").write(u)
f = VTKFile(tempfile + "u.pvd")
f.write(u, 0.)
f.write(u, 1.)
for file_option in file_options:
VTKFile(tempfile + "u.pvd", file_option).write(u)
def test_save_3d_tensor(tempfile, file_options):
mesh = UnitCubeMesh(MPI.COMM_WORLD, 8, 8, 8)
u = Function(TensorFunctionSpace(mesh, ("Lagrange", 2)))
u.vector.set(1)
VTKFile(tempfile + "u.pvd").write(u)
f = VTKFile(tempfile + "u.pvd")
f.write(u, 0.)
f.write(u, 1.)
for file_option in file_options:
VTKFile(tempfile + "u.pvd", file_option).write(u)
def test_save_1d_meshfunctions(tempfile, mesh_function_types, file_options,
type_conv):
mesh = UnitIntervalMesh(MPI.comm_world, 32)
for d in range(mesh.topology.dim + 1):
for t in mesh_function_types:
mf = MeshFunction(t, mesh, mesh.topology.dim - d, type_conv[t](1))
VTKFile(tempfile + "mf.pvd").write(mf)
f = VTKFile(tempfile + "mf.pvd")
f.write(mf, 0.)
f.write(mf, 1.)
def test_save_2d_tensor(tempfile, file_options):
mesh = UnitSquareMesh(MPI.comm_world, 16, 16)
u = Function(TensorFunctionSpace(mesh, ("Lagrange", 2)))
u.vector.set(1)
VTKFile(tempfile + "u.pvd").write(u)
f = VTKFile(tempfile + "u.pvd")
f.write(u, 0.)
f.write(u, 1.)
for file_option in file_options:
VTKFile(tempfile + "u.pvd", file_option).write(u)
def test_save_3d_vector(tempfile, file_options):
mesh = UnitCubeMesh(MPI.comm_world, 8, 8, 8)
u = Function(VectorFunctionSpace(mesh, "Lagrange", 2))
u.vector.set(1)
VTKFile(tempfile + "u.pvd").write(u)
f = VTKFile(tempfile + "u.pvd")
f.write(u, 0.)
f.write(u, 1.)
for file_option in file_options:
VTKFile(tempfile + "u.pvd", file_option).write(u)
def test_save_1d_scalar(tempfile, file_options):
mesh = UnitIntervalMesh(MPI.comm_world, 32)
u = Function(FunctionSpace(mesh, ("Lagrange", 2)))
u.vector.set(1.0)
VTKFile(tempfile + "u.pvd").write(u)
f = VTKFile(tempfile + "u.pvd")
f.write(u, 0.)
f.write(u, 1.)
for file_option in file_options:
VTKFile(tempfile + "u.pvd", file_option).write(u)
def test_save_3d_meshfunctions(tempfile, mesh_function_types, file_options,
type_conv, cell_types_3D):
mesh = UnitCubeMesh(MPI.comm_world, 8, 8, 8)
for d in range(mesh.topology.dim + 1):
for t in mesh_function_types:
for cell_type in cell_types_3D:
mf = MeshFunction(t, mesh, mesh.topology.dim - d, type_conv[t](1))
VTKFile(tempfile + "mf_{0:d}_{1:s}.pvd".format(mesh.topology.dim - d,
str(cell_type).split(".")[-1])).write(mf)
f = VTKFile(tempfile + "mf{0:d}_{1:s}.pvd".format(mesh.topology.dim - d,
str(cell_type).split(".")[-1]))
f.write(mf, 0.)
f.write(mf, 1.)
