def test_tiny(self, dims, inds, ndims, ninds):
dims, inds = qu.dim_compress(dims, inds)
assert dims == ndims
assert inds == ninds
def test_middle(self):
dims = [5, 3, 2, 5, 4, 3, 2]
coos = [1, 2, 3, 5]
ndims, ncoos = qu.dim_compress(dims, coos)
assert ndims == (5, 30, 4, 3, 2)
assert ncoos == (1, 3)
def test_single(self):
dims = [5, 3, 2, 5, 4, 3, 2]
coos = 3
ndims, ncoos = qu.dim_compress(dims, coos)
assert ndims == (30, 5, 24)
assert ncoos == (1, )
def test_edge(self):
dims = [2, 3, 2, 4, 5]
coos = [0, 4]
ndims, ncoos = qu.dim_compress(dims, coos)
assert ndims == (2, 24, 5)
assert ncoos == (0, 2)
