def test_fspace_one(out_shape): """Check one element.""" fspace = FunctionSpace(odl.IntervalProd(0, 1), out_dtype=(float, out_shape)) points = _points(fspace.domain, 4) mesh_shape = (5, ) mesh = _meshgrid(fspace.domain, mesh_shape) point = 0.5 values_points_shape = out_shape + (4, ) values_point_shape = out_shape values_mesh_shape = out_shape + mesh_shape f_one = fspace.one() assert all_equal(f_one(points), np.ones(values_points_shape)) if not out_shape: assert f_one(point) == 1.0 else: assert all_equal(f_one(point), np.ones(values_point_shape)) assert all_equal(f_one(mesh), np.ones(values_mesh_shape)) out_points = np.empty(values_points_shape) out_mesh = np.empty(values_mesh_shape) f_one(points, out=out_points) f_one(mesh, out=out_mesh) assert all_equal(out_points, np.ones(values_points_shape)) assert all_equal(out_mesh, np.ones(values_mesh_shape))
def test_fspace_one(): rect, points, mg = _standard_setup_2d() # real fspace = FunctionSpace(rect) one_vec = fspace.one() assert one_vec([0.5, 1.5]) == 1.0 assert all_equal(one_vec(points), np.ones(5, dtype=float)) assert all_equal(one_vec(mg), np.ones((2, 3), dtype=float)) # complex fspace = FunctionSpace(rect, field=odl.ComplexNumbers()) one_vec = fspace.one() assert one_vec([0.5, 1.5]) == 1.0 + 1j * 0.0 assert all_equal(one_vec(points), np.ones(5, dtype=complex)) assert all_equal(one_vec(mg), np.ones((2, 3), dtype=complex))