a
    Qe                     @   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   \/var/www/sunrise/env/lib/python3.9/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}g dg dg dg dg d	g d
g dg dg dg dg dg}|D ]>\}}}t|}t||fi | t|}t||fi |  qbt| \}}}t|}t||fi | t|}t||fi |  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   )numpyr   Znumpy.testingr   r   Zscipy.stats._tukeylambda_statsr   r   r   r   r   r   r   r   r   <module>   s
   #"