U
    Ãmœdt  ã                   @   s~  d Z ddlZddlZddlmZmZmZ ddlm	Z	 ddl
mZmZmZmZmZmZmZmZ dd„ Zdd	dd
gddddgddddgfdd	dd
gddddgddddgfdd	d	d
gddddgddddgfdd	d	d
gddddgddddgfdd	d	dgddddgddddgfdddd
gddddgddddgfddd	d
gddddgddddgfdddd
gddddgddddgfddd	d
gddddgddddgfdd	dd
dgdddddgdddddgfdd	dd
d	gdddddgdddddgfgZej de¡dd„ ƒZej d¡Ze ddd¡ dd
¡Ze dd dgdd dgdd dgd!d"d#gg¡egZej d$e¡d%d&„ ƒZ ej d'd(d(d)d*d+d,d-d-d.d/g
¡d0d1„ ƒZ!dS )2z=
Created on Sun Oct 16 17:33:56 2011

Author: Josef Perktold
é    N)Úassert_almost_equalÚassert_Úassert_equal)Úmoment_helpers)Úcov2corrÚmvsk2mcÚmc2mvskÚmnc2mcÚmc2mncÚcum2mcÚmc2cumÚmnc2cumc            	      C   s  t  d¡t  t  dd¡d d ¡ } t  dddgdddgdddgg¡}t| ƒ}t||dd	� | }t|ƒ}tt|t j	ƒƒ t
||ƒ t j | ¡}t|ƒ}t
||ƒ tt|t jjjƒƒ t jj| d
dd
gdd
d
gd
d
d
ggd�}t|ƒ}tt jj||dd�ƒ t
|j|jƒ d S )N)é   r   é   é   é   ç      à?gUUUUUUÕ?gUUUUUUÅ?é   )ÚdecimalFT)ÚmaskgVçž¯Ò<)Zatol)ÚnpZonesZdiagZarangeÚarrayr   r   r   Ú
isinstanceÚndarrayr   ÚmaÚcoreZMaskedArrayZallcloser   )	Zcov_aZcorr_aZcorrZcov_matZcorr_matZcov_maZcorr_maZcov_ma2Zcorr_ma2© r   úd/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/stats/tests/test_moment_helpers.pyÚtest_cov2corr   s,    $þ

þr   g        r   r   ç      ð?g      ð¿g      @g       @r   g      @r   g      "Àg      Àg      "@r   g      @g      à¿g      4ÀÚmomc                 C   s  t tt| d ƒƒ| d ƒ t t| d ƒ| d ƒ t| ƒdkrRt t| d ƒ| d ƒ t t| d ƒ| d ƒ t tt| d ƒƒ| d ƒ t| ƒdkr¤t t| d ƒ| d ƒ t tt| d ƒƒt| d ƒƒ t tt| d ƒƒ| d ƒ t| d ƒdk�rt tt| d ƒƒ| d ƒ d S )Nr   r   r   r   )	r   r   r
   Úlenr   r   r	   r   r   )r    r   r   r   Útest_moment_conversion<   s    r"   i90  éd   é   g      @g      $@g      T@g     `s@g      @Ú	test_valsc                 C   sL   t ttt| ƒjƒjƒj| ƒ t tt| ƒjƒj| ƒ t tt| ƒjƒj| ƒ d S )N)r   r   r   r
   ÚTr   r   r   )r%   r   r   r   Útest_multidimensionale   s    r'   Ú	func_namer   r   r
   r   r   r	   r   Zmvsk2mncc                 C   sÌ   t t| ƒ}t|ddddgƒtƒsFt|t ddddg¡ƒttjfƒsFt‚t|t ddddg¡ƒtƒsˆt|t ddddg¡ƒttjfƒsˆt‚t|tddddgƒƒtƒsÈt|t ddddg¡ƒttjfƒsÈt‚d S )Nr   r   r   r   )	Úgetattrr   r   Úlistr   r   Útupler   ÚAssertionError)r(   Úfuncr   r   r   Útest_moment_conversion_typesl   s    
 ÿ ÿ ÿr.   )"Ú__doc__Únumpyr   ZpytestZnumpy.testingr   r   r   Zstatsmodels.statsr   Z statsmodels.stats.moment_helpersr   r   r   r	   r
   r   r   r   r   ÚmsÚmarkZparametrizer"   ÚrandomZRandomStateÚrsÚrandintZreshapeZrandom_valsr   Zmultidimension_test_valsr'   r.   r   r   r   r   Ú<module>   sP   (         &&õ
ýü
    þ