(tempdir)
| 91 | |
| 92 | |
| 93 | def test_save_1d_vector(tempdir): |
| 94 | mesh = create_unit_interval(MPI.COMM_WORLD, 32) |
| 95 | |
| 96 | def f(x): |
| 97 | vals = np.zeros((2, x.shape[1])) |
| 98 | vals[0] = x[0] |
| 99 | vals[1] = 2 * x[0] * x[0] |
| 100 | return vals |
| 101 | |
| 102 | e = element("Lagrange", mesh.basix_cell(), 2, shape=(2,), dtype=default_real_type) |
| 103 | u = Function(functionspace(mesh, e)) |
| 104 | u.interpolate(f) |
| 105 | filename = Path(tempdir, "u.pvd") |
| 106 | with VTKFile(MPI.COMM_WORLD, filename, "w") as vtk: |
| 107 | vtk.write_function(u, 0.0) |
| 108 | |
| 109 | |
| 110 | @pytest.mark.parametrize("cell_type", cell_types_2D) |
nothing calls this directly
no test coverage detected