
    sg	                         d dl Z d dlZddlmZ e j
                  j                  dddddej                  e ej                  d      dg      d	        Z
d
 Zd Zy)    N   )_get_outputdtypef4float32	complex64
complex128c                    d}t        j                  |d      }| dn| }t        | |      }|j                  |k(  sJ |j                  t        j                  |      k(  sJ t        | |d      }|j                  dk(  sJ |j                  t        j                  |      k(  sJ t        j                  ||       }t        ||      }||u sJ y )Nr      r   r   r   )shape)npzerosr   r   r   )r   r   input_expected_dtyperesultoutputs         V/var/www/html/venv/lib/python3.12/site-packages/scipy/ndimage/tests/test_ni_support.pytest_get_output_basicr      s     EXXeY'F #(-YUN 'F<<5   <<288N3333 f5F<<6!!!<<288N3333 XXeU#F(FV    c                  ,   d} t        j                  |       }t        d |d      }|j                  | k(  sJ |j                  t        j                  d      k(  sJ t        j                  t        d      5  t        t        |d      }d d d        |j                  | k(  sJ |j                  t        j                  d      k(  sJ t        d|d      }|j                  | k(  sJ |j                  t        j                  d      k(  sJ y # 1 sw Y   xY w)Nr   Tcomplex_outputr	   z+promoting specified output dtype to complexmatchr   )	r   r   r   r   r   pytestwarnsUserWarningfloat)r   r   r   s      r   test_get_output_complexr!   *   s    EXXe_F vd;F<<5   <<288L1111 
k)V	W AUF4@A<<5   <<288L1111 fTBF<<5   <<288K0000A As   6D

Dc                  d   t        j                  dd      } t        j                  t        d      5  t        d| d       d d d        t        j                  t        d      5  t        t        j                  d      | d       d d d        t        j                  t        d      5  t        d|        d d d        t        j                  t        d	      5  t        t        j                  d
      |        d d d        y # 1 sw Y   xY w# 1 sw Y   xY w# 1 sw Y   exY w# 1 sw Y   y xY w)Nr   r   zoutput must have complex dtyper   Tr   zoutput must have numeric dtypevoidzshape not correctr   )r   r   r   raisesRuntimeErrorr   )r   s    r   test_get_output_error_casesr&   @   s    XXfi(F 
|+K	L <Ivd;<	|+K	L CBHHV$fTBC 
|+K	L $FF#$ 
|+>	? .BHHV$f-. .< <C C$ $. .s/   D$"D)D D&DDD#&D/)r   numpyr   _ni_supportr   markparametrizer   r    r   r   r!   r&    r   r   <module>r,      sf      %  	il


E8288D>

01,.r   