o jP@svddlZddlmZmZmZddlmZmZddlmZddl m Z ddl m Z m Z dZd d d ZGd d d e ZdS)N)max_tree area_closing area_opening)max_tree_local_maximadiameter_opening)diameter_closing)invert)assert_array_equalTestCaseg-q=FcKsn|||fi|}t||tjtjfD]}||}|||fi|}||} t|| q|tj} | d} |tj} | d} |rJ|d} n|} tjtjfD])}| |} || | fi|}| |} d| d|}|dk}|tks{JqR|tj }|d}|tj }|d}tj tj tj tj fD]}||}|||fi|}||}t||qdS)Ngo@?) r npuint32uint64astypefloat64float32sumepsint16int8int32int64)imgparamexpectedfuncZ param_scalekeywordsoutdtZimg_castZexp_castZ data_floatZexpected_floatZ param_castZ data_castZ error_imgerrorZ img_signedZ exp_signedZimg_sZexp_sr!g/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/skimage/morphology/tests/test_max_tree.py_full_type_test s>               r#c@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS) TestMaxtreec CsZtj}tjgdgdgdgdg|d}tjgdgdgdgd gtjd}tjgd tjd}tjtjtjtjfD]}||}t|d d \}}t ||t ||q>tj tj tj tjfD]}||}|d }t|d d \}}t ||t ||qb|t }|dd}tjtjfD]}||}t|d d \}}t ||t ||qdS)zTest for max tree) r& )r(r'r')r&r(r%r%)r'r'r%r%dtype)r+r+)r,r,r-)r+r,r-r%)r-r-r%r%)r,r'r+r&r-r( rr% r/Z connectivityr'r&g@N)r uint8arrayrZuint16rrrrr rrrfloatrr) selfZimg_typerZP_expZS_expPSZ img_shiftedZ img_floatr!r!r" test_max_tree6s<            zTestMaxtree.test_max_treecCsXtjgdgdgdgdgdgdgdgdgdgd gd gd g tjd }tjgdgd gdgdgdgdgdgdgdgdgdgd g tjd }tjgdgdgdgdgdgdgdgdgdgdgdgd g tjd }t|d|tddt|d|tddtt|dd\}}t|d|t||ddS)z/Test for Area Closing (2 thresholds, all types)) 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>dr>rArAr>r>r?r>) r?r?r?r?r?r?r?r>r?r?rB) r?rBr?r?r?rBr?r>rBrBrBr@) r?r?r?rAr?r?r?r>rBrBrBrB) r?r?r?rAr?r?r?r>r?r?rBrB) r?r?r?r?r?r@r?r>r>rArBrB) r?r@rBrBrBr@r?rBr?r?rBrB) r?r?r?r?r?r?r?rBrBrBrBrBr)) 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>rArAr>r>r?r>) r?rBr?r?r?rBr?r>rBrBrBrB) r?r?r?r?r?r@r?r>r>r?rBrB) r?r?rBrBrBr@r?rBr?r?rBrB) r>r?r?r>r>r>r>r>r>r>r>r>) r?r?r?r?r?r?r?r>r>r>rBrB) r?r?r?r?r?r?r?r>rBrBrBrB) r?r?r?r?r?r?r?r>r>r?rBrB) r?r?rBrBrBr?r?rBr?r?rBrBr/r6r,parentZtree_traverserN)r r8r7r#rrrr:rZ expected_2Z expected_4r;r<r!r!r"test_area_closing]shzTestMaxtree.test_area_closingcCsTtjgdgdgdgdgdgdgdgdgdgd gd gd g tjd }tjgdgd gdgdgdgdgdgdgdgdgdgd g tjd }tjgdgdgdgdgdgdgdgdgdgdgdgd g tjd }t|d|tddt|d|tddt|dd\}}t|d|t||ddS)z/Test for Area Opening (2 thresholds, all types)) r5r5r5r5r5r5r5r5r5r5r5r5) r57rGr5rGr5rGrGr5r5rGr5) r5rGr5r5r5r5r5r5r5rHr5) r5r5r5r5r5rIrIr5r5rGr5) rGrGrGrGrGrGrGr5rGrGrr) rGrrGrGrGrrGr5rrrrH) rGrGrGrIrGrGrGr5rrrr) rGrGrGrIrGrGrGr5rGrGrr) rGrGrGrGrGrHrGr5r5rIrr) rGrHrrrrHrGrrGrGrr) rGrGrGrGrGrGrGrrrrrr)) r5rGrGr5r5r5rGrGr5r5rGr5) r5rGrGr5r5r5r5r5r5r5rGr5) r5r5r5r5r5r5rIrIr5r5rGr5) rGrrGrGrGrrGr5rrrr) rGrGrGrGrGrHrGr5r5rGrr) rGrGrrrrHrGrrGrGrr) r5rGrGr5r5r5r5r5r5r5r5r5) rGrGrGrGrGrGrGr5r5r5rr) rGrGrGrGrGrGrGr5rrrr) rGrGrGrGrGrGrGr5r5rGrr) rGrGrrrrGrGrrGrGrrr/r6r,rCN)r r8r7r#rrrEr!r!r"test_area_openingshzTestMaxtree.test_area_openingcCsXtjgdgdgdgdgdgdgdgdgdgdgd gdg tjd }tjgdgdgdgdgdgdgd gdgdgdgd gdg tjd }tjgdgdgd gd gdgdgd gdgdgdgdgdg tjd }t|d|tddt|d|tddtt|dd\}}t|d|t||ddS)3Test for Diameter Opening (2 thresholds, all types)) a_]\[ZrQrQrPrOrNrM) rMrNrPYXrSrSrSrSrRrPrN) rN?rTrTrTVrUrUW+rWrP) rOrRrSrUUrXTrXrXrWrWrR) rPrSrVrXrYrYSrYrYrXrVrS) rQrSrUrXrYrZrZrZrYrXrUrS) rQrSrUrYrZrZRrZrZrYrUrS) rOrRr\rXrXrYrXrXr-r-rR) rNrPr\r\rVrUrUrUrVrSr-rPr)) rQrSrUrYrZrZrZrZrZrYrUrS) rNrTrTrTrTrUrUrUrVrYrYrP) rOrRrSrUrXrXrYrXrXrYrYrR) rOrRrYrYrXrXrYrXrXrYrYrR) rNrPrYrYrVrUrUrUrVrSrYrPr/r6r,rCN)r r8r7r#rrrrr:rZex2Zex4r;r<r!r!r"test_diameter_closingshz!TestMaxtree.test_diameter_closingcCsBtgdgdgdgdgdgdgdgdgdgdgd gdg }tgdgdgdgdgdgdgd gdgdgdgd gdg }tgdgdgd gd gdgdgd gdgdgd gdgdg }t|d|tddt|d|tddt|dd\}}t|d|t||ddS)rK) r.r(r'r3r1r1r1r1r1r3r'r() r(r%r3r2r4r4r5r4r4r2r3r%) r'r@r@r@r@r_r_r_<r`r3) r3r2r5r_rbrbrar`r`r2) r1r4r_rarbrcrcrbrar_r4) r1r4r_rbrcrcrcrcrcrbr_r4) r1r5r_rbrcrcrcrcrbr_r5) r3r2PrerarbrbrbrarArAr2) r'r3rerer_r_r_r_r_r5rAr3) r1r5r_rbrcrcrcrcrcrbr_r5) r'r@r@r@r@r_r_r_r_rbrbr3) r3r2r5r_rarbrbrbrarbrbr2) r3r2rbrbrarbrbrbrarbrbr2) r'r3rbrbr_r_r_r_r_r5rbr3r/r6r,rCN)r r8r#rrr]r!r!r"test_diameter_opening"sbz!TestMaxtree.test_diameter_openingc CsFtjgdgdgdgdgdgdgdgdgdgdg tjd }tjgd gd gd gd gd gd gd gd gd gd g tjd }tjtjtjtjfD]J}||}t|d d}|dk}t|||j |j kstJt |dks}Jt |\}}t|||d}t|||j |j ksJt |dksJqVdS)z#local maxima for various data types) r%r3r2r4r4r5r4r4r2r3) r3r2r5r_r_r_r_r_r5r2) r2r5r@r@rbrbrbr`r`r5) r4r_r@r@rcrcrcr`r`r_) r4r_rbrcrcrcrcrcrbr_) r5r_rbrcrcrdrcrcrbr_) r4r_rerercrcrcrArAr_) r2r5rererbrbrbrArAr5r) rrrrrrrrrr rrr+r+rrrr+r+r) rrrrrr+rrrrr+r6rr.rCN) r r8r7rrrrrr r*maxr) r:dataexpected_resultr*Z test_datarout_binr;r<r!r!r"test_local_maxima^sR       zTestMaxtree.test_local_maximac Cstjgdgdgdgdgdgdgdgdgdgdg tjd }tjgd gd gd gd gd gd gd gd gdgd g tjd }t|dd}|dk}t||t|dksdJdS)zspecific tests for float type) g?)\(?p= ף?Q?rp333333?rprprorn) rnrorq{Gz?rrrrrrrrrqro) rorq皙?rs ףp= ?rtrt333333?rurq) rprrrsrsRQ?rvrvrururr) rprrrtrvrvrvrvrvrtrr) rqgK?rtrvg&1?g?rvrvrtrr) rprr皙?rwrvrvrvg@r rr) rorqrwrwrtrtrtr r rqr)rgrh) rr+rrr+rrrrr) rrr+r+rrrr+rr) rrr+r+rrrrrrr+r6rr0N)r r8rr7rr ri)r:rjrkrrlr!r!r"test_extrema_floats@    zTestMaxtree.test_extrema_floatcCs\tjdtjd}tjdtjd}d|dddddf<d|d<d|dddddf<d|d<d|dd ddf<d|dd ddf<d|d dd dd f<d|d <d|d dd dd f<|d ddd dd fd7<d|d dd dd f<d|d dd dd f<d|dd dd dd f<d|d<d|d<t|}|d k}t||t|dksJdS)z$tests the detection of maxima in 3D.)r&r&r&r)rAr+r-r?)r/r/r/r.r&r)r+r0r0r/r(2r0rB)r(r(r(N)r Zzerosr7rrr ri)r:rZ local_maximarrlr!r!r"test_3ds*" zTestMaxtree.test_3dN) __name__ __module__ __qualname__r=rFrJr^rfrmrxrzr!r!r!r"r$5s'CC?<0 (r$)F)numpyr Zskimage.morphologyrrrrrrZ skimage.utilrZskimage._shared.testingr r rr#r$r!r!r!r"s   )