Translate chunks tuple to a set of slices in product order >>> slices_from_chunks(((2, 2), (3, 3, 3))) # doctest: +NORMALIZE_WHITESPACE [(slice(0, 2, None), slice(0, 3, None)), (slice(0, 2, None), slice(3, 6, None)), (slice(0, 2, None), slice(6, 9, None)), (slice(2, 4, N
(chunks)
| 247 | |
| 248 | |
| 249 | def slices_from_chunks(chunks): |
| 250 | """Translate chunks tuple to a set of slices in product order |
| 251 | |
| 252 | >>> slices_from_chunks(((2, 2), (3, 3, 3))) # doctest: +NORMALIZE_WHITESPACE |
| 253 | [(slice(0, 2, None), slice(0, 3, None)), |
| 254 | (slice(0, 2, None), slice(3, 6, None)), |
| 255 | (slice(0, 2, None), slice(6, 9, None)), |
| 256 | (slice(2, 4, None), slice(0, 3, None)), |
| 257 | (slice(2, 4, None), slice(3, 6, None)), |
| 258 | (slice(2, 4, None), slice(6, 9, None))] |
| 259 | """ |
| 260 | cumdims = [cached_cumsum(bds, initial_zero=True) for bds in chunks] |
| 261 | slices = [ |
| 262 | [slice(s, s + dim) for s, dim in zip(starts, shapes)] |
| 263 | for starts, shapes in zip(cumdims, chunks) |
| 264 | ] |
| 265 | return list(product(*slices)) |
| 266 | |
| 267 | |
| 268 | def graph_from_arraylike( |
no test coverage detected