o
    Eb                     @   sB  d dl Zd dlZd dlmZmZ d dlmZmZm	Z	m
Z
 d dlmZmZmZ d dlmZmZ zd dlZW n eyA   edZY nw zd dlZW n eyU   edZY nw eedd	d
 Zejjeedeedejddd Zejjeedeeddd Zeeddd Zejjeeddd ZdS )    N)MissingModulecheck_version)ArgIntArgmp_assert_allcloseassert_mpmath_equal)	compute_gcompute_alpha	compute_d)gammainc	gammainccsympympmathz0.19c               
   C   s   t d< t dt dd t dd t d d t d d t d	d
 t dd g} ttd|  W d    d S 1 sDw   Y  d S )N         i      i  i;  i % i' i `uicP l     
F    )mpworkdpsmpfr   r   )g r   N/usr/lib/python3/dist-packages/scipy/special/tests/test_precompute_gammainc.pytest_g   s    "r   z0.7zrtol only 2e-11, see gh-6938c                  C   s   t dF t dt dt dd t dd t d d t dd t dd t d	 d
 t dd g	} ttd|  W d    d S 1 sNw   Y  d S )Nr   r   r      $   i  i  irB  r   iS iX 	   )r   r   r   r   r	   )Zalphar   r   r   
test_alpha    s   (&"r   c                     s  t d ddt d fddt dfddt d fddt dfd	dt d
fd	dt d fddt dfddt d fddt d fddt dfddt d fddt dfddt dfddt d fddt dfddt d fddt d fddt dfddt d fddt d fg} td!d"  fd#d$| D }td%d& | }t|| W d    d S 1 sw   Y  d S )'N2   r   z 0.333333333333333333333333333333r   z#0.102618097842403080425739573227e-7r   z#0.185185185185185185185185185185e-2z#0.119516285997781473243076536700e-7   z#0.413359788359788359788359788360e-2z#0.140925299108675210532930244154e-7r   z#0.649434156378600823045267489712e-3z#0.191111684859736540606728140873e-7   z#0.861888290916711698604702719929e-3z#0.288658297427087836297341274604e-7   z#0.336798553366358150308767592718e-3z#0.482409670378941807563762631739e-7   z#0.531307936463992223165748542978e-3z#0.882860074633048352505085243179e-7r   z#0.344367606892377671254279625109e-3z#0.175629733590604619378669693914e-6   z#0.652623918595309418922034919727e-3z#0.377358774161109793380344937299e-6r   z#0.596761290192746250124390067179e-3z#0.870823417786464116761231237189e-6
      c                    s   g | ]\}}} | | qS r   r   ).0knstddr   r   
<listcomp>O   s    ztest_d.<locals>.<listcomp>c                 S   s   | d S )Nr    r   )xr   r   r   <lambda>P   s    ztest_d.<locals>.<lambda>)r   r   r   r
   mapr   )Zdatasetresr*   r   r+   r   test_d-   s6   
"r2   c                	   C   s2   t tdd tddddtddgddddd	 d S )
Nc                 S      t j| |ddS )NT)bregularizedr   r   ar.   r   r   r   r/   Z       ztest_gammainc.<locals>.<lambda>r   d   F)Zinclusive_aFFg<r   Znan_okZrtolr)   dps)r   r   r   r   r   r   r   test_gammaincT   s
   
r>   c                	   C   s\   t dd dd tddtddgdddd	d
 t tdd tddtddgddddd
 d S )Nc                 S   s   t | |ddS )N  )r=   )r   r7   r   r   r   r/   d   s    z test_gammaincc.<locals>.<lambda>c                 S   r3   NT)r8   r5   r6   r7   r   r   r   r/   e   r9      r:   Fr;   r   r?   r<   c                 S   r3   r@   r6   r7   r   r   r   r/   k   r9   r   r   )r   r   r   r   r   r   r   r   test_gammaincc_   s   
rB   )ZnumpyZnpZpytestZscipy.special._testutilsr   r   Zscipy.special._mptestutilsr   r   r   r   Z&scipy.special._precompute.gammainc_asyr   r	   r
   Z'scipy.special._precompute.gammainc_datar   r   r   ImportErrorr   r   r   ZmarkZslowZxfail_on_32bitr   Zxslowr2   r>   rB   r   r   r   r   <module>   s@    
	
	$

