def test_regulargrid_center():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
center = (1, 0, -2)
grid = RegularGrid(minpt, maxpt, shape)
assert all_equal(grid.center, center)
def test_regulargrid_stride():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
stride = (0.5, 0, 3)
grid = RegularGrid(minpt, maxpt, shape)
assert all_equal(grid.stride, stride)
def test_regulargrid_init():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
# Check correct initialization of coord_vectors
grid = RegularGrid(minpt, maxpt, shape)
vec1 = (0.75, 1.25)
vec2 = (0, )
vec3 = (-5, -2, 1)
assert all_equal(grid.coord_vectors, (vec1, vec2, vec3))
def test_regulargrid_getitem():
minpt = (0.75, 0, -5, 4)
maxpt = (1.25, 0, 1, 13)
shape = (2, 1, 5, 4)
grid = RegularGrid(minpt, maxpt, shape)
# Single indices yield points as an array
indices = [1, 0, 1, 1]
values = [vec[i] for i, vec in zip(indices, grid.coord_vectors)]
assert all_equal(grid[1, 0, 1, 1], values)
indices = [0, 0, 4, 3]
values = [vec[i] for i, vec in zip(indices, grid.coord_vectors)]
assert all_equal(grid[0, 0, 4, 3], values)
with pytest.raises(IndexError):
grid[1, 0, 1, 2, 0]
with pytest.raises(IndexError):
grid[1, 1, 6, 2]
with pytest.raises(IndexError):
grid[1, 0, 4, 6]
# Slices return new RegularGrid's
assert grid == grid[...]
# Use TensorGrid implementation as reference here
tensor_grid = TensorGrid(*grid.coord_vectors)
test_slice = np.s_[1, :, ::2, ::3]
assert all_equal(grid[test_slice].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[1:2, :, ::2, ::3].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[1:2, :, ::2, ..., ::3].coord_vectors,
tensor_grid[test_slice].coord_vectors)
test_slice = np.s_[0:1, :, :, 2:4]
assert all_equal(grid[test_slice].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[:1, :, :, 2:].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[:-1, ..., 2:].coord_vectors,
tensor_grid[test_slice].coord_vectors)
test_slice = np.s_[:, 0, :, :]
assert all_equal(grid[test_slice].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[:, 0, ...].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[0:2, :, ...].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[...].coord_vectors,
tensor_grid[test_slice].coord_vectors)
test_slice = np.s_[:, :, 0::2, :]
assert all_equal(grid[test_slice].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[..., 0::2, :].coord_vectors,
tensor_grid[test_slice].coord_vectors)
test_slice = np.s_[..., 1, :]
assert all_equal(grid[test_slice].coord_vectors,
tensor_grid[test_slice].coord_vectors)
assert all_equal(grid[:, :, 1, :].coord_vectors,
tensor_grid[test_slice].coord_vectors)
# Fewer indices
assert grid[1:] == grid[1:, :, :, :]
assert grid[1:, 0] == grid[1:, 0, :, :]
assert grid[1:, 0, :-1] == grid[1:, 0, :-1, :]
# Two ellipses not allowed
with pytest.raises(ValueError):
grid[1, ..., ..., 0]
# Too many axes
with pytest.raises(IndexError):
grid[1, 0, 1:2, 0, :]
# New axes not supported
with pytest.raises(ValueError):
grid[1, 0, None, 1, 0]
# Empty axes not allowed
with pytest.raises(ValueError):
grid[1, 0, 0:0, 1]
with pytest.raises(ValueError):
grid[1, 1:, 0, 1]
# One-dimensional grid
grid = RegularGrid(1, 5, 5)
assert grid == grid[...]
sub_grid = RegularGrid(1, 5, 3)
assert grid[::2], sub_grid
def test_regulargrid_insert():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
minpt2 = (1, 1)
maxpt2 = (3, 1)
shape2 = (5, 1)
grid = RegularGrid(minpt, maxpt, shape)
grid2 = RegularGrid(minpt2, maxpt2, shape2)
# Test all positions
ins_grid = grid.insert(0, grid2)
assert isinstance(ins_grid, RegularGrid)
ins_minpt = minpt2 + minpt
ins_maxpt = maxpt2 + maxpt
ins_shape = shape2 + shape
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(1, grid2)
ins_minpt = minpt[:1] + minpt2 + minpt[1:]
ins_maxpt = maxpt[:1] + maxpt2 + maxpt[1:]
ins_shape = shape[:1] + shape2 + shape[1:]
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(2, grid2)
ins_minpt = minpt[:2] + minpt2 + minpt[2:]
ins_maxpt = maxpt[:2] + maxpt2 + maxpt[2:]
ins_shape = shape[:2] + shape2 + shape[2:]
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(-1, grid2)
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(3, grid2)
ins_minpt = minpt + minpt2
ins_maxpt = maxpt + maxpt2
ins_shape = shape + shape2
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
# Insert a TensorGrid
vec = [-1, 0, 3]
tgrid = TensorGrid(vec)
ins_tgrid = grid.insert(3, tgrid)
assert isinstance(ins_tgrid, TensorGrid)
assert ins_tgrid == TensorGrid(*(grid.coord_vectors + (vec, )))
with pytest.raises(IndexError):
grid.insert(4, grid2)
with pytest.raises(IndexError):
grid.insert(-5, grid2)
with pytest.raises(TypeError):
grid.insert(0, [1, 2])
def test_regulargrid_is_subgrid():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 5)
# Optimized cases
grid = RegularGrid(minpt, maxpt, shape)
assert grid.is_subgrid(grid)
smaller_shape = (1, 1, 5)
not_sup_grid = RegularGrid(minpt, maxpt, smaller_shape)
assert not grid.is_subgrid(not_sup_grid)
larger_minpt = (0.85, 0, -4)
not_sup_grid = RegularGrid(larger_minpt, maxpt, shape)
assert not grid.is_subgrid(not_sup_grid)
smaller_maxpt = (1.15, 0, 0)
not_sup_grid = RegularGrid(minpt, smaller_maxpt, shape)
assert not grid.is_subgrid(not_sup_grid)
# Real checks
minpt_sup1 = (-0.25, -2, -5)
maxpt_sup1 = (1.25, 2, 1)
shape_sup1 = (4, 3, 9)
sup_grid = RegularGrid(minpt_sup1, maxpt_sup1, shape_sup1)
assert grid.is_subgrid(sup_grid)
assert not sup_grid.is_subgrid(grid)
minpt_sup2 = (0.5, 0, -5)
maxpt_sup2 = (1.5, 0, 1)
shape_sup2 = (5, 1, 9)
sup_grid = RegularGrid(minpt_sup2, maxpt_sup2, shape_sup2)
assert grid.is_subgrid(sup_grid)
assert not sup_grid.is_subgrid(grid)
shape_not_sup1 = (4, 3, 10)
not_sup_grid = RegularGrid(minpt_sup1, maxpt_sup1, shape_not_sup1)
assert not grid.is_subgrid(not_sup_grid)
assert not not_sup_grid.is_subgrid(grid)
minpt_not_sup1 = (-0.25, -2.5, -5)
not_sup_grid = RegularGrid(minpt_not_sup1, maxpt_sup1, shape_sup1)
assert not grid.is_subgrid(not_sup_grid)
assert not not_sup_grid.is_subgrid(grid)
maxpt_not_sup1 = (1.35, 2.0001, 1)
not_sup_grid = RegularGrid(minpt_sup1, maxpt_not_sup1, shape_sup1)
assert not grid.is_subgrid(not_sup_grid)
assert not not_sup_grid.is_subgrid(grid)
# Should also work for TensorGrid's
vec1_sup = (0.75, 1, 1.25, 7)
vec2_sup = (0, )
vec3_sup = (-5, -3.5, -3, -2, -0.5, 0, 1, 9.5)
tensor_sup_grid = TensorGrid(vec1_sup, vec2_sup, vec3_sup)
assert grid.is_subgrid(tensor_sup_grid)
vec1_not_sup = (1, 1.25, 7)
vec2_not_sup = (0, )
vec3_not_sup = (-4, -2, 1)
tensor_not_sup_grid = TensorGrid(vec1_not_sup, vec2_not_sup, vec3_not_sup)
assert not grid.is_subgrid(tensor_not_sup_grid)
# Fuzzy check
shape_sup = (4, 3, 9)
minpt_fuzzy_sup1 = (-0.24, -2, -5.01)
minpt_fuzzy_sup2 = (-0.24, -2, -5)
maxpt_fuzzy_sup1 = (1.24, 2, 1)
maxpt_fuzzy_sup2 = (1.25, 2, 1.01)
fuzzy_sup_grid = RegularGrid(minpt_fuzzy_sup1, maxpt_fuzzy_sup1, shape_sup)
assert grid.is_subgrid(fuzzy_sup_grid, atol=0.015)
assert not grid.is_subgrid(fuzzy_sup_grid, atol=0.005)
fuzzy_sup_grid = RegularGrid(minpt_fuzzy_sup2, maxpt_fuzzy_sup2, shape_sup)
assert grid.is_subgrid(fuzzy_sup_grid, atol=0.015)
assert not grid.is_subgrid(fuzzy_sup_grid, atol=0.005)
def test_regulargrid_init_raise():
# Check different error scenarios
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
nonpos_shape1 = (2, 0, 3)
nonpos_shape2 = (-2, 1, 3)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt, nonpos_shape1)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt, nonpos_shape2)
minpt_with_nan = (0.75, 0, np.nan)
minpt_with_inf = (0.75, 0, np.inf)
maxpt_with_nan = (1.25, np.nan, 1)
maxpt_with_inf = (1.25, np.inf, 1)
shape_with_nan = (2, np.nan, 3)
shape_with_inf = (2, np.inf, 3)
with pytest.raises(ValueError):
RegularGrid(minpt_with_nan, maxpt, shape)
with pytest.raises(ValueError):
RegularGrid(minpt_with_inf, maxpt, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt_with_nan, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt_with_inf, shape)
# Shape casting to int raises a TypeError
with pytest.raises(TypeError):
RegularGrid(minpt, maxpt, shape_with_nan)
with pytest.raises(TypeError):
RegularGrid(minpt, maxpt, shape_with_inf)
maxpt_smaller_minpt1 = (0.7, 0, 1)
maxpt_smaller_minpt2 = (1.25, -1, 1)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt_smaller_minpt1, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt_smaller_minpt2, shape)
too_short_minpt = (0.75, 0)
too_long_minpt = (0.75, 0, -5, 2)
too_short_maxpt = (0, 1)
too_long_maxpt = (1.25, 0, 1, 25)
too_short_shape = (2, 3)
too_long_shape = (2, 1, 4, 3)
bad_dim_shape = ((1, 2), (3, 4))
with pytest.raises(ValueError):
RegularGrid(too_short_minpt, maxpt, shape)
with pytest.raises(ValueError):
RegularGrid(too_long_minpt, maxpt, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, too_short_maxpt, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, too_long_maxpt, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt, too_short_shape)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt, too_long_shape)
maxpt_eq_minpt_at_shape_larger_than_1 = (0.75, 0, 1)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt_eq_minpt_at_shape_larger_than_1, shape)
with pytest.raises(ValueError):
RegularGrid(minpt, maxpt, bad_dim_shape)
def test_regular_is_subgrid(as_midp):
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 5)
# Optimized cases
grid = RegularGrid(minpt, maxpt, shape, as_midp=as_midp)
assert grid.is_subgrid(grid)
smaller_shape = (1, 1, 5)
not_sup_grid = RegularGrid(minpt, maxpt, smaller_shape, as_midp=as_midp)
assert not grid.is_subgrid(not_sup_grid)
larger_minpt = (0.85, 0, -4)
not_sup_grid = RegularGrid(larger_minpt, maxpt, shape, as_midp=as_midp)
assert not grid.is_subgrid(not_sup_grid)
smaller_maxpt = (1.15, 0, 0)
not_sup_grid = RegularGrid(minpt, smaller_maxpt, shape, as_midp=as_midp)
assert not grid.is_subgrid(not_sup_grid)
# Real checks
minpt_sup1 = (-0.25, -2, -5)
maxpt_sup1 = (1.25, 2, 1)
shape_sup1 = (4, 3, 9)
sup_grid = RegularGrid(minpt_sup1, maxpt_sup1, shape_sup1, as_midp=as_midp)
assert grid.is_subgrid(sup_grid)
assert not sup_grid.is_subgrid(grid)
minpt_sup2 = (0.5, 0, -5)
maxpt_sup2 = (1.5, 0, 1)
shape_sup2 = (5, 1, 9)
sup_grid = RegularGrid(minpt_sup2, maxpt_sup2, shape_sup2, as_midp=as_midp)
assert grid.is_subgrid(sup_grid)
assert not sup_grid.is_subgrid(grid)
shape_not_sup1 = (4, 3, 10)
not_sup_grid = RegularGrid(minpt_sup1, maxpt_sup1, shape_not_sup1,
as_midp=as_midp)
assert not grid.is_subgrid(not_sup_grid)
assert not not_sup_grid.is_subgrid(grid)
minpt_not_sup1 = (-0.25, -2.5, -5)
not_sup_grid = RegularGrid(minpt_not_sup1, maxpt_sup1, shape_sup1,
as_midp=as_midp)
assert not grid.is_subgrid(not_sup_grid)
assert not not_sup_grid.is_subgrid(grid)
maxpt_not_sup1 = (1.35, 2.0001, 1)
not_sup_grid = RegularGrid(minpt_sup1, maxpt_not_sup1, shape_sup1,
as_midp=as_midp)
assert not grid.is_subgrid(not_sup_grid)
assert not not_sup_grid.is_subgrid(grid)
# Should also work for TensorGrid's
vec1_sup = (0.75, 1, 1.25, 7)
vec2_sup = (0,)
vec3_sup = (-5, -3.5, -3, -2, -0.5, 0, 1, 9.5)
tensor_sup_grid = TensorGrid(vec1_sup, vec2_sup, vec3_sup,
as_midp=as_midp)
assert grid.is_subgrid(tensor_sup_grid)
vec1_not_sup = (1, 1.25, 7)
vec2_not_sup = (0,)
vec3_not_sup = (-4, -2, 1)
tensor_not_sup_grid = TensorGrid(vec1_not_sup, vec2_not_sup,
vec3_not_sup, as_midp=as_midp)
assert not grid.is_subgrid(tensor_not_sup_grid)
# Fuzzy check
shape_sup = (4, 3, 9)
minpt_fuzzy_sup1 = (-0.24, -2, -5.01)
minpt_fuzzy_sup2 = (-0.24, -2, -5)
maxpt_fuzzy_sup1 = (1.24, 2, 1)
maxpt_fuzzy_sup2 = (1.25, 2, 1.01)
fuzzy_sup_grid = RegularGrid(minpt_fuzzy_sup1, maxpt_fuzzy_sup1,
shape_sup, as_midp=as_midp)
assert grid.is_subgrid(fuzzy_sup_grid, tol=0.015)
assert not grid.is_subgrid(fuzzy_sup_grid, tol=0.005)
fuzzy_sup_grid = RegularGrid(minpt_fuzzy_sup2, maxpt_fuzzy_sup2,
shape_sup, as_midp=as_midp)
assert grid.is_subgrid(fuzzy_sup_grid, tol=0.015)
assert not grid.is_subgrid(fuzzy_sup_grid, tol=0.005)
def test_regular_grid_init():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
# Check correct initialization of coord_vectors
grid = RegularGrid(minpt, maxpt, shape)
vec1 = (0.75, 1.25)
vec2 = (0,)
vec3 = (-5, -2, 1)
assert all_equal(grid.coord_vectors, (vec1, vec2, vec3))
# Check different error scenarios
nonpos_shape1 = (2, 0, 3)
nonpos_shape2 = (-2, 1, 3)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt, nonpos_shape1)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt, nonpos_shape2)
minpt_with_nan = (0.75, 0, np.nan)
minpt_with_inf = (0.75, 0, np.inf)
maxpt_with_nan = (1.25, np.nan, 1)
maxpt_with_inf = (1.25, np.inf, 1)
shape_with_nan = (2, np.nan, 3)
shape_with_inf = (2, np.inf, 3)
with pytest.raises(ValueError):
grid = RegularGrid(minpt_with_nan, maxpt, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt_with_inf, maxpt, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt_with_nan, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt_with_inf, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt, shape_with_nan)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt, shape_with_inf)
maxpt_smaller_minpt1 = (0.7, 0, 1)
maxpt_smaller_minpt2 = (1.25, -1, 1)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt_smaller_minpt1, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt_smaller_minpt2, shape)
too_short_minpt = (0.75, 0)
too_long_minpt = (0.75, 0, -5, 2)
too_short_maxpt = (0, 1)
too_long_maxpt = (1.25, 0, 1, 25)
too_short_shape = (2, 3)
too_long_shape = (2, 1, 4, 3)
with pytest.raises(ValueError):
grid = RegularGrid(too_short_minpt, maxpt, shape)
with pytest.raises(ValueError):
grid = RegularGrid(too_long_minpt, maxpt, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, too_short_maxpt, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, too_long_maxpt, shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt, too_short_shape)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt, too_long_shape)
maxpt_eq_minpt_at_shape_larger_than_1 = (0.75, 0, 1)
with pytest.raises(ValueError):
grid = RegularGrid(minpt, maxpt_eq_minpt_at_shape_larger_than_1,
shape)
# Overrides for exact min and max
exact_min = np.array((0, 1, 0), dtype=float)
exact_max = np.array((1, 1, 3), dtype=float)
shape = np.array((3, 1, 7), dtype=int)
shift = (exact_max - exact_min) / (2 * shape)
minpt = exact_min + shift
maxpt = exact_max - shift
grid = RegularGrid(minpt, maxpt, shape, as_midp=True,
_exact_min=exact_min, _exact_max=exact_max)
assert all_equal(grid.min(), exact_min)
assert all_equal(grid.max(), exact_max)
def test_regulargrid_insert():
minpt = (0.75, 0, -5)
maxpt = (1.25, 0, 1)
shape = (2, 1, 3)
minpt2 = (1, 1)
maxpt2 = (3, 1)
shape2 = (5, 1)
grid = RegularGrid(minpt, maxpt, shape)
grid2 = RegularGrid(minpt2, maxpt2, shape2)
# Test all positions
ins_grid = grid.insert(0, grid2)
assert isinstance(ins_grid, RegularGrid)
ins_minpt = minpt2 + minpt
ins_maxpt = maxpt2 + maxpt
ins_shape = shape2 + shape
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(1, grid2)
ins_minpt = minpt[:1] + minpt2 + minpt[1:]
ins_maxpt = maxpt[:1] + maxpt2 + maxpt[1:]
ins_shape = shape[:1] + shape2 + shape[1:]
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(2, grid2)
ins_minpt = minpt[:2] + minpt2 + minpt[2:]
ins_maxpt = maxpt[:2] + maxpt2 + maxpt[2:]
ins_shape = shape[:2] + shape2 + shape[2:]
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(-1, grid2)
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
ins_grid = grid.insert(3, grid2)
ins_minpt = minpt + minpt2
ins_maxpt = maxpt + maxpt2
ins_shape = shape + shape2
assert ins_grid == RegularGrid(ins_minpt, ins_maxpt, ins_shape)
# Insert a TensorGrid
vec = [-1, 0, 3]
tgrid = TensorGrid(vec)
ins_tgrid = grid.insert(3, tgrid)
assert isinstance(ins_tgrid, TensorGrid)
assert ins_tgrid == TensorGrid(*(grid.coord_vectors + (vec,)))
with pytest.raises(IndexError):
grid.insert(4, grid2)
with pytest.raises(IndexError):
grid.insert(-5, grid2)
with pytest.raises(TypeError):
grid.insert(0, [1, 2])
