U ?hh@s|ddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z dZ ddZGd d d ZGd d d ejZdS) N) assert_equal)raiseswarns)expected_warnings)extrema-q=cCs:tj|tjd}tj|tjd}||d}t|S)Ndtype)npZasarrayfloat64summathsqrt)abtrW/var/www/html/venv/lib/python3.8/site-packages/skimage/morphology/tests/test_extrema.pydiffsrc@s\eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS) TestExtremacCsntjddddgddddgdd d d ggtjd }t|d}tjd dddgddddgddddggtjd }t||}|tks~tt|d}tjddd d gddddgddddggtjd }t||}|tksttjddgdd ggtj d }t|d }tjddgd d!ggtj d }t||}|tks*tt|d }tjdd"gddggtj d }t||}|tksjtd#S)$z*Adding/subtracting a constant and clippingd r h`riiiiiiN) r arrayuint8rZ_add_constant_cliprepsAssertionErrorZ_subtract_constant_clipint16)selfdataZimg_constant_addedexpectederrorZimg_constant_subtractedrrrtest_saturated_arithmeticsZ                   z%TestExtrema.test_saturated_arithmeticcCs&tjddddddddddg ddddddddddg dddddddd d dg ddddd d d d d dg dddd d d d d ddg dddd d d d d ddg dddd d d d d ddg ddd d d d d d d dg ddd d dddd d dg ddddddddddg g tjd}tjddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg g tjd}tjtjtjtjfD]4}||}t|d}t ||}|t kst qdS)zh-maxima for various data typesr r((<Prrrr N) r r1r2uint64int8int64astyperh_maximarr3r4r6r7expected_resultr outr9rrr test_h_maxima@s>       zTestExtrema.test_h_maximacCs@tjddddddddddg ddddddddddg dddddddd d dg ddddd d d d d dg dddd d d d d ddg dddd d d d d ddg dddd d d d d ddg ddd d d d d d d dg ddd d dddd d dg ddddddddddg g tjd}d |}tjddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg g tjd}tjtjtjtjfD]F}||}t|d}t ||}|t ks&t |j |j kst qdS)zh-minima for various data typesrr;r<r(r=r>r?r@rArBrCrDrrrr N) r r1r2rErFrGrHrh_minimarr3r4r rJrrr test_h_minimaasB       zTestExtrema.test_h_minimacCs,tjddddddddddg ddddddddddg dddddddd d dg ddddd d d d d dg dddd d d d d ddg dd dd d d d d ddg dddd d d d d ddg ddddd d d dddg ddddddddddg ddddddddddg g tjd}d|}t|d}tjddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg g tjd}t||}|tkstt |d}t||}|tks(tdS)zspecific tests for float type皙?)\(?p= ף?Q?333333?{Gz?皙? ףp= ?333333?RQ?粝K?&1?皙?皙??rg~jth?rr N) r r1float32rrIr2rr3r4rN)r6r7Z inverted_datarLrKr9rrrtest_extrema_floats     zTestExtrema.test_extrema_floatc Csd}tjd|d|f\}}dd||dd||dd}d|ddddf<d|ddd d f<d |d d ddf<d |d d d d f<|tj}t|}d ||dk<dD]&}t||}t||}|tkst qdS)z,specific tests for h-maxima float image typerrrCr\r r?rrAr)r$rDrr^gfffff3@rgư>gMbP?g{Gz?rPrPN) r mgridrHr_ zeros_likerrIrr3r4 r6wxyr7rKhrLr9rrrtest_h_maxima_float_images(     z%TestExtrema.test_h_maxima_float_imagec Cstjdddddgdddddgdddddgdddddgdddddggtjd}tddd}d}|D]P}|ddkrzd g}ng}t|t||}W5QRX|d dkrb|d7}qb|dkstd S) z-specific tests for h-maxima float h parameterrr!rrr^@rr (possible precision loss converting imager r N)r r1r2linspacerrrIr4)r6r7h_valsfailuresrhmsgsmaximarrrtest_h_maxima_float_hs&        z!TestExtrema.test_h_maxima_float_hc Cstjdddddgdddddgdddddgdddddgdddddggtjd}t|d}t|dksjttjdddddgdddddgdddddgdddddgdddddggtjd}t|d}t|dkstdS) z.test that h-maxima works correctly for large hrr<r(rrr@N)r r1r2rrIr r4r_r6r7rqrrrtest_h_maxima_large_hs(          z!TestExtrema.test_h_maxima_large_hc Csd}tjd|d|f\}}dd||dd||dd}d|ddddf<d|ddd d f<d |d d ddf<d |d d d d f<|tj}t|}d ||dk<dD]&}t||}t||}|tkst qdS)z,specific tests for h-minima float image typerrr\r rr)r$xrr^g33333f@raN) r rbrHr_rcrrNrr3r4rdrrrtest_h_minima_float_images(     z%TestExtrema.test_h_minima_float_imagec Cstjdddddgdddddgdddddgdddddgdddddggtjd}tddd}d}|D]P}|ddkrzdg}ng}t|t||}W5QRX|d dkrb|d7}qb|dkstd S) z-specific tests for h-minima float h parameterrr rrr^rjrrkrlN)r r1r2rmrrrNr4)r6r7rnrorhrpZminimarrrtest_h_minima_float_hs&        z!TestExtrema.test_h_minima_float_hc Cstjdddddgdddddgdddddgdddddgdddddggtjd}t|d}t|dksjttjdddddgdddddgdddddgdddddgdddddggtjd}t|d}t|dkstdS) z.test that h-minima works correctly for large hr(r;rrrrrsN)r r1r2rrNr r4r_rtrrrtest_h_minima_large_hs(          z!TestExtrema.test_h_minima_large_hN) __name__ __module__ __qualname__r:rMrOr`rirrrurzr{r|rrrrrs(!"-rc@seZdZdZejejejejej ej ej ej ej ejg ZejdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddggejdZejdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddggedZejdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddggedZdd Zd d Zd d ZddZddZddZddZddZddZddZddZddZ d S)!TestLocalMaximaz1Some tests for local_minima are included as well.r rr r!rrcCs tjtggdd}|jdks$t|jtks2t|jdks@ttjtgdd}t |t sbtt |dksrt|djdkst|djtj ksttjtggdd}t |t stt |dkst|djdkst|djtj kst|djdkst|djtj kstdS) zTest result with empty image.F)indicesr)r rTr r N) r local_maximar r1sizer4r boolshape isinstancetuplelenintp)r6resultrrr test_emptyVs zTestLocalMaxima.test_emptycCs<|jD]0}t|j|}|jtks*tt||j qdS)zATest results with default configuration for all supported dtypes.N) supported_dtypesrrimagerHr rr4rexpected_default)r6r rrrr test_dtypesks zTestLocalMaxima.test_dtypescCstjddddddddddg ddddddddddg dddddddd d dg ddddd d d d d dg dddd d d d d ddg dddd d d d d ddg dddd d d d d ddg ddd d d d d d d dg ddd d dddd d dg ddddddddddg g tjd}tjddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg g td}|jD]4}||}t|}|jtkst t ||qdS)z Test results with default configuration and data copied from old unit tests for all supported dtypes. rr;r<r(r=r>r?r@rArBrCrDrrrr N) r r1r2rrrHrrr r4r)r6r7r8r rrrrrtest_dtypes_oldrsB      zTestLocalMaxima.test_dtypes_oldcCstj|jdd}|jtkstt||jtj|jdd}|jtksHtt||jtj|jdd}|jtksrtt||jdS)z&Test results if footprint is a scalar.r ) connectivityr r!N) rrrr rr4rexpected_crossr)r6 result_conn1 result_conn2Z result_conn3rrrtest_connectivitys  z!TestLocalMaxima.test_connectivitycCstjdddgdddgdddggtd}tj|j|d}|jtksDtt||j dtj dtj dtj dtj dtj dtdfD].}tj|j|d}|jtkstt||j q~tjdddgdddgdddggtd}tjdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddggtd}tj|j|d}|jtkstt||dS)z#Test results if footprint is given.rr r footprint)TTTrr)r!r!N)r r1rrrrr r4rronesr r2r)r6Zfootprint_crossZresult_footprint_crossrZresult_footprint_squareZ footprint_xZexpected_footprint_xZresult_footprint_xrrrtest_footprintsJ  &       zTestLocalMaxima.test_footprintcCsTt|j}tj|jddd}t||t|j}tj|jddd}t||dS)z,Test output if indices of peaks are desired.r T)rrr N)r Znonzerorrrrrr)r6Zexpected_conn1rZexpected_conn2rrrr test_indicess     zTestLocalMaxima.test_indicescCs*tj|jddd}|jtks tt||jtj dddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddgdddddddddddddddggtd}tj|jddd}|jtkstt||dS)z*Test maxima detection at the image border.r T)r allow_bordersrrFN) rrrr rr4rrr r1)r6Zresult_with_boderZexpected_without_borderZresult_without_borderrrrtest_allow_borderss0        z"TestLocalMaxima.test_allow_bordersc Cstdddddddddddg }tjdddddddddddg td}t|}|jtks\tt||tjdtj d}tjdtd}d|dddddf<d|d <d|dddddf<d|d <d|d d ddf<d|d d ddf<d|dd d d d f<d|d <d|dd d d d f<|ddd d d d fd7<d|dd d d d f<d|dd d d d f<d|dd dd dd f<d|d<d|d<t|}|jtkstt||dS)z%Test one- and three-dimensional case.r rr r!r)r'r'r'rr)r r r rr')r r0r0r)2r0r#)r)r)r)N) r r1rrrr r4rzerosr2)r6Zx_1dZ expected_1dZ result_1dZx_3dZ expected_3dZ result_3drrrtest_nds6   " zTestLocalMaxima.test_ndcCstjddtjd}tjdtjd}|jD]R}||}t|}|jt ksNt t ||t |}|jt kspt t ||q(dS)z!Test behaviour for 'flat' images.)r)r0*rN) r fullr2rrrHrrr rr4r local_minima)r6Z const_imager8r rrrr test_constants     zTestLocalMaxima.test_constantcCs*tjddddddddddg ddddddddddg dddddddd d dg ddddd d d d d dg dddd d d d d ddg dd dd d d d d ddg dddd d d d d ddg ddddd d d dddg ddddddddddg ddddddddddg g tjd}d|}tjddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg g td}t|}|jtkstt||t |}|jtkstt||dS)zSpecific tests for float type.rPrQrRrSrTrUrVrWrXrYrZr[r\r]r^rrr N) r r1r_rrrr r4rr)r6rZinverted_imagerKrrrrr`&sF      z"TestLocalMaxima.test_extrema_floatc Csttdd tj|jtjdtddW5QRXttdd tj|jtjdtddW5QRXttdd tj|jtjdtddW5QRXttdd tj|jtjd tddW5QRXttd dttj d tj dW5QRXd S) z5Test if input validation triggers correct exceptions.znumber of dimensionsmatch)r!r!r!rr)r!zdimension size)r r!)rrzfloat16 which is not supportedr N) r ValueErrorrrrr rr TypeErroremptyZfloat16)r6rrrtest_exceptionsOs,    zTestLocalMaxima.test_exceptionsc Csd}tddg}tj|ddtt|dtj|dd}W5QRXt|ddg|jtksbt tddgddgg}tj|ddd tt|dtj|ddd }W5QRXt|tj d tj d |djtj kst |djtj kst d S) aTest output for arrays with dimension smaller 3. If any dimension of an array is smaller than 3 and `allow_borders` is false a footprint, which has at least 3 elements in each dimension, can't be applied. This is an implementation detail so `local_maxima` should still return valid output (see gh-3261). If `allow_borders` is true the array is padded internally and there is no problem. z0maxima can't exist .* any dimension smaller 3 .*rr T)rrFr )rr)r rrN) r r1rrr UserWarningrr rr4rr)r6Z warning_msgrfrrrrtest_small_arrayds z TestLocalMaxima.test_small_arrayN)!r}r~r__doc__r r2Zuint16Zuint32rErFr5Zint32rGr_r rr1rrrrrrrrrrrrrr`rrrrrrr0sl                     %&$)r)rZunittestnumpyr Z numpy.testingrZpytestrrZskimage._shared.testingrZskimage.morphologyrr3rrZTestCaserrrrrs