U ?hX@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=FcKs`|||f|}t||tjtjfD]0}||}|||f|}||} t|| q$|tj} | d} |tj} | d} |r|d} n|} tjtjfD]N}| |} || | f|}| |} d| d|}|dk}|tkst q|tj }|d}|tj }|d}tj tj tj tj fD]2}||}|||f|}||}t||q(dS)Ngo@?)r npuint32uint64astypefloat64float32sumepsAssertionErrorint16int8int32int64)imgparamexpectedfuncZ param_scalekeywordsoutdtZimg_castZexp_castZ data_floatZexpected_floatZ param_castZ data_castZ error_imgerrorZ img_signedZ exp_signedZimg_sZexp_sr"X/var/www/html/venv/lib/python3.8/site-packages/skimage/morphology/tests/test_max_tree.py_full_type_test s<              r$c@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS) TestMaxtreec Cstj}tjddddgddddgddddgddddgg|d}tjddddgddddgddddgddddggtjd}tjdd ddd ddd dd d dddddgtjd}tjtjtjtjfD]2}||}t|d d\}}t ||t ||qtj tj tj tjfD]<}||}|d}t|d d\}}t ||t ||q|t }|dd}tjtjfD]4}||}t|d d\}}t ||t ||q`dS)zTest for max tree  dtype r Z connectivityg@N)r uint8arrayrZuint16rrrrr rrrfloatrr) selfZimg_typerZP_expZS_expPSZ img_shiftedZ img_floatr"r"r# test_max_tree9sZ               zTestMaxtree.test_max_treecCs(tjddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg g tjd}tjddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg g tjd}tjddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg g tjd}t|d|tddt|d |tddtt|dd\}}t|d |t||d d S) z/Test for Area Closing (2 thresholds, all types)(dr*r0r7r-parentZtree_traverserN)r r9r8r$rrrr;rZ expected_2Z expected_4r<r=r"r"r#test_area_closingasn    zTestMaxtree.test_area_closingcCs$tjddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg g tjd}tjddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg g tjd}tjddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg ddddddddddddg g tjd}t|d|tddt|d |tddt|dd\}}t|d |t||d d S) z/Test for Area Opening (2 thresholds, all types)r67rr*r0r7r-rDN)r r9r8r$rrrFr"r"r#test_area_openings(  zTestMaxtree.test_area_openingcCs(tjddddddddddddg ddddddddddddg dd d d d d d d d d d dg dddd d d dd d d d dg ddd d dddddd d dg ddd d dddddd d dg ddd dddddddd dg ddd d dddddd d dg ddd d dddddd d dg ddddd d dd d dddg ddddd d d d d dddg ddddddddddddg g tjd}tjddddddddddddg ddddddddddddg dd d d d d d d d d d dg dddd d d dd d d d dg ddd d dddddd d dg ddd d dddddd d dg ddd dddddddd dg ddd d dddddd d dg ddd d dddddd d dg ddddd d dd d dddg ddddd d d d d dddg ddddddddddddg g tjd}tjddddddddddddg ddddddddddddg dd d d d d d d d dddg dddd d d dd d dddg ddd d dddddd d dg ddd d dddddd d dg ddd dddddddd dg ddd d dddddd d dg ddd d dddddd d dg ddddd d dd d dddg ddddd d d d d dddg ddddddddddddg g tjd}t|d|tddt|d|tddtt|dd\}}t|d|t||ddS)3Test for Diameter Opening (2 thresholds, all types)a_]\[ZYX?VW+UTSRr.r*r0r7r-rDN)r r9r8r$rrrrr;rZex2Zex4r<r=r"r"r#test_diameter_closingsh    z!TestMaxtree.test_diameter_closingcCstddddddddddddg ddddddd dddddg dd d d d d d d d d d dg ddd d d dddd d d dg ddd d dddddd d dg ddd dddddddd dg dd d dddddddd d g ddd dddddddd dg ddd d dddddd d dg ddddd dddd dddg ddddd d d d d d ddg ddddddd dddddg g }tddddddddddddg ddddddd dddddg dd d d d d d d d d d dg ddd d d dddd d d dg ddd d dddddd d dg ddd dddddddd dg dd d dddddddd d g ddd dddddddd dg ddd d dddddd d dg ddddd dddd dddg ddddd d d d d d ddg ddddddd dddddg g }tddddddddddddg ddddddd dddddg dd d d d d d d d dddg ddd d d dddd dddg ddd d dddddd d dg ddd dddddddd dg dd d dddddddd d g ddd dddddddd dg ddd d dddddd d dg ddddd dddd dddg ddddd d d d d d ddg ddddddd dddddg g }t|d|tddt|d|tddt|dd\}}t|d|t||ddS)rLr/r)r(r4r2r&r3r5r6rA<PrBr0r7r-rDN)r r9r$rrr^r"r"r#test_diameter_opening s\   z!TestMaxtree.test_diameter_openingc Cstjddddddddddg 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]}||}t|dd}|dk}t|||j |j ks*t t |dks>t t |\}}t|||d}t|||j |j kstt t |dkst qdS)z#local maxima for various data typesr&r4r3r5r6r`rArcrardrerfrBr*rr,r7r/rDN) r r9r8rrrrrr r+rmaxr) r;dataexpected_resultr+Z test_datarout_binr<r=r"r"r#test_local_maximaQsR         zTestMaxtree.test_local_maximacCstjddddddddddg 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}tjddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg ddddddddddg g tjd}t|dd}|dk}t||t|dks tdS)zspecific tests for float typeg?g)\(?gp= ף?gQ?g333333?g{Gz?g?g ףp= ?g333333?gRQ?gK?g&1?g?g?g@r r*rr,r7r1N)r r9rr8rr rhr)r;rirjrrkr"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|dksZtdS)z$tests the detection of maxima in 3D.)r'r'r'r*rBr,r.r@)r0r0r0r/r'r)r,r1r1r0r)2r1rC)r)r)r)N)r Zzerosr8rrr rhr)r;rZ local_maximarrkr"r"r#test_3ds*" zTestMaxtree.test_3dN) __name__ __module__ __qualname__r>rGrKr_rgrlrmror"r"r"r#r%7s(;P41+*r%)F)numpyr Zskimage.morphologyrrrrrrZ skimage.utilrZskimage._shared.testingr r rr$r%r"r"r"r#s   +