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(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")
