sg ddlZddlZddlmZej j dddddejeejddgd Z d 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)Nrrr r)shape)npzerosrrr)rrinput_expected_dtyperesultoutputs V/var/www/html/venv/lib/python3.12/site-packages/scipy/ndimage/tests/test_ni_support.pytest_get_output_basicrs E XXeY 'F#(-YUN 'F <<5  <<288N3 33 3f 5F <<6 !! ! <<288N3 33 3XXeU #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&@s 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 rrr!r&rrr,sf %  il E8288D>   01, .r