o :ήcX@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)imgparamexpectedfunc param_scalekeywordsoutdtimg_castexp_cast data_floatexpected_float param_cast data_cast error_imgerror img_signed exp_signedimg_sexp_sr-M/tmp/pip-target-vg8gfxp4/lib/python/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) r2 )r4r3r3)r2r4r1r1)r3r3r1r1dtype)r7r7)r8r8r9)r7r8r9r1)r9r9r1r1)r8r3r7r2r9r4 rr1 r; connectivityr3r2g@N)r uint8arrayruint16rrrrr rrrfloatrr) selfimg_typerP_expS_expPS img_shifted img_floatr-r-r. test_max_tree9sJ              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)) rQrQrQrQrQrQrQrQrQrQrQ) rQrRrQrRrQrRrRrQrQrRrQ) rQrR(rQrQrQrQrQrQrQrSrQ) rQrQrQrQdrQrTrTrQrQrRrQ) rRrRrRrRrRrRrRrQrRrRrU) rRrUrRrRrRrUrRrQrUrUrUrS) rRrRrRrTrRrRrRrQrUrUrUrU) rRrRrRrTrRrRrRrQrRrRrUrU) rRrRrRrRrRrSrRrQrQrTrUrU) rRrSrUrUrUrSrRrUrRrRrUrU) rRrRrRrRrRrRrRrUrUrUrUrUr5) rQrRrRrQrQrQrRrRrQrQrRrQ) rQrRrRrQrQrQrQrQrQrQrRrQ) rQrQrQrQrQrQrTrTrQrQrRrQ) rRrUrRrRrRrUrRrQrUrUrUrU) rRrRrRrRrRrSrRrQrQrRrUrU) rRrRrUrUrUrSrRrUrRrRrUrU) rQrRrRrQrQrQrQrQrQrQrQrQ) rRrRrRrRrRrRrRrQrQrQrUrU) rRrRrRrRrRrRrRrQrUrUrUrU) rRrRrRrRrRrRrRrQrQrRrUrU) rRrRrUrUrUrRrRrUrRrRrUrUr;rBr8parenttree_traverserN)r rErDr/rrrrHr expected_2 expected_4rLrMr-r-r.test_area_closingasl     zTestMaxtree.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)) rArArArArArArArArArArArA) rA7r]rAr]rAr]r]rArAr]rA) rAr]rArArArArArArAr^rA) rArArArArAr_r_rArAr]rA) r]r]r]r]r]r]r]rAr]r]rr) r]rr]r]r]rr]rArrrr^) r]r]r]r_r]r]r]rArrrr) r]r]r]r_r]r]r]rAr]r]rr) r]r]r]r]r]r^r]rArAr_rr) r]r^rrrr^r]rr]r]rr) r]r]r]r]r]r]r]rrrrrr5) rAr]r]rArArAr]r]rArAr]rA) rAr]r]rArArArArArArAr]rA) rArArArArArAr_r_rArAr]rA) r]rr]r]r]rr]rArrrr) r]r]r]r]r]r^r]rArAr]rr) r]r]rrrr^r]rr]r]rr) rAr]r]rArArArArArArArArA) r]r]r]r]r]r]r]rArArArr) r]r]r]r]r]r]r]rArrrr) r]r]r]r]r]r]r]rArAr]rr) r]r]rrrr]r]rr]r]rrr;rBr8rVN)r rErDr/rrrYr-r-r.test_area_openingsf      zTestMaxtree.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_]\[Zrgrgrfrerdrc) rcrdrfYXririririrhrfrd) rd?rjrjrjVrkrkW+rmrf) rerhrirkUrnTrnrnrmrmrh) rfrirlrnroroSrorornrlri) rgrirkrnrorprprprornrkri) rgrirkrorprpRrprprorkri) rerhrrrnrnrornrnr9r9rh) rdrfrrrrrlrkrkrkrlrir9rfr5) rgrirkrorprprprprprorkri) rdrjrjrjrjrkrkrkrlrororf) rerhrirkrnrnrornrnrororh) rerhrorornrnrornrnrororh) rdrfrororlrkrkrkrlrirorfr;rBr8rVN)r rErDr/rrrrrHrex2ex4rLrMr-r-r.test_diameter_closingsf        z!TestMaxtree.test_diameter_closingcCsBtgdgdgdgdgdgdgdgdgdgdgd gdg }tgdgdgdgdgdgdgd gdgdgdgd gdg }tgdgdgd gd gdgdgd gdgdgd gdgdg }t|d|tddt|d|tddt|dd\}}t|d|t||ddS)ra) r:r4r3r?r=r=r=r=r=r?r3r4) r4r1r?r>r@r@rAr@r@r>r?r1) r3rSrSrSrSrwrwrw<rxr?) r?r>rArwrzrzryrxrxr>) r=r@rwryrzr{r{rzryrwr@) r=r@rwrzr{r{r{r{r{rzrwr@) r=rArwrzr{r{r{r{rzrwrA) r?r>Pr}ryrzrzrzryrTrTr>) r3r?r}r}rwrwrwrwrwrArTr?) r=rArwrzr{r{r{r{r{rzrwrA) r3rSrSrSrSrwrwrwrwrzrzr?) r?r>rArwryrzrzrzryrzrzr>) r?r>rzrzryrzrzrzryrzrzr>) r3r?rzrzrwrwrwrwrwrArzr?r;rBr8rVN)r rEr/rrrsr-r-r.test_diameter_opening sZ     z!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) r1r?r>r@r@rAr@r@r>r?) r?r>rArwrwrwrwrwrAr>) r>rArSrSrzrzrzrxrxrA) r@rwrSrSr{r{r{rxrxrw) r@rwrzr{r{r{r{r{rzrw) rArwrzr{r{r|r{r{rzrw) r@rwr}r}r{r{r{rTrTrw) r>rAr}r}rzrzrzrTrTrAr5 rrrrrrrrrr rrr7r7rrrr7r7r) rrrrrr7rrrrr7rBrr:rVN) r rErDrrrrrr r6maxr) rHdataexpected_resultr6 test_datarout_binrLrMr-r-r.test_local_maximaQsT         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?r333333?rrrr) rrr{Gz?rrrrrr) rr皙?r ףp= ?rr333333?rr) rrrrRQ?rrrrr) rrrrrrrrrr) rgK?rrg&1?g?rrrr) rr皙?rrrrg@r r) rrrrrrrr r rr5rr) rr7rrr7rrrrr) rrr7r7rrrr7rr) rrr7r7rrrrrrr7rBrr<N)r rErrDrr r)rHrrrrr-r-r.test_extrema_float|s<     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.)r2r2r2r5rTr7r9rR)r;r;r;r:r2r)r7r<r<r;r42r<rU)r4r4r4N)r zerosrDrrr r)rHr local_maximarrr-r-r.test_3ds*" zTestMaxtree.test_3dN) __name__ __module__ __qualname__rPr\r`rvr~rrrr-r-r-r.r07s(;P41+ *r0)F)numpyr skimage.morphologyrrrrrr skimage.utilrskimage._shared.testingr r rr/r0r-r-r-r.s   +