a dp<@sddlZddlZddlmZddlmZmZm Z ddl m Z m Z m Z ddlmZddlmZddlmZmZddlmZmZejd d Zejd d ZGd ddZGdddZejejejej ej!ej"gZ#ej$%de#ddZ&ddZ'ej ejgZ(ej$%de(ddZ)ddZ*ddZ+ddZ,e-gdgdgd gd!gd"gZ.e-gd#gd$gd%gd&gd'gZ/e-gd(gd)gd*gd+gd,gZ0e-gd-gd.gd/gd0gd1gZ1e-gd2gd(gd3gd4gd5gZ2d6d7Z3d8d9Z4d:d;Z5dej$%d?d@ej$%dAdBdCgdDdEZ7ej$%dgd>ej$%dFdGej$%dHdGej$%dAdBdCgdIdJZ8ej$%dgd>ej$%dKdLej$%dAdCgdMdNZ9ej$%dgd>ej$%dOdPej$%dQdRej$%dAdCgdSdTZ:ej$%dgd>ej$%d?dUej$%dAdBdCgdVdWZ;ej$%dgd>ej$%dKdXej$%dAdCgdYdZZTestEccentricStructuringElements.test_white_tophat_white_pixelcCs6|jD]*}t|j|}t|d|jksJqdSrG)r r r(rCrrHrTrrrtest_black_tophat_black_pixelas z>TestEccentricStructuringElements.test_black_tophat_black_pixelcCs0|jD]$}t|j|}t|dksJqdSrP)r r r'rCrrHrTrrrtest_white_tophat_black_pixelfs z>TestEccentricStructuringElements.test_white_tophat_black_pixelcCs0|jD]$}t|j|}t|dksJqdSrP)r r r(rDrrHrTrrrtest_black_tophat_white_pixelks z>TestEccentricStructuringElements.test_black_tophat_white_pixelN) r,r:r;rFrLrNrOrRrSrUrVrWrXrrrrr<;sr<functioncCstjdd}tgdgdgdgdgdgdgdgdgdgdgdgdgdg tj}|||}||}t||dS)Nr )radius) rrrrrrrrrr) rrr r r r r r rr) rrr r r rrr rr)r r*rarrayrBr)rYr2r/Z im_expectedZim_testrrrtest_default_footprintvs(    r\cCsntdt}d|ddddddf<t|}tjdtd}tdd|ddddddf<t||dS)Nr^r^r r@r?)rzerosboolr r%ndigenerate_binary_structurer)r/openedimage_expectedrrr"test_3d_fallback_default_footprints   "rhcCsNtdt}d|ddddddf<tjdtjd}|||}t||dS)Nr]r r@r_)r`r`r`r?)rrbrcrArBr)rYr/cube new_imagerrrtest_3d_fallback_cube_footprints   rkcCstjdtd}d|dddddf<d|dddddf<d|dddddf<tdgt|}Wdn1sz0Ytdd}tdg(tj|jtj d|d }Wdn1s0Yt ||dS) Nr]r?r r@r!r`raoperator.*deprecated|\A\Z footprint) rrbrcr r r'rdreviewrBrr/rjrnrgrrrtest_3d_fallback_white_tophats (  $rqcCstjdtd}d|dddddf<d|dddddf<d|dddddf<tdgt|}Wdn1sz0Ytdd }tdg(tj|jtj d|d }Wdn1s0Yt ||dS) Nr]r?rr@r!r`rarlr rm) rrArcr r r(rdrerorBrrprrrtest_3d_fallback_black_tophats (  $rrcCstdtj}d|ddddf<d|ddddf<d|d <t|}t|}tdd }tj||d }tj ||d }t ||t ||dS) N) rsr@r_r`r=r>r rm) rrbrBr r%r&rdreZ grey_openingZ grey_closingr)r/rfclosedrnZndimage_openedZndimage_closedrrrtest_2d_ndimage_equivalences    rx)皙? ףp= ?g333333?HzG?zG?)g?)\(?{Gz?Q?RQ?)HzG?(\?g= ףp=?(\?Q?) ףp= ?{Gz?(\?(\?ףp= ?)\(\?g?q= ףp?rQ?)ryr}r{r|r)r}r}r}rr)rrrrr)rrrrr)rrrrr)rzrzr~rr{)rr~rrr)rrrrr)rrrrr)rrrrr)ryryr{r{r|)ryr}r{r}r)r}rrrr)rrrrr)rrrrr)rzrzrzr{r{)rrrrr)rrrrr)rrrrrcCsDtttttttttttttt tt dSrM) rr r#imerodedr$dilatedr%rfr&rwrrrr test_floatsrcCsbtttttttg\}}}}}tt ||tt ||tt ||tt ||dSrM) mapr rrrrfrwrr r#r$r%r&)Zim16Zeroded16Z dilated16Zopened16Zclosed16rrr test_uint16s  rcCstgdgdgdgtj}tdtj}|ddddddf}tgdgdgdgdgd gtj}tgd gdgd gdgd gtj}tj||d t||tj||d t||dS)N)rar@)r^r@r@)r`rar )rarar@)r^rrrr)rrrrr)r^rr^rr@)r^rrarra)rarr@rr@)r@rr@rr )r`rr rr )out)rr[rBrbr r$rr#)r/Z out_array_bigZ out_arrayZexpected_dilationZexpected_erosionrrrtest_discontiguous_out_arrays8    rcCs4tgd}tgd}t|}t||dS)N)r r@r`r@r )r r r@r r )rr[r r#r)r/r8rrrrtest_1d_erosions r)r#r$r&r%r'r(size)r^ decompositionZ separablesequencec CsLtj|dd}tj||d}tt|}|||d}|||d}t||dStValidate footprint decomposition for various shapes. comparison is made to the case without decomposition. Nrrm)r r)getattrr r) rrYrrfootprint_ndarrayrnr3r8rrrrtest_square_decomposition#s    rnrows)r` ncolsc CsPtj||dd}tj|||d}tt|}|||d}|||d} t|| dSr)r rErr r) rrYrrrrrnr3r8rrrrtest_rectangle_decomposition6s    rrZ)r@r`c CsLtj|dd}tj||d}tt|}|||d}|||d}t||dSr)r r*rr r) rrYrZrrrnr3r8rrrrtest_diamond_decompositionKs    rm)rr r`rar1)rr r@r`c Cs|dkrL|dkrLtt tj|||dWdq1s@0YnLtj||dd}tj|||d}tt|}|||d}|||d} t|| dS)rrrNrm)pytestZraises ValueErrorr Zoctagonrr r) rrYrr1rrrnr3r8rrrrtest_octagon_decomposition^s  0   r)rac CsLtj|dd}tj||d}tt|}|||d}|||d}t||dSr)r rirr r) rrYrrrrnr3r8rrrrtest_cube_decompositionvs    r)r`c CsLtj|dd}tj||d}tt|}|||d}|||d}t||dSr)r Z octahedronrr r) rrYrZrrrnr3r8rrrrtest_octahedron_decompositions    r)=numpyrrZscipyrrdZ numpy.testingrrrrrrrZskimage._shared._warningsr Zskimage._shared.testingr Zskimage.morphologyr r Z skimage.utilr rZfixturerrrr<r#r$r%r&r'r(Zgray_functionsmarkZ parametrizer\rhZgray_3d_fallback_functionsrkrqrrrxr[rrrrfrwrrrrrrrrrrrrrrs     "6