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)