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)