a6dFWRdZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZdZejgd gd gd gd gd gd gdgdgdgdgdgdgdgdgdgd gd gd gd gd gd gZdZGddejZdZdZdZdZdZdZdZdS)z1test_watershed.py - tests the watershed function N)ndimage)gaussian)label) watershedg-q=)rrrrrrrrrrrrrrr)rrrrrr r r r r rrrrr)rrrr r r r r r r r rrr)rrr r r ogrr r r r r rr)rrr r r ^H4rrrr r r rr)rrr r r r'rrrr r r rr)rrr r r r rrrrr r r r rr)rrrr r r r rrr r r r rrr)rrrrr r r rrr r r rrrr)rrrrrr r rrr r rrrrrct|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 Ilib/python3.11/site-packages/skimage/segmentation/tests/test_watershed.pydiffr('sb a $ $ JqMM a $ $ JqMM QW 1<<sAa1fI BL"-000 BL"-000 Jq"- ( ( Jq"- ( (fqvo ! & & ( (QVaf_q,@+E+E+G+G G JqMM HHRZ  JqMM HHRZ !eaZ     9Q<<cxeZdZejdeZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdS) TestWatershed)r,c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j}tjgdgdgdgdgdgdgdgdgdgdg }t ||}|tksJdS) z watershed 1rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr2r2r2r2r2r2r2r2rrrrrr2N)rarrayuint8int8reightr(eps)selfdatamarkersoutexpectederrors r'test_watershed01zTestWatershed.test_watershed01@sx...111111444444444444444444444 68:x A A(333444444"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9"9 ;"$ * *gtz228999222222222222222222222222222 4 5 5Xs##s{{{{{{r)ctjgdgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdgdg tj}t ||}t gdgdgdgdgdgdgdgdgdgdgdg |}||tkd S) z watershed 2r.r/r0r1r3r4)r2r2rrrr2r2r5Nrr6r7r8rr( assertTruer:r;r<r=r>r@s r'test_watershed02zTestWatershed.test_watershed02esx................................. 024 ; ;(222111111111111111111111111111111 357G = =g&&222222222222222222222222222222222 469 : :  $$$$$r)c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t ||}t gdgdgdgdgdgdgdgdgdgdg |}||tkd S) z watershed 3r.r/rrrrrrrrrrrr,rrrrrrrrr2r4)r2rrrr,rr2r2rrrr,r,r2NrCrEs r'test_watershed03zTestWatershed.test_watershed03sx.............................. 024 ; ;(111111111111111111111111111222 468W > >g&&222222222222222222222222222222 469 : :  $$$$$r)c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j}t gdgdgdgdgdgdgdgdgdgdg |}||tkdS) z watershed 4r.r/rHrIrJr4rKN rr6r7r8rr9r(rDr:rEs r'test_watershed04zTestWatershed.test_watershed04sx...444444444444444444444444444 68:x A A(111111111111111111111111111222 468W > >gtz22222222222222222222222222222222 469 : :  $$$$$r)c tjgdgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdgdg tj}t |||j}t gdgdgdgdgdgdgdgdgdgdg |}||tkdS) z watershed 5r.r/rH)rrr,rrrrrJr4)r2r,r,rrrr2NrNrEs r'test_watershed05zTestWatershed.test_watershed05sx.............................. 024 ; ;(111111111111111111111111111222 468W > >gtz22222222222222222222222222222222 469 : :  $$$$$r)c tjgdgdgdgdgdgdgdgdgdg tj}tjgdgdgdgdgdgdgdgdgdg tj}t |||j}t gdgdgdgdgdgdgdgdgdg |}||tkdS) z watershed 6r0r/r.r3r1r5r4NrNrEs r'test_watershed06zTestWatershed.test_watershed06s}x...........................024;;(111111111111111111111111222468W>>gtz22222222222222222222222222222469::  $$$$$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 failedrrrrWmaskrVN) blobrzerosrintrr9rrDabsr;r<r[r=r>size1size2s r'test_watershed07zTestWatershed.test_watershed07s (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 rW rrrUrrXrZrVN) r\copyrr]rr^rr9rrDr_r`s r'test_watershed08zTestWatershed.test_watershed08$syy{{R1W  (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 )rjrrj)drrreflect)modeN) rr]randomuniformr r^rrr9ndi watershed_iftuint16)r;imagecoordsr=idxxys r'test_watershed09zTestWatershed.test_watershed095s &&""1dH55<z2TestWatershed.test_watershed11..msg A A ARWahhuoob(1,qxxr/AA.E F F A A Ar))r~rrrN) rr]r^r6rmgriddstackminrDall) r;rtr= structurer>ddminrrs @@r'test_watershed11zTestWatershed.test_watershed11Ys@""(8S)) H222000222455 w 22x"ad #1 I A A A A A? A A A B Bva|| qAs1a4y{!23t;<<=====r)ctjgdgdgdgdgdgdgdgdgd gd gd gd gd gdgdgdg}|dk}t|}t||dd}t t dgdD]4\}}|tj||k|k5dS)zThe watershed line)rrr r r r r r r r r r r r r )rrrr r r frrrrrr r r r )rrrrr r rrMrrrr r rr)rrrrr r r rrrrr r rrr)rrrrrrr r r r r r rrrr)r rrrrrrrrrrrrrrr)r rrrrrrrrrrrrrrr)r r r rrrrrrrrrrrrr )rrr r r r rrrrrrrr r r )rrrrrr rrrrrrrr rr )r3rrrr rrrrr r r r rr )rrrrr r rrrrrr rrrr )rrrrr rrrrrrr rrrr )rrrrr rrrrrrr rrrr )rrrrr rrrrr r r rrrr )rrrrr rrrr r r r r r r r rrT) connectivitywatershed_linerl)"JrrN)rr6rrziprangerDr)r;r< markerbinmarkerwslabareas r'test_watershed12zTestWatershed.test_watershed12ss2xiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiikll 1W y!! tV!D I I IU1XX}}}55 7 7IC OOBF29--5 6 6 6 6 7 7r)N)__name__ __module__ __qualname__ronesboolr9rArFrLrOrQrSrcrhryr{rrr)r'r+r+=s BGFD ! !E###J$%$%$%L!%!%!%F!%!%!%F!%!%!%F%%%@111 111"HHH&%%%">>>477777r)r+ctjd}d|ddddf<tjdt}d|d<d|d<t||d }tjgd gd gd gd gd gt}tj||t||}tj|jt}d|dddf<tj||dS) NrrVrr,r)rrr)rr,{Gz? compactness)rrrrrr) rr]r^rr6testing assert_equalrr)rtseedscompactr?normals r'test_compact_watershedrs HV  EE!!!QRR%L HV3 ' ' 'EE$KE$Ku$777Gx+++++++++++++++ -58 999H JGX... ue $ $Fwu{#...HHQUOJFH-----r)ctjd}d|ddddf<t|dd}tjgdgdgdgdgdgtj }tj||dS) z>Test that passing just the number of seeds to watershed works.rrNr,rrr)rrrrrrr)rr]rr6int32rr)rtrr?s r'test_numeric_seed_watershedrs HV  EE!!!QRR%Lqd333Gx+++++++++++++++ -57H >>>H JGX.....r)ctjd}tjd}tjt5t ||ddddS#1swxYwYdS)NrrrWrrpytestraises ValueErrorr)rtr=s r'test_incorrect_markers_shapers GFOOEgfooG z " """%!!!""""""""""""""""""sA  A$'A$ctjd}tjd}tjt5t |d|ddddS#1swxYwYdS)Nrrrl)r=r[r)rtr[s r'test_incorrect_mask_shapers GFOOE 76??D z " "//%....//////////////////sA""A&)A&ct}|dk}t|dd|}tj||dksJdS)Nrr)rr[r)r\rrrr<r[r>s r'test_markers_in_maskrsL D CKD D"14 8 8 8C 6#te*/ " """"""r)cvt}|dk}t||}tj|dksJdS)NrrZr)r\rrmaxrs r'test_no_markersrsA D CKD 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,xrkrkiTr)rk2(Pr r)decimalsrrr)r[rrrVr)EiOiY+ii,)rriOib0i,N) rindices logical_orrqdistance_transform_edtaroundwhereruniquerrrr)rwrxx1y1x2y2x3y3x4y4r1r2r3r4 mask_circle1 mask_circle2 mask_circle3 mask_circle4rtDummyBT DummyBT_disImg_mask labels_c1 labels_c2rrs r'test_connectivityrst :j ! !DAq%K"BBBB$NBBFQ;!b&1,r1u4LFQ;!b&1,r1u4LFQ;!b&1,r1u4LFQ;!b&1,r1u4L M, 5 5E M% . .E M% . .E(//G)GbLQ777:Kx q(!Q//H# +(&*,,,I# +(&*,,,I 9Y   %a (A - - - - 9Y   %a (A - - - -q#K#K#KLL00 Tvi3&''4/////q#E#E#EFF00 Tvi3&''4/////00r))__doc__r"unittestnumpyrrscipyrrqskimage._shared.filtersrskimage.measurer _watershedrr:r6r\r(TestCaser+rrrrrrrrr)r'rs  ,,,,,,!!!!!!"""""" rxaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa)cdd.,L7L7L7L7L7H%L7L7L7` ...& / / /"""///###(0(0(0(0(0r)