a d@sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z eZ ejddd Zd d Zd d ZddZddZeddZddZdS)N)assert_array_equal)moon)CENSURE)run_in_parallel)testing)rescalecCs8tjdd}tjdd}t|t|dS)z>Censure feature detector should work on 2D image of any shape.i,N)nprandomZrandrdetect)Z rect_imageZ square_imager k/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/feature/tests/test_censure.py"test_censure_on_rectangular_imagess rcCs@tt"ttdWdn1s20YdS)z?Censure keypoints can be extracted from gray-scale images only.)rN)rraises ValueErrorrr r zerosr r r r 4test_keypoints_censure_color_image_unsupported_errors rcCs8tttddWdn1s*0YdS)zJMode argument in keypoints_censure can be either DoB, Octagon or STAR.dummymodeNrrrrr r r r *test_keypoints_censure_mode_validity_errors rcCs:tttdddWdn1s,0YdS)z~Difference between the the max_scale and min_scale parameters in keypoints_censure should be greater than or equal to two.)Z min_scaleZ max_scaleNrr r r r (test_keypoints_censure_scale_range_error#s rc Cszt}|ttddgddgddgddgd d gd d gd dgddgddgg }tgd}t||jt||jdS)znVerify the actual Censure keypoints and their corresponding scale with the expected values for DoB filter.i$.wi^iieitiii) rr'rrrrrrN)rr imgr arrayr keypointsscalesdetectorZexpected_keypointsZexpected_scalesr r r %test_keypoints_censure_moon_image_dob*s    r.cCsrtdd}|ttddddtddgd d gd d gd dgddgg}tgd}t||jt||jdS)zrVerify the actual Censure keypoints and their corresponding scale with the expected values for Octagon filter.Zoctagonr?FconstantZ anti_aliasingrYWj;oC)rrrr'N rr rr(r r)rr*r+r,r r r )test_keypoints_censure_moon_image_octagon>s    r>c Cs~tdd}|ttddddtddgd d gd d gd dgddgddgddgg}tgd}t||jt||jdS)zoVerify the actual Censure keypoints and their corresponding scale with the expected values for STAR filter.Zstarrr/Fr0r1r2r3r4r5Vkr9m@r:r;qF)rrr'rr<rrNr=r,r r r &test_keypoints_censure_moon_image_starTs     rF)numpyr Zskimage._shared.testingrZ skimage.datarZskimage.featurerrZskimage._sharedrZskimage.transformrr(r seedrrrrr.r>rFr r r r s