def test_initialise_vector(): mesh = CuboidMesh(1, 1, 1, 1, 1, 1) v = vector_field(mesh, lambda r: 2 * r) assert very_close(v, np.array((1, 1, 1)))
def test_save_vector_field(tmpdir): mesh = CuboidMesh(4, 3, 2, 4, 3, 2) s = vector_field(mesh, lambda r: (r[0], r[1], r[2])) vtk = VTK(mesh, directory=str(tmpdir), filename="save_vector") vtk.save_vector(s, name="s") assert same_as_ref(vtk.write_file(), REF_DIR)
def test_save_vector_field(): mesh = CuboidMesh(4, 3, 2, 4, 3, 2) s = vector_field(mesh, lambda r: (r[0], r[1], r[2])) vtk = VTK(mesh, directory=OUTPUT_DIR, filename="save_vector") vtk.save_vector(s, name="s")