ó
áp7]c           @   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 „  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 j d e ƒ d „  ƒ Z e j j d ƒ Z e j j d	 d d ƒ j d d
 ƒ Z e j d d d g d d d g d d d g d  d! d" g g ƒ e g Z e j j d# e ƒ d$ „  ƒ Z  e j j d% d& d& d' d( d) d* d+ d+ d, d- g
 ƒ d. „  ƒ Z! d S(/   s=   
Created on Sun Oct 16 17:33:56 2011

Author: Josef Perktold
iÿÿÿÿN(   t   assert_almost_equalt   assert_t   assert_equal(   t   moment_helpers(   t   cov2corrt   mvsk2mct   mc2mvskt   mnc2mct   mc2mnct   cum2mct   mc2cumt   mnc2cumc    	      C   sœ  t  j d ƒ t  j t  j d d ƒ d d ƒ }  t  j d d d d d g d d d d d d g d d d d d d g g ƒ } t |  ƒ } t | | d d ƒ|  } t | ƒ } t t | t  j	 ƒ ƒ t
 | | ƒ t  j j |  ƒ } t | ƒ } t
 | | ƒ t t | t  j j j ƒ ƒ t  j j |  d	 t t t g t t t g t t t g g ƒ} t | ƒ } t t  j j | | d
 d ƒƒ t
 | j | j ƒ d  S(   Ni   i   i   i   g       @g      @t   decimali   t   maskt   atolgVçž¯Ò<(   i   i   (   t   npt   onest   diagt   aranget   arrayR   R    R   t
   isinstancet   ndarrayR   t   mat   coret   MaskedArrayt   Falset   Truet   allcloseR   (	   t   cov_at   corr_at   corrt   cov_matt   corr_matt   cov_mat   corr_mat   cov_ma2t   corr_ma2(    (    sJ   lib/python2.7/site-packages/statsmodels/stats/tests/test_moment_helpers.pyt   test_cov2corr   s(    0!g        i   i    i   g      ð?g      ð¿g      @g       @i   g      @i   g      "Àg      Àg      "@g      à?g      @g      à¿g      4Àt   momc         C   s^  t  t t |  d ƒ ƒ |  d ƒ t  t |  d ƒ |  d ƒ t |  ƒ d k rl t  t |  d ƒ |  d ƒ n  t  t |  d ƒ |  d ƒ t  t t |  d ƒ ƒ |  d ƒ t |  ƒ d k rØ t  t |  d ƒ |  d ƒ n  t  t t |  d ƒ ƒ t |  d ƒ ƒ t  t t |  d ƒ ƒ |  d ƒ t |  d ƒ d k rZt  t t |  d ƒ ƒ |  d ƒ n  d  S(   Ni    i   i   i   (	   R   R   R   t   lenR
   R	   R   R   R   (   R&   (    (    sJ   lib/python2.7/site-packages/statsmodels/stats/tests/test_moment_helpers.pyt   test_moment_conversion<   s    !!'!i90  id   i   g      @g      $@g      T@g     `s@g      @t	   test_valsc         C   sj   t  t t t |  ƒ j ƒ j ƒ j |  ƒ t  t t |  ƒ j ƒ j |  ƒ t  t t |  ƒ j ƒ j |  ƒ d  S(   N(   R    R	   R   R   t   TR
   R   R   (   R)   (    (    sJ   lib/python2.7/site-packages/statsmodels/stats/tests/test_moment_helpers.pyt   test_multidimensionale   s    (t	   func_nameR	   R
   R   R   R   R   R   t   mvsk2mncc         C   sœ  t  t |  ƒ } t | d d d d g ƒ t ƒ s t | t j d d d d g ƒ ƒ t ƒ s t | t j d d d d g ƒ ƒ t j ƒ s t ‚ t | t j d d d d g ƒ ƒ t ƒ st | t j d d d d g ƒ ƒ t ƒ st | t j d d d d g ƒ ƒ t j ƒ st ‚ t | t d d d d g ƒ ƒ t ƒ s˜t | t j d d d d g ƒ ƒ t ƒ s˜t | t j d d d d g ƒ ƒ t j ƒ s˜t ‚ d  S(   Ng      ð?i   i    i   (	   t   getattrR   R   t   listR   R   t   tupleR   t   AssertionError(   R,   t   func(    (    sJ   lib/python2.7/site-packages/statsmodels/stats/tests/test_moment_helpers.pyt   test_moment_conversion_typesl   s    !*3**3'*("   t   __doc__t   numpyR   t   pytestt   numpy.testingR    R   R   t   statsmodels.statsR   t    statsmodels.stats.moment_helpersR   R   R   R   R   R	   R
   R   R%   t   mst   markt   parametrizeR(   t   randomt   RandomStatet   rst   randintt   reshapet   random_valsR   t   multidimension_test_valsR+   R3   (    (    (    sJ   lib/python2.7/site-packages/statsmodels/stats/tests/test_moment_helpers.pyt   <module>   s:   :	0000000009? $		