dask / dask-image / 159 / 3

100%

main: 100%

Pull #78

travis-ci

Use `labeled_comprehension` in `center_of_mass`

Rewrite `center_of_mass` to use `labeled_comprehension` along with a
custom kernel. This simplifies the Dask graph significantly, improves
the performance noticeably, and cleans up the code a bit as the whole
center of mass computation is now handled by the custom kernel function.
Only a little bit of bookkeeping is needed to bind a few extra arguments
(since `labeled_comprehension` does not handle keyword arguments) to the
custom kernel and to convert the custom structured array type (needed
since `labeled_comprehension` cannot handle results with a shape) into a
normal array before returning it user.

220 of 230 branches covered (95.65%)

Branch coverage included in aggregate %.

724 of 730 relevant lines covered (99.18%)

0.99 hits per line