ó  ‰\c@sedZddlZddlZddlmZddlmZddl m Z ddd„ƒYZ dS( s^ Tests for Morphological structuring elements (skimage.morphology.selem) Author: Damian Eads iĸĸĸĸN(tdata_dir(tselem(t assert_equalt TestSElemcBsteZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z RS( cCsUxNtddƒD]=}tj|ƒ}tj||fddƒ}t||ƒqWdS(s Test square structuring elementsiitdtypetuint8N(trangeRtsquaretnptonesR(tselftkt actual_maskt expected_mask((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_square_selemscCsrxktddƒD]Z}xQtddƒD]@}tj||ƒ}tj||fddƒ}t||ƒq&WqWdS(s#Test rectangle structuring elementsiiRRN(RRt rectangleRR R(R titjR R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_rectangle_selems cCsXxQtddƒD]@}tj|ƒ}tj|||fddƒ}t||ƒqWdS(sTest cube structuring elementsiiRRN(RRtcubeRR R(R R R R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_cube_selem"scCs—tjtjjt|ƒƒ}d}xlt|ƒD]^}||}||ƒ}|jdkrx|dd…tjf}nt ||ƒ|d}q1WdS(Nii(i( RtloadtostpathtjoinRtsortedtshapetnewaxisR(R tfntfunct matlab_masksR tarrnameR R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyt strel_worker)s   c Cstjtjjt|ƒƒ}d}xčt|ƒD]Ú}||}||ƒ}|jdkrx|dd…tjf}nt |jddƒ}t |||dd…dd…fƒt ||dd…|dd…fƒt ||dd…dd…|fƒ|d}q1WdS(Niii(i( RRRRRRRRRtintR( R RRRR RR R tc((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pytstrel_worker_3d4s  &&&cCs|jdtjƒdS(sTest disk structuring elementssdisk-matlab-output.npzN(R Rtdisk(R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_diskEscCs|jdtjƒdS(s!Test diamond structuring elementssdiamond-matlab-output.npzN(R Rtdiamond(R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_diamondIscCs|jdtjƒdS(sTest ball structuring elementssdisk-matlab-output.npzN(R#Rtball(R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_ballMscCs|jdtjƒdS(s$Test octahedron structuring elementssdiamond-matlab-output.npzN(R#Rt octahedron(R ((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_octahedronQscCs"tjdddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg g dtjƒ}tjddƒ}tjdddgdddgdddggdtjƒ}tjddƒ}t||ƒt||ƒdS(s!Test octagon structuring elementsiiRiiN(RtarrayRRtoctagonR(R texpected_mask1t actual_mask1texpected_mask2t actual_mask2((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_octagonUs$*$$$$$$$$$6  cCsĘtjdddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg dddddddddddg gdtjƒ}tjddƒ}tjdddgdddgdddggdtjƒ}tjddƒ}t||ƒt||ƒt|tjddƒjƒt|tjddƒjƒdS(s!Test ellipse structuring elementsiiRiiN(RR,RRtellipseRtT(R R.R/R0R1((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_ellipsejs *$$$$$6   cCsētjdddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg dddddddddddddg g dtjƒ}tjdƒ}tjdddgdddgdddggdtjƒ}tjdƒ}t||ƒt||ƒdS(sTest star structuring elementsiiRiN(RR,RRtstarR(R R.R/R0R1((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyttest_selem_star}s*0***********0   (t__name__t __module__RRRR R#R%R'R)R+R2R5R7(((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyRs          (( t__doc__tos.pathRtnumpyRtskimageRtskimage.morphologyRtskimage._shared.testingRR(((sBlib/python2.7/site-packages/skimage/morphology/tests/test_selem.pyts