ó
ØÆ÷Xc           @` s¨   d  d l  m Z m Z m Z d  d l Z d  d l Z d  d l m Z m	 Z	 m
 Z
 m Z m Z m Z d  d l m Z d Z d e	 f d „  ƒ  YZ e d k r¤ e ƒ  n  d S(	   i    (   t   divisiont   absolute_importt   print_functionN(   t   assert_t   TestCaset   assert_array_equalt   assert_allcloset   run_module_suitet   suppress_warnings(   t   sixui   t   TestRegressionc           B` sq   e  Z e d  „ Z e d „ Z e d „ Z e d „ Z e 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   levelt   x(    (    s=   /tmp/pip-build-X4mzal/numpy/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   R   (    (    s=   /tmp/pip-build-X4mzal/numpy/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   R   t   at   bt   c(    (    s=   /tmp/pip-build-X4mzal/numpy/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   R   (    (    s=   /tmp/pip-build-X4mzal/numpy/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   R   (    (    s=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyt   test_masked_array_repeat)   s    c         C` s    t  t j j t d ƒ ƒ ƒ d  S(   Nt   Unicode(   t   reprR   R   R   R	   (   R   (    (    s=   /tmp/pip-build-X4mzal/numpy/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=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyt   test_atleast_2d1   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=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyt   test_set_fill_value_unicode_py38   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=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyt    test_var_sets_maskedarray_scalar>   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=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyt   test_ddof_corrcoefE   s    $(   t   __name__t
   __module__t   rlevelR   R   R   R    R#   R&   R*   R-   R5   R@   (    (    (    s=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyR
      s   				t   __main__(   t
   __future__R    R   R   t   warningst   numpyR   t   numpy.testingR   R   R   R   R   R   t   numpy.compatR	   RC   R
   RA   (    (    (    s=   /tmp/pip-build-X4mzal/numpy/numpy/ma/tests/test_regression.pyt   <module>   s   .C