U
    Kvft                     @   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   O/tmp/pip-unpacked-wheel-2v6byqio/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rt 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st|tddddgttjfstt|tddddgtst|tddddgttjfstd S )Nr   r   r   r   )	getattrr   r   listr   r   tupler   AssertionError)r(   funcr   r   r   test_moment_conversion_typesl   s    
   r.   )"__doc__Z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   (         &&

    