1-PhP&dZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl mZddl mZddlmZd Zejgd gd gd gd gd gdgdgdgdgdgdgdgdgdgdgdgd gd gd gd gd gZdZGddejZdZdZdZdZdZdZejdej ej!ej"ej#ej$ej%ej&ej'gdZ(dZ)d Z*d!Z+d"Z,d#Z-dS)$z0test_watershed.py - tests the watershed functionN)ndimage)gaussian)peak_local_max)label) watershedg-q=)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 )r r r r r ogrrr r r r r r )r r r r r^H4rrrrr r r r )r r r r rr'rrrrr r r r )r r r r r rrrrrrr r r r r )r r r r r r rrrrr r r r r r )r r r r r r r rrr r r r r r r )r r r r r r r rrr r r r r r r ct|tjstj|}t|tjstj|}d|jvr d|jvrdSd||dk<|jtjtjfvs|jtjtjfvrtj|tj}tj|tj}|j|jz dz |j |j z dz z}ntj|}| tj }tj|}| tj }||z dz }tj|S)Nrgr) isinstancenpndarrayasarrayshapedtype complex64 complex128realsumimagastypefloat64mathsqrt)abts i/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/segmentation/tests/test_watershed.pydiffr)+sk a $ $ JqMM a $ $ JqMM QW 1<<sAa1fIw2<///17  ?44 Jq"- ( ( Jq"- ( (fqvo! # ( ( * *qv1.D-I-I-K-K K JqMM HHRZ  JqMM HHRZ !e\    9Q<<c~eZdZejdeZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdS) TestWatershedr.c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j}tjgdgdgdgdgdgdgdgdgdgdg }t ||}|tksJdS) z watershed 1rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr4r4r4r4r4r4r4r4rrrrrr4N)rarrayuint8int8reightr)eps)selfdatamarkersoutexpectederrors r(test_watershed01zTestWatershed.test_watershed01Fsx%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  H  (&&&%%%%%%%%%%%%%%%%%%%%%%%%%%%  G  gtz228,,,,,,,,,''''''''''''''',,,,,,    Xs##s{{{{{{r*ctjgdgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdgdg tj}t ||}t gdgdgdgdgdgdgdgdgdgdgdg |}||tkd S) z watershed 2r0r1r2r3r5r6)r4r4rrrr4r4r7Nrr8r9r:rr) assertTruer<r=r>r?r@rBs r(test_watershed02zTestWatershed.test_watershed02xsx%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  H   (&&&%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  G   g&&,,,,,,,,,,,,)))'''''''''))),,,,,,      $$$$$r*c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t ||}t gdgdgdgdgdgdgdgdgdgdg |}||tkd S) z watershed 3r0r1rrrrrrrrrrrr.rrrrrrrrr4r6)r4rrrr.rr4r4rrrr.r.r4NrErGs r(test_watershed03zTestWatershed.test_watershed03sx%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  H  (%%%%%%%%%%%%%%%%%%%%%%%%%%%&&&  G  g&&,,,''''''''''''''',,,,,,,,,,,,      $$$$$r*c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j}t gdgdgdgdgdgdgdgdgdgdg |}||tkdS) z watershed 4r0r1rJrKrLr6rMN rr8r9r:rr;r)rFr<rGs r(test_watershed04zTestWatershed.test_watershed04x%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  H  (%%%%%%%%%%%%%%%%%%%%%%%%%%%&&&  G  gtz22,,,''''''''''''''',,,,,,,,,,,,      $$$$$r*c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j}t gdgdgdgdgdgdgdgdgdgdg |}||tkdS) z watershed 5r0r1rJ)rrr.rrrrrLr6)r4r.r.rrrr4NrPrGs r(test_watershed05zTestWatershed.test_watershed05rRr*c tjgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdg tj}t |||j}t gdgdgdgdgdgdgdgdgdg |}||tkdS) z watershed 6r2r1r0r5r3r7r6NrPrGs r(test_watershed06zTestWatershed.test_watershed06Cs}x%%%%%%%%%%%%%%%%%%%%%%%%%%%  H   (%%%%%%%%%%%%%%%%%%%%%%%%&&&  G   gtz22'''''''''''',,,,,,,,,,,,,,,       $$$$$r*cTt}|dk}tj|jt}d|d<d|d<t |||j|}tj|dk}tj|dk}|t||z dkdS) z3A regression test of a competitive case that failedr rrrZmaskrYN) blobrzerosrintrr;rrFabsr=r>r^r?r@size1size2s r(test_watershed07zTestWatershed.test_watershed07rss{(4:s++ gtz=== sax  sax   EEM**a/00000r*ct}d|dddf<|dk}tj|jt }d|d<d|d <t |||j| }tj|dk}tj|dk}| t||z d kd S) z2The border pixels + an edge are all the same valuer rZ r rrXrr[r]rYN) r_copyrr`rrarr;rrFrbrcs r(test_watershed08zTestWatershed.test_watershed08syy{{R1W s{(4:s++ gtz=== sax  sax   EEM**a/00000r*ctjd}tjdddt }tjdt }d}|D]\}}d|||f<||||f<|dz }t |dd}t|||jtj |tj ||jd S) zTest on an image of reasonable size This is here both for timing (does it take forever?) and to ensure that the memory constraints are reasonable )rmrrm)drrreflect)sigmamodeN) rr`randomuniformr!rarrr;ndi watershed_iftuint16)r=imagecoordsr?idxxys r(test_watershed09zTestWatershed.test_watershed09s &&""1dH55<z2TestWatershed.test_watershed11..sa   B%2-!3ahhuoo6Jq5PQQ   r*)rrrrN) rr`rar8rmgriddstackminrFall) r=rxr? structurer@ddminrrs @@r(test_watershed11zTestWatershed.test_watershed11s4""(8S)) H ! ! !#5#5#57K7K7K L  w 22x"ad #1 I     B      va|| qAs1a4y1}!45=>>?????r*ctjgdgdgdgdgdgdgdgdgd gd gd gd gd gdgdgdg}|dk}t|}t||dd}t t dgdD]4\}}|tj||k|k5dS)zThe watershed line)r rr r r r r r r r r r r r r )rr rr r r frrrrrr r r r )rr rrr r rrMrrrr r rr)rr r rr r r rrrrr r rrr )rrr rrrr r r r r r rrr r )r rr r r rrrrrrrrr r r)r rrrr r r r r r r r r r rr)r r r rrrrrr rrrrrrr )rrr r r r rrr rrr rr r r )rrrrrr rr r rrrrr rr )r3rrrr rr rrr r r r rr )rrrrr r rr rrrr rrrr )rrrrr rrr rr rr rrrr )rrrrr rr r rrrr rrrr )rrrrr rr rrr r r rrrr )rrrrr rr rr r r r r r r r rrT) connectivitywatershed_linero)"JrrN)rr8rrziprangerFr)r=r> markerbinmarkerwslabareas r(test_watershed12zTestWatershed.test_watershed12sx$$$$$$$$$$$$$$$_a c c H AI y!! tV!D I I IU1XX'7'7'788 7 7IC OOBF29--5 6 6 6 6 7 7r*ctjd}tjdtj}gd|gdgdf<tj|}t ||}tj||tj ||krJdS)z.Test to ensure input markers are not modified.r)sizer)rrr.ro)rrr)rrrrN) rrs default_rngr`r9rjrtesting assert_equalr)r=rxr?original_markersresults r(!test_watershed_input_not_modifiedz/TestWatershed.test_watershed_input_not_modifieds %%''..H.==(8284442>,,./77++5'**  0':::6&G+,,,,,,,r*N)__name__ __module__ __qualname__ronesboolr;rCrHrNrQrTrVrfrkr}rrrrr*r(r,r,Cs BGFD ! !E000d3%3%3%j0%0%0%d0%0%0%d0%0%0%d-%-%-%^111 111"HHH& % % %@@@8j7j7j7X -----r*r,c|tjd}d|dddf<d|ddddf<tjdt}d|d<d|d<t||d }tjgd gd gd gd gd gt}tj||t||}tjgd gd gd gd gd gt}tj||t||d d }||k}d||dk<tj|rJdS)NrrYrr.rror)rr)rr{Gz? compactness)rrrrrrrrrrrrT)rrFr)rr`rarr8rrany)rxseedscompactrAnormal compact_wsl differences r(test_compact_watershedrs HV  EE!!!Q$KE!!!QRR%L HV3 ' ' 'EE$KE$Ku$777Gx                    HJGX... ue $ $Fx                    HJFH---E5d4PPPK'J#(J{a vj!!!!!!!r*ctjd\}}d\}}}}d\}}||z dz||z dzz|dzk}||z dz||z dzz|dzk} tj|| } tj| } t | tjd| } | dddfxxd z cc<tj| jt } d | t| j <tj | \}}t| || }tj|| }|djdksJtjddt&dd}tj|dj||djdksJ|djdksJdS)a Check edge case behavior reported in gh-6632 While we initially viewed the behavior described in gh-6632 [1]_ as a bug, we have reverted that decision in gh-7661. See [2]_ for an explanation. So this test now actually asserts the behavior reported in gh-6632 as correct. .. [1] https://github.com/scikit-image/scikit-image/issues/6632. .. [2] https://github.com/scikit-image/scikit-image/issues/7661#issuecomment-2645810807 )Pr)r,r)rr-) footprintlabelsNrrYrTr])intensity_imageri)startstoprr4rhi)rindices logical_orrudistance_transform_edtrrr`rrtupleTrrskimagemeasure regionpropsextentarangerreshaperrimage_intensity num_pixels)r{r|x1y1x2y2r1r2 mask_circle1 mask_circle2rxdistanceryr^r?_rpropsexpected_regions r("test_watershed_with_markers_offsetr3s :h  DAq#NBB FBFq=AFq=02q58LFq=AFq=02q58L M, 5 5E )%00H H N N NF 111a4LLLALLL 8HN$ / / /D Dvx4JGQ y' 6 6 6F O ' ' ' J JE 8?a    ic???GGANNOJE!H4oFFF 8 " $ $ $ $ 8 $ & & & & & &r*ctjgdgdg}tjgdgdg}tjgdgdg}t||}tj||tjgd }tjgd}tjgd }t|||d k }tj||d S) a Test edge case behavior when markers spill over into another basin / compete. While we initially viewed the behavior described in gh-6632 [1]_ as a bug, we have reverted that decision in gh-7661. See [2]_ for an explanation. So this test now actually asserts the behavior reported in gh-6632 as correct. .. [1] https://github.com/scikit-image/scikit-image/issues/6632. .. [2] https://github.com/scikit-image/scikit-image/issues/7661#issuecomment-2645810807 ) rYrror.rr.rrr) rrrrrrrrr) rrrrrrrrr) rrrrrrrrr) rrrrrrrrrr?)rrrrrrr.)rrrrrrr)rrrrrrrrr?r^N)rr8rrrassert_array_equalrxr?rArs r(%test_watershed_simple_basin_overspillr`s%x444444677Ex444444677Gx444444677Hug . . .FJFH---X+++ , , ,Eh,,,--Gx---..H ugEQJ ? ? ?FJ!!&(33333r*ctjgd}tjgd}tjgd}t||}tj||tjgd}tjgd}t||}tj||dS)a Edge case: Basins should be distributed evenly between contesting markers. Markers should be prevented from spilling over into another basin and conquering it against other markers with the same claim, just because they get to the basin one step earlier. ) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrrr) rrrrrrrrrrNrr8rrrrs r(+test_watershed_evenly_distributed_overspillrsx66677Ex66677Gx66677H ug . . .FJFH---x66677Ex66677H ug . . .FJFH-----r*ctjgdgdg}tjgdgdg}tjgdgdg}t||}tj||dS) zCheck that markers placed at maxima don't conquer other pixels. Regression test for gh-7661 [1]_. .. [1] https://github.com/scikit-image/scikit-image/issues/7661 )rrrr.rorro)rrrr.rororo)rrrrrrrr0)rrrrrrr)rrrrrrrrNrrs r(test_markers_on_maximarsx......011Ex......011Gx......011H ug . . .FJFH-----r*ctjd}d|ddddf<t|dd}tjgdgdgdgdgdgtj }tj||dS) z>Test that passing just the number of seeds to watershed works.rrNr.rrrrr)rr`rr8int32rr)rxrrAs r(test_numeric_seed_watershedrs HV  EE!!!QRR%Lqd333Gx                 h   HJGX.....r*rctjd}tjd|}t||}|j|jksJdS)N)rnrn)rr`rr)rrxr?r@s r(test_watershed_output_dtypersN HZ Ehz5))G E7 # #C 9 % % % % % %r*ctjd}tjd}tjt5t ||ddddS#1swxYwYdS)NrrrZrrpytestraises ValueErrorr)rxr?s r(test_incorrect_markers_shapers GFOOEgfooG z " """%!!!""""""""""""""""""sA  A$'A$ctjd}tjd}tjt5t |d|ddddS#1swxYwYdS)Nrrrorr)rxr^s r(test_incorrect_mask_shapers GFOOE 76??D z " "//%....//////////////////sA""A&)A&ct}|dk}t|dd|}tj||dksJdS)Nr r)rr^r)r_rrrr>r^r@s r(test_markers_in_maskrsL D 3;D D"14 8 8 8C 6#te*/ " """"""r*cvt}|dk}t||}tj|dksJdS)Nr r]r)r_rrmaxrs r(test_no_markersr sA D 3;D Dt $ $ $C 6#;;!      r*c"tjd\}}d\}}}}}}}} d\} } } } ||z dz||z dzz| dzk}||z dz||z dzz| dzk}||z dz||z dzz| dzk}||z dz|| z dzz| dzk}tj||}tj||}tj||}tj|}tj|dz ddz}tj|dkdd}td |z |dd }td |z |dd }tj|j dd ksJtj|j dd ksJttd gdD]#\}}tj ||k|ksJ$ttd gdD]#\}}tj ||k|ksJ$dS)z Watershed segmentation should output different result for different connectivity when markers are calculated where None is supplied. Issue = 5084 )ii)i,xrnrniTr )rn2(rr r)decimalsrr r)r^rrrYr)EiOiY+ii,)rriOib0i,N) rrrruraroundwhereruniquerrrr)r{r|rrrrx3y3x4y4rrr3r4rr mask_circle3 mask_circle4rxDummyBT DummyBT_disImg_mask labels_c1 labels_c2rrs r(test_connectivityr%sv :j ! !DAq%K"BBBB$NBBFq=AFq=02q58LFq=AFq=02q58LFq=AFq=02q58LFq=AFq=02q58L M, 5 5E M% . .E M% . .E(//G)GbL1555:Kx q(!Q//H kqdI kqdI 9Y   %a (A - - - - 9Y   %a (A - - - -q#K#K#KLL00 Tvi3&''4/////q#E#E#EFF00 Tvi3&''4/////00r*).__doc__r#unittestnumpyrrscipyrruskimage.measurerskimage._shared.filtersrskimage.featurerr _watershedrr<r8r_r)TestCaser,rrrrrrmark parametrizer9r:rwint16uint32ruint64int64rrrrr r%rr*r(r5s66  ,,,,,,******!!!!!!"""""" rxaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)cdd00 - - - - -H% - - -D+"+"+"\*'*'*'Z444@...,..."///$ Xrw 28RY")RXV&& &"""///###*0*0*0*0*0r*