
    ^Mh                     b    d dl Zd dlmZmZ d dlmZ d dlmZm	Z	m
Z
 d Zd Zd Zd Zd	 Zd
 ZdS )    N)assert_allcloseassert_)FuncData)gammagammalnloggammac                  b   t          j        g d          } |                                 }t          j        | |          \  } }| d|z  z                                   }t          j        |t          |          f          j        }d }t          ||dddd          	                                 d S )Ng     Xg      #g            ?g      #@g     X@              ?c                 D    t          j        t          |                     S )N)npexpr   zs    a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/special/tests/test_loggamma.pyfztest_identities1.<locals>.f   s    vhqkk"""    r      +=rtolatol)
r   arraycopymeshgridflattenvstackr   Tr   checkxyr   datasetr   s        r   test_identities1r%      s    
44455A	A;q!DAq	
RTAiE!HH&&(G# # # QAE666<<>>>>>r   c                     t          j        g d          } |                                 }t          j        | |          \  } }| d|z  z                                   }t          j        |t          j        |          t          |          z   f          j        }d }t          ||dddd          
                                 d S )Nr
   r   c                 &    t          | dz             S )Nr   )r   r   s    r   r   ztest_identities2.<locals>.f   s    Ar   r   r   r   r   )r   r   r   r   r   r   logr   r   r   r    r!   s        r   test_identities2r)      s    
44455A	A;q!DAq	
RTAiBF1II34557G   QAE666<<>>>>>r   c                  &   t           j        t          j        dd           t          j        dd          f         dz   } t          j        | t	          |           f          j        }d }t          ||dddd                                           d S )	N
   r   c                 V    t          j        | d          } t          |           j        S )N
complex128)dtype)r   r   r   realr   s    r   r   z)test_complex_dispatch_realpart.<locals>.f+   s%    HQl+++{{r   r   r   r   r   )r   r_logspacer   r   r   r   r    )r"   r$   r   s      r   test_complex_dispatch_realpartr3   $   s     	r{2s###R[b%9%99:S@AiGAJJ((*G      QAE666<<>>>>>r   c                     t          j        dd          dz   } t          j        | t          |           f          j        }t          t          |dddd                                           t          t          d          t           j	        k               t          t          j
        t          d                               d S )	Nr,   r+   r   r   r   r   r   )r   r2   r   r   r   r   r   r    r   infisnan)r"   r$   s     r   test_real_dispatchr8   2   s    
Cs"AiGAJJ((*GXw15u===CCEEEHQKK26!"""BHXb\\""#####r   c                      t          t          dd                    } t          t          dd                    }t          | |                                dd           d S )Ng333333               V瞯<r   r   )r   complexr   	conjugate)r   zbars     r   test_gh_6536r@   ;   sW    t$$%%AGD$''((DAt~~''e!<<<<<<r   c                      t          j        ddd           } t          j        d | D                       }t          j        d | D                       }t          ||                                dd           d S )	Ni,  id   c                 .    g | ]}t          |d           S )r:   r=   .0x0s     r   
<listcomp>z#test_branch_cut.<locals>.<listcomp>D   s"    111GB$$111r   c                 .    g | ]}t          |d           S )r;   rD   rE   s     r   rH   z#test_branch_cut.<locals>.<listcomp>E   s"    555Rwr4((555r   r<   r   r   )r   r2   asarrayr   r>   )r"   r   r?   s      r   test_branch_cutrK   A   s{    	S#s	#	##A

11q11122A:55155566DAt~~''e!<<<<<<r   )numpyr   numpy.testingr   r   scipy.special._testutilsr   scipy.specialr   r   r   r%   r)   r3   r8   r@   rK    r   r   <module>rQ      s        2 2 2 2 2 2 2 2 - - - - - - 2 2 2 2 2 2 2 2 2 2? ? ?? ? ?? ? ?$ $ $= = == = = = =r   