o  h @sfddlZddlZddlmZejdddddejee ddgd d Z ejj d d Z d dZ dS)N) _get_outputdtypef4float32 complex64 complex128cCsd}tj|dd}|durdn|}t||}|j|ksJ|jt|ks'Jt||dd}|jdks5J|jt|ks?Jtj||d}t||}||usQJdS)Nrr)rr r)shape)npzerosrr r)rr input_expected_dtyperesultoutputrW/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/ndimage/tests/test_ni_support.pytest_get_output_basics   rcCsd}t|}td|dd}|j|ksJ|jtdksJtjtddtt|dd}Wdn1s8wY|j|ksDJ|jtdksNJtd|dd}|j|ks\J|jtdksfJdS)Nr Tcomplex_outputrz+promoting specified output dtype to complexmatchr) r rrr rpytestwarns UserWarningfloat)r rrrrrtest_get_output_complex*s rcCstdd}tjtddtd|ddWdn1swYtjtddttd|ddWdn1s@wYtjtdd td|Wdn1s\wYtjtd dttd |WddS1s|wYdS) Nr rzoutput must have complex dtyperTrzoutput must have numeric dtypevoidzshape not correctr )r rrraises RuntimeErrorr)rrrrtest_get_output_error_casesAs  "r")rnumpyr _ni_supportrmark parametrizerrrr thread_unsaferr"rrrrs