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