U ?h)@sdZddlZddlZddlmZddlmZddlm Z GdddZ ej de j d d fe jd d fe jd d fe jd d fe jd d fe jd d fe jd d fe jd d fe jd d fe jd d fg ej d ejejgddZej dddgej dddddddddddddd d!gd"d#Zej dddddddddd g ej d$d d gd%d&Zej d'ddddgej d(ddddgd)d*Zd+d,Zd-d.ZdS)/zX Tests for Morphological footprints (skimage.morphology.footprint) Author: Damian Eads N) assert_equal)fetch) footprintsc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS)TestFootprintscCs:tddD]*}t|}tj||fdd}t||q dS)zTest square footprintsruint8dtypeN)rangersquarenponesrselfk actual_mask expected_maskrZ/var/www/html/venv/lib/python3.8/site-packages/skimage/morphology/tests/test_footprints.pytest_square_footprints z$TestFootprints.test_square_footprintcCsLtddD]<}tddD],}t||}tj||fdd}t||qq dS)zTest rectangle footprintsrrrrN)r r rectangler r r)rijrrrrrtest_rectangle_footprints  z'TestFootprints.test_rectangle_footprintcCs<tddD],}t|}tj|||fdd}t||q dS)zTest cube footprintsrrrrN)r rcuber r rrrrrtest_cube_footprint s z"TestFootprints.test_cube_footprintcCsbtt|}d}t|D]B}||}||}|jdkrJ|ddtjf}t|||d}qdS)Nrr)r loadrsortedshapenewaxisr)rfnfunc matlab_masksrarrnamerrrrr strel_worker's   zTestFootprints.strel_workerc Cstt|}d}t|D]}||}||}|jdkrJ|ddtjf}t|jdd}t|||ddddft||dd|ddft||dddd|f|d}qdS)Nrrr)r rrrr r!intr) rr"r#r$rr%rrcrrrstrel_worker_3d2s  zTestFootprints.strel_worker_3dcCs|dtjdS)zTest disk footprintsdata/disk-matlab-output.npzN)r&rdiskrrrrtest_footprint_diskCsz"TestFootprints.test_footprint_diskcCs|dtjdS)zTest diamond footprintsdata/diamond-matlab-output.npzN)r&rdiamondr-rrrtest_footprint_diamondGsz%TestFootprints.test_footprint_diamondcCs|dtjdS)zTest ball footprintsr+N)r*rballr-rrrtest_footprint_ballKsz"TestFootprints.test_footprint_ballcCs|dtjdS)zTest octahedron footprintsr/N)r*r octahedronr-rrrtest_footprint_octahedronOsz(TestFootprints.test_footprint_octahedroncCsptjdddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg g tjd}tdd}tjdddgdddgdddggtjd}tdd}t||t||dS)zTest octagon footprintsrrrrN)r arrayrroctagonrrZexpected_mask1Z actual_mask1Zexpected_mask2Z actual_mask2rrrtest_footprint_octagonTs0     z%TestFootprints.test_footprint_octagoncCs8tjdddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg gtjd}tdd}tjdddgdddgdddggtjd}tdd}t||t||t|tddjt|tddjdS)zTest ellipse footprintsrrrrr6N)r r7rrellipserTr9rrrtest_footprint_ellipsejs,     z%TestFootprints.test_footprint_ellipsecCstjdddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg g tjd}td}tjdddgdddgdddggtjd}td}t||t||dS)zTest star footprintsrrrN)r r7rrstarrr9rrrtest_footprint_star~s4      z"TestFootprints.test_footprint_starN)__name__ __module__ __qualname__rrrr&r*r.r1r3r5r:r=r@rrrrrs rz/function, args, supports_sequence_decomposition)r6T)r6r>Fr csN||di}|jkst|rJ||dd}tfdd|DsJtdS)Nr sequence)r decompositioncsg|]}|djkqS)rr).0Zfp_tuplerrr sz(test_footprint_dtype..)r AssertionErrorall)functionargsZsupports_sequence_decompositionr Z footprintrDrrrtest_footprint_dtypes rLrJr,r2radiusrr'r6r>r 2KdcCstt|}||ddd}||ddd}t|}|j|jks@ttt|t |t }|dkrv|dkstn"|dkrdnd}||j |kstdS) NF strict_radiusrErDrrr,g?g333333?) getattrrfootprint_from_sequencer rHr sumabsastyper(size)rJrMfp_funcexpectedfootprint_sequence approximateerror max_errorrrr!test_nsphere_series_approximations   rerXcCsvtj}|||dd}|||dd}t|}|j|jkssH               "