3
Od                 @   sD   d dl Zd dlmZmZ d dlmZmZ dd Zdd Z	dd	 Z
dS )
    N)assert_allcloseassert_equal)tukeylambda_variancetukeylambda_kurtosisc              C   s   t d} t| tjd d dd td}t|ddd t d} t| d	tj dd td}dtjd  tjd	 d d  d }t||dd t d} t| ddd td}t|ddd t d} t| ddd td}t|ddd dS )z/Compare results with some known exact formulas.r         g-q=)atolg333333?g|=g      ?   g      @   g      ?   Ng?gUUUUUU?g333333gUUUUUU?g333333)r   r   nppir   )varkurtZdesired r   \/var/www/html/virt/lib64/python3.6/site-packages/scipy/stats/tests/test_tukeylambda_stats.py"test_tukeylambda_stats_known_exact   s"    
$r   c              C   s   t ddd} t ddd}d$ddgd%d	d
gd&ddgd'ddgdddgdddgdddgdddgddd(gdddgd d!d"gg}x>|D ]6\}}}t|}t||f| t|}t||f|  qzW t| \}}}t|}t||f| t|}t||f|  d#S ))zACompare results with some values that were computed using mpmath.g|=r   )r   Zrtolg-q=g?g;@g2V#H@gQI?g9@gk+\)@g?gxuh~@g@gMbP?g٤{i
@ghYn?gӦNFb:
@g4\>?g      ?gt@g<n\?gU+2C@gӤB?gGI8@gx@As?g333333?gڽP?g@᜿?g      $@gds5O?gX\@g      4@g ?gc@NggQIggMbPg@᜿)dictr   r   r   zip)Za10Za12datalamZvar_expectedZkurt_expectedr   r   r   r   r   test_tukeylambda_stats_mpmath+   s.    r   c              C   sT   ddg} t | }t|tjtjtjg ddg} t| }t|tjtjtjg dS )	z;Test values of lambda outside the domains of the functions.g      ?g      ?g      ?Ng      g      g      g      п)r   r   r   arraynaninfr   )r   r   r   r   r   r   test_tukeylambda_stats_invalidM   s    r   )Znumpyr   Znumpy.testingr   r   Zscipy.stats._tukeylambda_statsr   r   r   r   r   r   r   r   r   <module>   s
   #"