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.)