def test_compute_point_values(V, W, mesh):
u = Function(V)
v = Function(W)
with u.vector().localForm() as u_local, v.vector().localForm() as v_local:
u_local.set(1.0)
v_local.set(1.0)
u_values = u.compute_point_values(mesh)
v_values = v.compute_point_values(mesh)
u_ones = np.ones_like(u_values, dtype=np.float64)
assert np.all(np.isclose(u_values, u_ones))
v_ones = np.ones_like(v_values, dtype=np.float64)
assert np.all(np.isclose(v_values, v_ones))
u_values2 = u.compute_point_values()
assert all(u_values == u_values2)
def test_compute_point_values(V, W, mesh):
from numpy import all
u = Function(V)
v = Function(W)
u.vector()[:] = 1.
v.vector()[:] = 1.
u.vector().update_ghosts()
v.vector().update_ghosts()
u_values = u.compute_point_values(mesh)
v_values = v.compute_point_values(mesh)
u_ones = numpy.ones_like(u_values, dtype=numpy.float64)
assert all(numpy.isclose(u_values, u_ones))
v_ones = numpy.ones_like(v_values, dtype=numpy.float64)
assert all(numpy.isclose(v_values, v_ones))
u_values2 = u.compute_point_values()
assert all(u_values == u_values2)
