Q(h ddlZddlZddlmZej j dddddejeejddgd Z ej jd Z d Z y) N) _get_outputdtypef4float32 complex64 complex128cd}tj|d}|dn|}t||}|j|k(sJ|jtj|k(sJt||d}|jdk(sJ|jtj|k(sJtj||}t||}||usJy)Nrr)rr r)shape)npzerosrrr)rrinput_expected_dtyperesultoutputs a/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/ndimage/tests/test_ni_support.pytest_get_output_basicrs E XXe9 -F#(-YUN 'F <<5  <<288N3 33 3f 5F <<6 !! ! <<288N3 33 3XXe5 )F  (F V  c,d}tj|}td|d}|j|k(sJ|jtjdk(sJt j td5tt|d}ddd|j|k(sJ|jtjdk(sJtd|d}|j|k(sJ|jtjdk(sJy#1swYxYw)Nr Tcomplex_outputr z+promoting specified output dtype to complexmatchr) rrrrrpytestwarns UserWarningfloat)rrrs rtest_get_output_complexr!*s E XXe_Fvd ;F <<5  <<288L1 11 1 k)V WAUF4@A <<5  <<288L1 11 1fT BF <<5  <<288K0 00 0AAs 6D  Dcdtjdd}tjtd5t d|ddddtjtd5t tjd|ddddtjtd5t d|dddtjtd 5t tjd |dddy#1swYxYw#1swYxYw#1swYexYw#1swYyxYw) Nr rzoutput must have complex dtyperTrzoutput must have numeric dtypevoidzshape not correctr )rrrraises RuntimeErrorr)rs rtest_get_output_error_casesr&As XXfi (F |+K L<Ivd;< |+K LCBHHV$fTBC |+K L$FF#$ |+> ?.BHHV$f-..<<CC$$..s/D$"D) D D&D DD#&D/)rnumpyr _ni_supportrmark parametrizerr rr thread_unsafer!r&rrr-s %  il E8288D>   011, .r