a Ł ždFć @s"ddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z e gd¢”Ze gd ¢”Ze gd ¢”Ze gd ¢”Ze gd ¢gd ¢gd ¢gd¢gd¢gd¢gd¢gd ¢gd ¢gd ¢g ”Ze gd¢gd¢gd¢gd¢gd¢gd¢gd¢gd¢gd¢gd¢g ”Ze gd ¢gd¢gd¢gd¢gd¢gd¢gd¢gd¢gd ¢gd ¢g ”Ze gd ¢gd ¢gd!¢gd"¢gd#¢g”Ze gd$¢gd%¢gd&¢gd'¢gd'¢g”Ze gd(¢gd)¢gd(¢gd(¢ggd(¢gd*¢gd(¢gd(¢ggd(¢gd)¢gd(¢gd+¢ggd(¢gd(¢gd(¢gd+¢gg”Ze gd,¢gd,¢gd,¢gd-¢ggd,¢gd,¢gd,¢gd.¢ggd,¢gd,¢gd/¢gd0¢ggd1¢gd1¢gd/¢gd0¢gg”Ze gd2¢gd3¢gd)¢g”Ze  d4eed5feed5feed6feed5feed7feed8fg”d9d:„ƒZe  d;d8d5g”e  dd?„ƒƒZd@dA„ZdS)Bé)Śndimage)ŚdataN)Śmeasure)Ś expand_labels)Śtesting)Śassert_array_equal)rrérrrrrrér rrrrrr)rrrrrrr r r r r r r r rrr)rrrrrrrrr r rrrrrr)rrrrrrr r r r r r r rrr) rrrrrrrrrrr) rrr r rrrrrrr) rr r r rrrrrrr) rr r rrrrrrrr) rrr rrrrrrér) 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)rérr)rr r r)rrér)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ų?rr cCst||ƒ}t||ƒdS)N)rr)Z input_arrayZexpected_outputZexpand_distanceŚexpanded©rśv/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/segmentation/tests/test_expand_labels.pyŚtest_expand_labelss rŚndimŚdistanceéc Cs®tjdd|d}t |”}t||d}|dk}t||||ƒ|| t”}t  |”}||}|j dkr~t   ||k”s~J‚|| t”} | j dkrŖt   | |k”sŖJ‚dS)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š™™™™™©?)ŚlengthZblob_size_fractionZn_dim)rrN) rZ binary_blobsrŚlabelrrZastypeŚboolŚndiZdistance_transform_edtŚsizeŚnpŚall) rrŚimgŚlabelsr Z original_maskZexpanded_only_maskZ distance_mapZexpanded_distancesZbeyond_expanded_distancesrrrŚtest_binary_blobs‘s      rcCs.ttdƒ}ttjdƒ}t ||jk”r*J‚dS)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)rŚSAMPLE_EDGECASE_BEHAVIOURŚTrr)r Zexpanded_transposerrrŚtest_edge_case_behaviour«s  r!) ZscipyrrZskimagerŚnumpyrrZ#skimage.segmentation._expand_labelsrZskimage._sharedrZskimage._shared.testingrŚarrayZSAMPLE1DZSAMPLE1D_EXPANDED_3Z EDGECASE1DZEDGECASE1D_EXPANDED_3ZSAMPLE2DZSAMPLE2D_EXPANDED_3ZSAMPLE2D_EXPANDED_1_5Z EDGECASE2DZEDGECASE2D_EXPANDED_4ZSAMPLE3DZSAMPLE3D_EXPANDED_2rZ parametrizerŚrangerr!rrrrŚsž      ÷’ ÷’÷’ ü’ü’żżżżń’żżżżń’śž