B
    `                 @   s   d dl Z d dlZd dlmZmZ d dlZd dlmZ ddl	m
Z
 ddgZdZd	Zd
ddddddddddddddddddgZdgZdd Zejjejd e d!d" Zd)d#d$Zd%d& Zd'd( ZdS )*    N)assert_allclosesuppress_warnings)stats   )distconti  i  g      ?g      ?ZburrZchi2Z
gausshyperZgenexponZgengammaZkappa4ZksoneZkstwoZmielkeZncfZncx2Zpearson3ZpowerlognormZ
truncexponZtukeylambdaZvonmisesZ
wrapcauchyZlevy_stableZtrapzZerlangc              c   s(   x"t D ]\} }| tkr| |fV  qW d S )N)r   skip_fit)distnamearg r
   I/home/dcms/DCMS/lib/python3.7/site-packages/scipy/stats/tests/test_fit.pycases_test_cont_fit.   s    r   zdistname,argc             C   s  | t krTyttjd  }W n tk
r4   d}Y nX |rTd|  }|d7 }t| tt| }t	
|ddgg}t	t	|t t	|jd tgd}xtD ]}t	jd	 t	jd
d@ t .}|jtdd |j|d|i}	||	}
W d Q R X W d Q R X |
| }t	t	|	 t tg|d< t	t	|
rFtdqt	t	||krP qW dt | }|dt |
 7 }|dt | 7 }td|j! | d S )NZSCIPY_XFAILTz$Fitting %s doesn't work reliably yetzH [Set environment variable SCIPY_XFAIL=1 to run this test nevertheless.]g        g      ?   r   i  ignore)allz
.*frechet_)categorymessagesizeznan returned in fitzparameter: %s
zestimated: %s
zdiff     : %s
zfit not very good in %s
)"failing_fitsintosenviron	Exceptionpytestxfailgetattrr   npZhstackmaxZvstackthresh_percentfullZnumargs
thresh_min	fit_sizesrandomseedZerrstater   filterDeprecationWarningrvsfitabsZmeananyisnanAssertionErrorr   strname)r   r	   r   msgZdistfnZtrueargZdiffthresholdZfit_sizesupr&   ZestZdifftxtr
   r
   r   test_cont_fit7   s<    


 
r1   c             C   s4   t t| }||dd  }t|||d|  d d S )Nr   z"poor mle fit of (loc, scale) in %s)atolerr_msg)r   r   r'   r   )r-   dataZdesiredr2   dactualr
   r
   r   _check_loc_scale_mle_fitf   s    
r7   c           
   C   sB   t ddddddddg} td| ddgd td| dd	gd d S )
Ng)\(?g{Gz?gGz?g       @uniformgGz?gMbP?expongp=
ף?)r   arrayr7   )r4   r
   r
   r   "test_non_default_loc_scale_mle_fitm   s    r;   c              C   s:   ddddddddg} t jj| dd}t|ddgdd dS )zgh-6167r   r   )Zflocg      ?gMbP?)r2   N)r   r9   r'   r   )r4   Zphatr
   r
   r   test_expon_fits   s    r<   )N)r   numpyr   Znumpy.testingr   r   r   Zscipyr   Ztest_continuous_basicr   r!   r   r    r   r   r   markZslowZparametrizer1   r7   r;   r<   r
   r
   r
   r   <module>   sD   	.
