
    tKg                     `    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y)    N)assert_allcloseassert_)FuncData)gammagammalnloggammac                  H   t        j                  g d      } | j                         }t        j                  | |      \  } }| d|z  z   j	                         }t        j
                  |t        |      f      j                  }d }t        ||dddd      j                          y )Ng     Xg      #g            ?g      #@g     X@              ?c                 >    t        j                  t        |             S )N)npexpr   zs    e/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/special/tests/test_loggamma.pyfztest_identities1.<locals>.f   s    vvhqk""    r      +=rtolatol)
r   arraycopymeshgridflattenvstackr   Tr   checkxyr   datasetr   s        r   test_identities1r%      s    
45A	A;;q!DAq	
RTAiiE!H&((G# QAE6<<>r   c                  t   t        j                  g d      } | j                         }t        j                  | |      \  } }| d|z  z   j	                         }t        j
                  |t        j                  |      t        |      z   f      j                  }d }t        ||dddd      j                          y )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5A	A;;q!DAq	
RTAiiBFF1I34577G QAE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      j                          y )	N
   r   c                 Z    t        j                  | d      } t        |       j                  S )N
complex128)dtype)r   r   r   realr   s    r   r   z)test_complex_dispatch_realpart.<locals>.f+   s"    HHQl+{r   r   r   r   r   )r   r_logspacer   r   r   r   r    )r"   r$   r   s      r   test_complex_dispatch_realpartr3   $   ss     	r{{2s##R[[b%99:S@AiiGAJ(**G  QAE6<<>r   c                  h   t        j                  dd      dz   } t        j                  | t        |       f      j                  }t        t        |dddd      j                          t        t        d      t         j                  k(         t        t        j                  t        d                   y )	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   sx    
Cs"AiiGAJ(**GXw15u=CCEHQK266!"BHHXb\"#r   c                      t        t        dd            } t        t        dd            }t        | |j                         dd       y )Ng333333               V瞯<r   r   )r   complexr   	conjugate)r   zbars     r   test_gh_6536r@   ;   s9    t$%AGD$'(DAt~~'e!<r   c            	      6   t        j                  ddd       } t        j                  | D cg c]  }t        |d       c}      }t        j                  | D cg c]  }t        |d       c}      }t	        ||j                         dd       y c c}w c c}w )	Ni,  id   r:   r;   r<   r   r   )r   r2   asarrayr=   r   r>   )r"   x0r   r?   s       r   test_branch_cutrE   A   s|    	S#s	##A


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