o
    `^hf                     @   sd   d dl Zd dl mZmZmZ d dlmZmZ d dlm	Z	 d dl
mZ dZdd Zdd	 Zd
d ZdS )    N)pilogsqrt)assert_assert_equal)FuncDatagox?c                  C   s\   t jt ddd t dddf } t | d t| fj}ttj|dddd	d
  d S )N   i   i,  y                r      gvIh%,=T)rtolnan_ok)	npr_logspacevstackscdigammaTr   check)xdataset r   ^/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/scipy/special/tests/test_digamma.pytest_consistency   s   $r   c                  C   s  dt  fddtd t  fdt dtd  dtd d  t  fdt d dtd  t  fdt td d dtd  dtd d  t  fd	t d d
td  ttdtd  tdtd  td  t  fg} t| } ttj| dddd	  d S )Nr
   g      ?   gUUUUUU?   g      ?gUUUUUU?g      ?   r   g+=)r   )
eulerr   r   r   r   asarrayr   r   r   r   )r   r   r   r   test_special_values   s   (4D

r    c                  C   sR   ddt jg} t j t jt jg}tt| | ttt tt j dg d S )Ng        g       )r   infr   r   r   r   allisnan)ptsstdr   r   r   test_nonfinite)   s   $r'   )numpyr   r   r   r   numpy.testingr   r   scipy.special._testutilsr   scipy.specialspecialr   r   r   r    r'   r   r   r   r   <module>   s    
