U
    »mœ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   é   é   çê-�™—q=)Úatolç333333ó?ç»½×Ùß|Û=g      à?é   g«ªªªªªú?é   gUUUUUUÕ?g333333ó¿gUUUUUUµ?N)r   r   ÚnpÚpir   )ÚvarÚkurtZdesired© r   úa/home/sam/Atlas/atlas_env/lib/python3.8/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}|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   gš™™™™™¹¿g®Ÿ�ö;@g�2V#æH@g“©‚QI�°¿g�ú9¨@g›k+ë\)@gš™™™™™©¿gx¤ºuh~@géîíå@gü©ñÒMbP¿gòÙ¤{i
@göhY±nó?gü©ñÒMbP?gÓ¦NFb:
@gÚ4›\>øò?g       ?g®tö™¬@g<°n¯\¾é?gš™™™™™©?gûÔU+2C@gÓ¤B’ã?g“©‚QI�°?gGI¸û8@gx§@As„Þ?r
   gÚ½PæþÎ?gÃ@áœ¿ó¿g      $@gòd¹s5O?gXîò\@g      4@gù ÿø?g‘c‹”�@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   )Únumpyr   Znumpy.testingr   r   Zscipy.stats._tukeylambda_statsr   r   r   r   r   r   r   r   r   Ú<module>   s
   #"