ó
ÐEe]c           @` sd   d  d l  m Z m Z m Z d  d l Z d  d l m Z m Z m	 Z	 m
 Z
 d e f d „  ƒ  YZ d S(   i    (   t   divisiont   absolute_importt   print_functionN(   t   assert_t   assert_array_equalt   assert_allcloset   suppress_warningst   TestRegressionc           B` st   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z	 d „  Z
 d	 „  Z d
 „  Z d „  Z RS(   c         C` st   t  j j d d d d d d d d g d d d d d d d d d g ƒ} t t  j j | ƒ d d d d	 g g ƒ d  S(
   Ni    i   i   i   i   i   i   t   maski   (   t   npt   mat   masked_arrayR   t   nonzero(   t   selft   x(    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_masked_array_create
   s    '!c         C` s   t  j j d d d g ƒd  S(   Ni   R   (   R	   R
   t   array(   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_masked_array   s    c         C` sU   d d l  m } m } t j d ƒ } t j | j | ƒ } | | | ƒ } | | d  S(   Ni    (   t   masked_wheret   MaskTypei   (   i   i   (   t   numpy.maR   R   R	   t   zerost   shape(   R   R   R   t   at   bt   c(    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_mem_masked_where   s
    c         C` sE   t  j j d ƒ } t  j j | d <t  j d ƒ } | | | | d  S(   Ni   i   i   i    (   i   i   (   i   i    (   i   i   (   R	   R
   R   t   masked(   R   R   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_masked_array_multiply   s
    c         C` s&   t  j j d g d t ƒj d ƒ d  S(   Ni   R   i
   (   R	   R
   R   t   Falset   repeat(   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_masked_array_repeat$   s    c         C` s   t  t j j d ƒ ƒ d  S(   Nu   Unicode(   t   reprR	   R
   R   (   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_masked_array_repr_unicode(   s    c         C` si   t  j j d d d g d t t t g ƒ} t  j | ƒ } t | j j d k ƒ t | j j d k ƒ d  S(   Ng        g333333ó?g      @R   i   i   (	   R	   R
   R   R   t   Truet
   atleast_2dR   R   t   ndim(   R   R   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_atleast_2d,   s    *c         C` sJ   t  j j d d d g d d d d g ƒ} d | _ t | j d k ƒ d  S(   NR   R   R   R   i   i    t   X(   R	   R
   R   t
   fill_valueR   (   R   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_set_fill_value_unicode_py33   s    *	c         C` s`   t  j j t  j d ƒ d t ƒ} t  j j d d t ƒ} | j d | ƒ t | j d k ƒ d  S(   Ni   R   iÿÿÿÿt   dtypet   outi    (	   R	   R
   R   t   arangeR"   t   floatt   varR   t   _data(   R   R   t   mout(    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt    test_var_sets_maskedarray_scalar9   s    !c      	   C` s±   t  j j d d d d d g d ƒ } t  j d d d d d g ƒ } t ƒ  ` } | j t d ƒ t  j j | | d	 d
 ƒ} t  j j | | d	 d ƒ} t | j	 | j	 ƒ Wd  QXd  S(   Ni   i   i   i   i   g      @gÍÌÌÌÌÌ@s   bias and ddof have no effectt   ddofi    (
   R	   R
   t   masked_equalR   R   t   filtert   DeprecationWarningt   corrcoefR   t   data(   R   R   t   yt   supt   r0t   r1(    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_ddof_corrcoef@   s    $c         C` s   t  j j d d g d t t g ƒ} t | j j d k ƒ t  j | d ƒ } t | j j d k ƒ t | j d	 k ƒ t | j j d
 k ƒ d  S(   Ng      ð?g       @R   i   i   (   i   (   i   i   (   i   (   i   i   (   i   i   (   R	   R
   t   MaskedArrayR   R   R   R   t   tile(   R   R   R   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_mask_not_backmangledL   s    $c         C` s:   t  j j d
 d d g d d ƒ} t | g  | d	  ƒ d  S(   Ni   g      ð?i   g       @i   g      @R)   s   i4,f4i    (   i   g      ð?(   i   g       @(   i   g      @(   R	   R
   R<   R   (   R   R
   (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   test_empty_list_on_structuredV   s    !(   t   __name__t
   __module__R   R   R   R   R   R!   R%   R(   R0   R;   R>   R?   (    (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyR   	   s   											
(   t
   __future__R    R   R   t   numpyR	   t   numpy.testingR   R   R   R   t   objectR   (    (    (    s=   lib/python2.7/site-packages/numpy/ma/tests/test_regression.pyt   <module>   s   "