a6dF ddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z ejgdZejgd Zejgd Zejgd Zejgd gd gd gdgdgdgdgd gd gd g Zejgdgdgdgdgdgdgdgdgdgdg Zejgd gdgdgdgdgdgdgdgd gd g Zejgd gd gd!gd"gd#gZejgd$gd%gd&gd'gd'gZejgd(gd)gd(gd(ggd(gd*gd(gd(ggd(gd)gd(gd+ggd(gd(gd(gd+ggZejgd,gd,gd,gd-ggd,gd,gd,gd.ggd,gd,gd/gd0ggd1gd1gd/gd0ggZejgd2gd3gd)gZe jd4eed5feed5feed6feed5feed7feed8fgd9Ze jd:d8d5ge jd;ed<d=Zd>ZdS)?)ndimage)dataN)measure) expand_labels)testing)assert_array_equal)rrrrrrrrr rrrrrr)r r r r r r r r r r r r r r rrr)rrr rrrrrr r rrrrrr)r r r r r r r r r r r r r rrr) rrrrrrrrrrr) rrr r rrrrrrr) rr r r rrrrrrr) rr r rrrrrrrr) rrr rrrrrrr) rrrrrrrrrr r) r r r r r r rrrrr) r r r r r r r rrr r) r r r r r r r r r r r ) r r r r r r rr r r r ) r r r r r r r r r r r ) r r r r r rr r r r r ) r r r r r rrr r r r ) rrr rrrrr r r r ) rr r r r rrrrrr) r r r r r rrrrrr) r r r r r rrrr r r ) r r r r rrrrr r r ) rr r r rrrrr r r ) rrrrrrrrr r r ) rrrrrr r rrrr) rrr r rr r rrrr) rr r r rr rrrrr) rrr rrrrrrrr) r r r r r r r r r r r) r r r r r r r r r r r ) r r r r r r r r r r r ) r r r r r r r r r r r)rrrr)rrr)rr r r)rrr)r r r r )rr r r)rr r r )r r r r )r r r r )r r r r)rr rr)r rrrz-input_array, expected_output, expand_distancer g?r r cFt||}t||dS)N)rr) input_arrayexpected_outputexpand_distanceexpandeds Mlib/python3.11/site-packages/skimage/segmentation/tests/test_expand_labels.pytest_expand_labelsrs)[/::Hx11111ndimdistancectjdd|}tj|}t ||}|dk}t ||||||z t}tj |}||}|j dkrtj ||ksJ||t} | j dkrtj | |ksJdSdS)a2Check some invariants with label expansion. - New labels array should exactly contain the original labels array. - Distance to old labels array within new labels should never exceed input distance. - Distance beyond the expanded labels should always exceed the input distance. @g?)lengthblob_size_fractionn_dim)rrN) r binary_blobsrlabelrrastypeboolndidistance_transform_edtsizenpall) rrimglabelsr original_maskexpanded_only_mask distance_mapexpanded_distancesbeyond_expanded_distancess rtest_binary_blobsr.s  2$d K K KC ]3  FVh777HaKMvm,h}.EFFF"V+33D99-}n==L%&89""v(H455555 ,hood.C.C-C D %))v/(:;;;;;*);;rcttd}ttjd}tj||jkrJdS)a Check edge case behavior to detect upstream changes For edge cases where a pixel has the same distance to several regions, lexicographical order seems to determine which region gets to expand into this pixel given the current upstream behaviour in scipy.ndimage.distance_map_edt. As a result, we expect different results when transposing the array. If this test fails, something has changed upstream. r N)rSAMPLE_EDGECASE_BEHAVIOURTr%r&)rexpanded_transposes rtest_edge_case_behaviourr3sM6::H&'@'BAFFvh"4"667777777r) scipyrr"skimagernumpyr%r#skimage.segmentation._expand_labelsrskimage._sharedrskimage._shared.testingrarraySAMPLE1DSAMPLE1D_EXPANDED_3 EDGECASE1DEDGECASE1D_EXPANDED_3SAMPLE2DSAMPLE2D_EXPANDED_3SAMPLE2D_EXPANDED_1_5 EDGECASE2DEDGECASE2D_EXPANDED_4SAMPLE3DSAMPLE3D_EXPANDED_2r0 parametrizerranger.r3rrrIs ======######666666 28GGG H HbhRRRSSRXFFF G G  !Q!Q!QRR 28&&&&&&&&&&&&&&&&&&&&&&&&&&&&&& (  bh(((((((((((((((((((((((((((((( *   !(((((((((((((((((((((((((((((( * + +RX&&&&&&&&&&&&&&& ( !((((((((((((((( *++ 28                     *RX                   *%BHlllLLL,,,%OPP3 "A& "A& $c*&*&* "A&   22  2 VaV$$Zq**<<+*%$<0 8 8 8 8 8r