U
    Hvf                     @   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         -q=)atol333333?绽|=g      ?   g?   gUUUUUU?g333333gUUUUUU?N)r   r   nppir   )varkurtZdesired r   L/tmp/pip-unpacked-wheel-96ln3f52/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}|D ]6\}}}t|}t||f| t|}t||f|  qxt| \}}}t|}t||f| t|}t||f|  d&S )'zACompare results with some values that were computed using mpmath.r   r   )r	   Zrtolr   gg;@g2V#H@gQIg9@gk+\)@ggxuh~@g@gMbPg٤{i
@ghYn?gMbP?gӦNFb:
@g4\>?g      ?gt@g<n\?g?gU+2C@gӤB?gQI?gGI8@gx@As?r
   gڽP?g@᜿g      $@gds5O?gX\@g      4@g ?gc@N)dictr   r   r   zip)Za10Za12datalamZvar_expectedZkurt_expectedr   r   r   r   r   test_tukeylambda_stats_mpmath+   s0    r   c                  C   sT   ddg} t | }t|ttjtjg ddg} t| }t|ttjtjg dS )z;Test values of lambda outside the domains of the functions.g      g      g      пN)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
   #"