a =b@sddlZddlZddlmZmZddlmZddlm Z m Z m Z ddl m Z ddlmZeZe ejdgdd d Zejdgdd d Ze e d dddZddZdS)N)assert_almost_equal assert_equal)data) test_parallelxfailarch32)ORB)_convertdtype)float32float64Zuint8Zuint16Zint64cCstt|}tdddd}||tgd}tgd}tgd}tgd}tgd }t|tjkr|jjtjksJ|j jtjksJ|j jtjksJn6|jjtj ksJ|j jtj ksJ|j jtj ksJt ||j ddd ft ||j ddd ft ||jt ||j d t |t|j d |tt ||j ddd ft ||j ddd fdS)N 皙?) n_keypointsfast_nfast_threshold) a@[@gRj@g``@gbX9j@gP@gi@g f@rr) g0t@gt@gp= ףq@g@r@gvq@g@U@g@p@gq@ǧt@p@) rg ףp= ?rgS?rrrg333333?r) gG'JgYk(M@g͊n4XgfvbgBWg^nFgkxIgZ.d\@gTO@gS) gp/?gv?g/C?gU#K?g?gUk/?g_vj.?gjK>ރ?g ?gt(?rr)r imgrdetectnparrayr r scales responses orientationsr r keypointsrad2degdetect_and_extractr Z_imgdetector_extractorZexp_rowsZexp_colsZ exp_scalesZexp_orientationsZ exp_responser%=lib/python3.9/site-packages/skimage/feature/tests/test_orb.py*test_keypoints_orb_desired_no_of_keypointss2     r'cCstt|}tdddddd}||tgd}tgd}tgd}tgd }tgd }t||jddd ft||jddd ft||jt||j t|t |j d | tt||jddd ft||jddd fdS)NrgQ?)rrrZ downscaleZn_scales)rg`i@ga@g@P@gM@)gPr@rg@i@g@`@g0r@)?r*r*r*r*)gk b@glF=DsLgjSgIKMgfc)g7Ԕƿg"-D>?g܌:g㳆?gO?rr) r rrrrrrr rrr!rr"r#r%r%r&4test_keypoints_orb_less_than_desired_no_of_keypoints;s(      r,zKnown test failure on 32-bit platforms. See links for details: https://github.com/scikit-image/scikit-image/issues/3091 https://github.com/scikit-image/scikit-image/issues/2529)Z conditionreasoncCs\tddd}tjgdgdgdgdgdgd gd gd gd gd gdgdgdgdgdgdgdgdgdgdgtd}|t|t|j|j|j t ||j ddddf| tt ||j ddddf|jj d}||j j dksJ||j j dks,J||jj dksBJ||jj dksXJdS)Nrr)rr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr) rrrrrrrrrr)r dxr r)rrrboolrrextractr rrrZ descriptorsr"shaper)r$Zexp_descriptorsZkeypoints_countr%r%r&test_descriptor_orb[sR      r4cCsHtd}t}tt||Wdn1s:0YdS)N)r5)rZonesrpytestZraises RuntimeErrorr")rr$r%r%r&!test_no_descriptors_extracted_orbs  r8)Znumpyrr6Z numpy.testingrrZskimagerZskimage._shared.testingrrrZskimage.featurerZskimage.util.dtyper ZcoinsrZmarkZ parametrizer'r,r4r8r%r%r%r&s*    )  )