def test_cell_sizes():
vec1 = np.array([1, 3])
vec2 = np.array([-1, 0, 1])
vec3 = np.array([2, 4, 5, 10])
scalar = 0.5
cs1, cs2, cs3 = [np.diff(vec) for vec in (vec1, vec2, vec3)]
csscal = 0
# Grid as set
grid = TensorGrid(vec1, vec2, vec3, as_midp=False)
assert all_equal(grid.cell_sizes(), (cs1, cs2, cs3))
grid = TensorGrid(vec1, scalar, vec3, as_midp=False)
assert all_equal(grid.cell_sizes(), (cs1, csscal, cs3))
# Grid as tesselation
cs1 = (2, 2)
cs2 = (1, 1, 1)
cs3 = (2, 1.5, 3, 5)
grid = TensorGrid(vec1, vec2, vec3, as_midp=True)
assert all_equal(grid.cell_sizes(), (cs1, cs2, cs3))
grid = TensorGrid(vec1, scalar, vec3, as_midp=True)
assert all_equal(grid.cell_sizes(), (cs1, csscal, cs3))
