ó
ÐEe]c           @` sÒ  d  Z  d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l	 m
 Z
 m Z d d l m Z m Z m Z m Z d d l m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z d d l m Z m Z m  Z  m! Z! m" Z" m# Z# m$ Z$ m% Z% m& Z& m' Z' m( Z( m) Z) m* Z* m+ Z+ m, Z, m- Z- m. Z. m/ Z/ m0 Z0 m1 Z1 m2 Z2 m3 Z3 m4 Z4 m5 Z5 m6 Z6 m7 Z7 m8 Z8 m9 Z9 m: Z: m; Z; m< Z< m= Z= m> Z> d e? f d	 „  ƒ  YZ@ d
 e? f d „  ƒ  YZA d e? f d „  ƒ  YZB d e? f d „  ƒ  YZC d e? f d „  ƒ  YZD d e? f d „  ƒ  YZE d e? f d „  ƒ  YZF d e? f d „  ƒ  YZG d e? f d „  ƒ  YZH d e? f d „  ƒ  YZI d e? f d „  ƒ  YZJ d e? f d „  ƒ  YZK d  e? f d! „  ƒ  YZL d" e? f d# „  ƒ  YZM d S($   sæ   Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant

:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $

i    (   t   divisiont   absolute_importt   print_functionN(   t   assert_warnst   suppress_warnings(   t   assert_t   assert_array_equalt   assert_equalt   assert_almost_equal(   t   arrayt   aranget   maskedt   MaskedArrayt   masked_arrayt   getmaskarrayt   shapet   nomaskt   onest   zerost   count(!   t
   atleast_1dt
   atleast_2dt
   atleast_3dt   mr_t   dott   polyfitt   covt   corrcoeft   mediant   averaget   uniquet   setxor1dt	   setdiff1dt   union1dt   intersect1dt   in1dt   ediff1dt   apply_over_axest   apply_along_axist   compress_ndt   compress_rowcolst   mask_rowcolst   clump_maskedt   clump_unmaskedt   flatnotmasked_contiguoust   notmasked_contiguoust   notmasked_edgest
   masked_allt   masked_all_liket   isint   diagflatt   stackt   vstackt   TestGenericc           B` s>   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` s4  t  d d t ƒ} t d d g d d d g d t ƒ} t | | ƒ t j i d d g d 6d d g d	 6ƒ } t  d d | ƒ} t d d g d d d g d | ƒ} t | | ƒ t  d d | ƒ} t d d g d d g g d d d g d d g g d | ƒ} t | | ƒ t j d d d d g f g ƒ } t  d d | ƒ} t d d  f d d! f g d d d" f d d# f g d | ƒ} t | | ƒ t  d$ d | ƒ} t d d% f d d& f g d d d' f d d( f g d | ƒ} t | | ƒ t  d) d | ƒ} t d d* f g g d d d+ f g g d | ƒ} t | | ƒ d  S(,   Ni   t   dtypei   t   maskt   at   bt   namest   ft   formatsi    t   bat   bb(   i   (   i   (   i    i    (   i    i    (   i   i   (   i   i   (   i   i   (   i    i    (   i    i    (   i    i    (   i    i    (   i   i   (   i   i   (   i   i   (   i   i   (   R8   R;   (   R=   R;   (   R>   R;   (   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   R/   t   floatR	   R   t   npR6   (   t   selft   testt   controlt   dt(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_masked_all&   s0    $)$	!!!0c         C` sY  t  d d g d t ƒ} t | ƒ } t  d d g d d d g d t ƒ} t | | ƒ t j i d d g d 6d d g d	 6ƒ } t  d d g d d d g d | ƒ} t | ƒ } t  d d g d d d g d | ƒ} t | | ƒ t j d d d d g f g ƒ } t  d d f d d f g d d d f d d f g d | ƒ} t | ƒ } t | | ƒ d  S(   Ni   i   R6   R7   R8   R9   R:   R;   R<   i    i
   R=   R>   (   i    i    (   i    i    (   i   i   (   i   i   (   i
   i
   (   i
   i
   (   i   i   (   i   i   (   R8   R;   (   R=   R;   (   R>   R;   (   i   i   (   i   i   (   i   i   (   i   i   (   R	   R?   R0   R   R@   R6   (   RA   t   baseRB   RC   RD   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_masked_all_likeD   s    $)$$!!c   	      C` s  xt  d d ƒ D]÷ } xî t  d | ƒ D]Ü } t j | d t ƒ} t j | | d t ƒ} t d | ƒ } | d | @d k | _ d } x+ | | ƒ D] } | | j | j ƒ  7} q” W| t	 k rÝ t
 | j ƒ  j ƒ  | ƒ q' | j | _ t
 | j ƒ  j ƒ  | ƒ q' Wq Wd  S(   Ni   i   i   R6   i    (   t   rangeR@   R
   t   intt   fullR   R7   t   datat   sumR+   R   t
   compressed(	   RA   R;   t   it   jt   kt   jaR8   t   st   sl(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   check_clumpX   s    c         C` s…   t  t j d ƒ ƒ } t | d d d d d d g <t | ƒ } t d d ƒ t d d	 ƒ t d d ƒ g } t | | ƒ |  j t ƒ d  S(
   Ni
   i    i   i   i   i   i	   i   i   (   R   R@   R
   R   R*   t   sliceR   RT   (   RA   R8   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_clump_maskedh   s    *c         C` sy   t  t j d ƒ ƒ } t | d d d d d d g <t | ƒ } t d d ƒ t d	 d ƒ g } t | | ƒ |  j t ƒ d  S(
   Ni
   i    i   i   i   i   i	   i   i   (   R   R@   R
   R   R+   RU   R   RT   (   RA   R8   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_clump_unmaskeds   s    c         C` sá   t  d ƒ } t | ƒ } t | t d | j ƒ g ƒ t j d d t ƒ| _ t | t d | j ƒ g ƒ t	 | | d k  | d k B| d k B<t | ƒ } t | t d d ƒ t d d ƒ g ƒ t	 | (t | ƒ } t | g  ƒ d  S(	   Ni
   i    R6   i   i   i   i   i	   (
   R
   R,   R   RU   t   sizeR@   R   t   boolR7   R   (   RA   R8   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_flatnotmasked_contiguous}   s    $%(   t   __name__t
   __module__RE   RG   RT   RV   RW   RZ   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR5   $   s   					
t   TestAveragec           B` s5   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z RS(   c      	   C` s½  t  d d d d g d t t t t g ƒ} t d t | d d ƒƒ t d t | d d d d d g ƒƒ t | d d d d d g d	 d
 ƒ\ } } t d | ƒ t | d k ƒ t | (t t | d d ƒj t g ƒ t  d d d d g d t t t t g ƒ} | j d d ƒ } t | d  d  … d
 f <t t | d d ƒd d g ƒ t t | d d
 ƒj d t g ƒ t d d g t | d d ƒƒ t | d d d	 d
 ƒ\ } } t | d d g ƒ d  S(   Ng        g      ð?g       @g      @R7   t   axisi    t   weightst   returnedi   g      @i   (	   R	   t   Truet   FalseR   R   R   R   R7   t   reshape(   RA   t   ottt   resultt   wts(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_testAverage1’   s     *%**#c         C` s²  d d d d d d g } d d d d d d g d d d d d d g g } t  d d t j ƒ} t t | d d ƒd ƒ t t | d d d | ƒd ƒ t t  d d t j ƒd t  d ƒ g ƒ } t t | d  ƒ t j j t j  d ƒ ƒ d	 d
 ƒ t t | d d ƒt j  d ƒ d	 d ƒ t t | d d ƒt | d d ƒt | d d ƒd g ƒ t t | d  d | ƒd ƒ t t | d d d | ƒd d d d	 d d g ƒ t t | d d ƒt | d d ƒt | d d ƒd g ƒ t	 d ƒ } d d d d d d g } d d d d d d g d d d d d d g g } t
 d ƒ } d d d d d d g }	 t t t | | ƒ d d ƒd ƒ t t t | | ƒ d d ƒd ƒ t t t | | ƒ d d ƒj t g ƒ t t t | |	 ƒ d d ƒd ƒ t t t t | | ƒ d d ƒƒ d ƒ t | | ƒ }
 t t |
 d  ƒ d ƒ t t |
 d d ƒd d d d d d g ƒ t t |
 d d ƒd d g ƒ t t |
 d d d | ƒd d d d d d g ƒ d  S(   Ni    i   i   R6   R^   g      @R_   g       @g      @g      (@g      4@g      @g        g      ð?g      @g      $@g     ÀX@g      @g      @g«ªªªªª
@g«ªªªªª
@(   R
   R@   t   float_R   R   R	   t   Nonet   addt   reduceR   R   R   R7   Ra   R   (   RA   t   w1t   w2t   xt   yt   m1t   m2t   m3t   m4t   m5t   z(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_testAverage2¥   s>    0+3*))0""("(+c         C` s  t  d ƒ } t  d ƒ d } t | | g | | g g d d d d ƒ\ } } t t | ƒ t | ƒ ƒ t | j | j ƒ t t d ƒ d d d d d g d d ƒ\ } } t t | ƒ t | ƒ ƒ t t d ƒ d d ƒ\ } } t t | ƒ t | ƒ ƒ t t d ƒ d t d ƒ d d ƒ\ } } t t | ƒ t | ƒ ƒ t d d g d d	 g g t ƒ } t | t t g t	 t g g ƒ } t | d d ƒ}	 t |	 d
 d g ƒ t | d d ƒ}
 t |
 d d g ƒ t | d d  ƒ}
 t |
 d ƒ t | d d ƒ}
 t |
 d d g ƒ d  S(   Ni   i   R^   i   R`   i   i    R_   i   g      à?g      @g      ð?g      @g      ø?g      @(   i   i   i   (   i   i   i   (   i   i   i   (   i   i   i   g«ªªªªª@(   R
   R   R   R   R   R	   R?   R   Rb   Ra   Ri   (   RA   R8   R9   t   r1Rl   t   r2Rm   t   a2dt   a2dmt   a2dat   a2dma(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_testAverage3Ç   s*    00*!!c         C` sf   t  t d d g ƒ ƒ } t | d ƒ t  t d d d d g d t t t t g ƒƒ } t | d ƒ d  S(   Ni   i   g      ø?i   i   R7   (   R   R	   R   Rb   Ra   (   RA   R8   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_onintegers_with_maskß   s    0c         C` sù  t  j d d d d d g d d d d d g g d t ƒ} t d d d d d g d d d d d g g d | ƒ} t | ƒ } t  j | j ƒ  ƒ } t | j | j ƒ t | j | j ƒ t | d d ƒ} t | j d d ƒt | j d d ƒd } t | j | j ƒ t | j | j ƒ t | d d ƒ} t | j d d ƒt | j d d ƒd } t | j | j ƒ t | j | j ƒ t  j d d d d d g d d d d d g g ƒ }	 t | d |	 ƒ}
 t  j | j ƒ  d |	 | ƒ} t |
 j | j ƒ t |
 j | j ƒ t | d |	 d d ƒ} t | j d |	 d d ƒt | j d |	 d d ƒd } t | j | j ƒ t | j | j ƒ t | d |	 d d ƒ} t | j d |	 d d ƒt | j d |	 d d ƒd } t | j | j ƒ t | j | j ƒ d  S(    Ni    i   R6   y               @i   y              @i   y              @i   y               @y              "@y              ð?i   y              @i   y              @y              @R7   R^   g      à?g      ð?g       @R_   y      ð?       @y      @      @y      @      @y      @       @y              ð?y       @      @y      @      @y      @      @(	   R@   R	   RY   R   R   RM   R   t   realt   imag(   RA   R7   R8   t   avt   expectedt   av0t	   expected0t   av1t	   expected1Rf   t   wavt   wav0t   wav1(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_complexæ   sB    !	,,   (   R[   R\   Rg   Rv   R}   R~   RŠ   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR]      s
   		"		t   TestConcatenatorc           B` s#   e  Z d  „  Z d „  Z d „  Z RS(   c         C` sÚ   t  t d	 t d d d d d d g ƒ ƒ t d ƒ } d d d d d g } t | d | ƒ} t | d d | f } t t | t ƒ ƒ t  | d d d d d d d d d d d d g ƒ t  | j t | d d | f ƒ d  S(
   Ni   i   i   i   i   i   i    R7   (   i   i   i   i   i   i   (	   R   R   R	   R   R   R   t
   isinstanceR   R7   (   RA   R9   t   mt   dt   c(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_1d  s    )1c         C` s¦  t  j j d d ƒ } t  j j d d ƒ } t  j t  j j d d ƒ d ƒ } t  j t  j j d d ƒ d ƒ } t | d | ƒ} t | d | ƒ} t d | | f } t | j d k ƒ t | d  d  … d  d … f | ƒ t | d  d  … d d  … f | ƒ t | j	 t  j
 d | | f ƒ t | | f } t | j d k ƒ t | d  d … d  d  … f | ƒ t | d d  … d  d  … f | ƒ t | j	 t  j
 | | f ƒ d  S(   Ni   i    R7   t   1i
   (   i   i
   (   i
   i   (   R@   t   randomt   randt   round_R   R   R   R   R   R7   t   r_(   RA   t   a_1t   a_2t   m_1t   m_2t   b_1t   b_2RŽ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_2d!  s     !!## ##c         C` s“   t  t j j d f } t | j t t g ƒ t | j d d ƒ t  d d g t j j f } t | j t t t g ƒ t | j d  d d g ƒ d  S(   Ni   i   (	   R   R@   t   maR   R   R7   Ra   Rb   RK   (   RA   t   actual(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_masked_constant5  s    (   R[   R\   R   Rœ   RŸ   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR‹     s   		t   TestNotMaskedc           B` s   e  Z d  „  Z d „  Z RS(   c         C` sJ  t  t j d ƒ j d d ƒ d d d d d d g d d d d d g d d d d d g d d d d d g d d d d d g g ƒ} t | d  ƒ } t | d d g ƒ t | d ƒ } t | d d d g ƒ t | d d d g ƒ t | d ƒ } t | d d d g ƒ t | d d d g ƒ t | j d  ƒ } t | d d g ƒ t | j d ƒ } t | d d d g ƒ t | d d d g ƒ t | j d
 ƒ } t | d d d g ƒ t | d d d g ƒ t | d <t | d ƒ } t | d d d g ƒ t | d d d g ƒ t | d
 ƒ } t | d d  d! g ƒ t | d d" d# g ƒ d  S($   Ni   i   R7   i    i   i   i   i   i   iÿÿÿÿiþÿÿÿ(   i    i    i   i    i    (   i    i   i   i   i   (   i   i   i   i   i   (   i    i   i   i   i   (   i    i   i   i   i   (   i    i    i   i    i   (   i    i   i   i   i   (   i   i   i   i   i   (   i    i    i    i    i    (   i    i   i   i   i   (   i   i   i   i   i   (   i    i   i   i   i   (   i    i   i   i   i   (   i    i    i    i    i    (   i    i   i   i   i   (   i   i   i   i   i   (   i    i    i   i    i    (   i    i   i   i   i   (   i   i   i   i   i   (   i    i   i   i   i   (   i    i   i   i   (   i    i    i   i   (   i    i   i   i   (   i   i   i   i   (	   R   R@   R
   Rc   R.   Ri   R   RK   R   (   RA   RK   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
   test_edgesB  s:    
c         C` sð  t  t j d ƒ j d d ƒ d d d d d d d d d g d d d d d d d d g d d d d d d d d g g ƒ} t | d  ƒ } t | t d d d  ƒ t d d	 d  ƒ t d
 d d  ƒ g ƒ t | d ƒ } t | t d d d  ƒ t d d d  ƒ g t d d d  ƒ t d d d  ƒ g t d d d  ƒ t d d d  ƒ g t d d d  ƒ t d d d  ƒ g t d d d  ƒ g t d d d  ƒ g g  t d d d  ƒ g g ƒ t | d ƒ } t | t d d d  ƒ g g  t d d d  ƒ t d d d  ƒ g g ƒ d  S(   Ni   i   i   R7   i    i   i   i   i   i   i   i   i   (   R   R@   R
   Rc   R-   Ri   R   RU   (   RA   R8   t   tmp(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_contiguousd  s0    $!!!!(   R[   R\   R¡   R£   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR    ?  s   	"t   TestCompressFunctionsc           B` s>   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` s5  t  j t t d@ ƒ ƒ ƒ j d d d ƒ } t  j dA ƒ j t ƒ } t | dB <t | d | ƒ} t	 | ƒ } t
 | d d d d g d d	 d
 d g d d d d g g d d d d g d d d d g d d d d g g g ƒ t	 | d ƒ } t
 | d d d d d g d d d d d g d d  d	 d
 d g d d! d d d g g d d" d d d g d# d$ d% d& d' g d d( d d d g d d) d d d g g g ƒ t	 | d ƒ } t
 | d d d d d g d d  d	 d
 d g d d! d d d g g d* d+ d, d- d. g d/ d0 d1 d2 d3 g d4 d5 d6 d7 d8 g g d d" d d d g d d( d d d g d d) d d d g g g ƒ t	 | dC ƒ } t	 | d9 ƒ } t	 | dD ƒ } t
 | | ƒ t
 | | ƒ t
 | | ƒ t	 | d ƒ } t
 | d d d d g d d d d g d d	 d
 d g d d d d g g d* d, d- d. g d: d; d< d= g d/ d1 d2 d3 g d4 d6 d7 d8 g g d d d d g d# d% d& d' g d d d d g d d d d g g g ƒ t	 | dE ƒ } t	 | d> ƒ } t	 | dF ƒ } t
 | | ƒ t
 | | ƒ t
 | | ƒ t	 | dG ƒ } t
 | d d d d d g d d  d	 d
 d g d d! d d d g g d d" d d d g d d( d d d g d d) d d d g g g ƒ t	 | dH ƒ } t
 | | ƒ t	 | dI ƒ } t
 | d d d d g d d	 d
 d g d d d d g g d* d, d- d. g d/ d1 d2 d3 g d4 d6 d7 d8 g g d d d d g d d d d g d d d d g g g ƒ t	 | dJ ƒ } t	 | dK ƒ } t	 | dL ƒ } t
 | | ƒ t
 | | ƒ t
 | | ƒ t	 | dM ƒ } t
 | d d d d g d d d d g d d	 d
 d g d d d d g g d d d d g d# d% d& d' g d d d d g d d d d g g g ƒ t	 | dN ƒ } t
 | | ƒ d  S(O   Ni   i   i   i   R7   i    i   i
   i   i   i   i   i   i   i   i(   i*   i+   i,   i2   i4   i5   i6   i7   i9   i:   i;   i   i   i   i	   i   i   i)   i-   i.   i/   i0   i1   i3   i8   i   i   i   i   i   i   i   i    i!   i"   i#   i$   i%   i&   i'   iþÿÿÿi   i   i   i   iÿÿÿÿi   i<   (   i   i   i   (   i   i   i   (   i   (   iþÿÿÿ(   i   (   iÿÿÿÿ(   i    i   (   i    iþÿÿÿ(   i   i   (   iþÿÿÿi   (   i   iÿÿÿÿ(   iþÿÿÿiÿÿÿÿ(   i    i   (   i    iÿÿÿÿ(   R@   R	   t   listRH   Rc   R   t   astypeRY   Ra   R'   R   (   RA   Rn   R   R8   t   a2t   a3t   a4(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_compress_nd‡  s¶    *
c         C` s­  t  t j d ƒ j d d ƒ d d d d g d d d g d d d g g ƒ} t t | ƒ d d g d d	 g g ƒ t t | d ƒ d d d g d
 d d	 g g ƒ t t | d ƒ d d g d d g d d	 g g ƒ t  | j d d d d g d d d g d d d g g ƒ} t t | ƒ d d g d
 d	 g g ƒ t t | d ƒ d d d g d
 d d	 g g ƒ t t | d ƒ d d g d d g d
 d	 g g ƒ t  | j d d d d g d d d g d d d g g ƒ} t t | ƒ d	 g g ƒ t t | d ƒ d
 d d	 g g ƒ t t | d ƒ d g d g d	 g g ƒ t  | j d d d d g d d d g d d d g g ƒ} t t | ƒ j d ƒ t t | d ƒ j d ƒ t t | d ƒ j d ƒ d  S(   Ni	   i   R7   i   i    i   i   i   i   i   i   (   R	   R@   R
   Rc   R   R(   t   _dataRX   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_compress_rowcols÷  s"    -%.19%.19"(9c         C` sÁ  t  t j d ƒ j d d ƒ d d d d g d d d g d d d g g ƒ} t t | ƒ j d d d g d d d g d d d g g ƒ t t | d ƒ j d d d g d d d g d d d g g ƒ t t | d ƒ j d d d g d d d g d d d g g ƒ t  | j d d d d g d d d g d d d g g ƒ} t t | ƒ j d d d g d d d g d d d g g ƒ t t | d ƒ j d d d g d d d g d d d g g ƒ t t | d ƒ j d d d g d d d g d d d g g ƒ t  | j d d d d g d d d g d d d g g ƒ} t t | ƒ j d d d g d d d g d d d g g ƒ t t | d ƒ j d d d g d d d g d d d g g ƒ t t | d ƒ j d d d g d d d g d d d g g ƒ t  | j d d d d g d d d g d d d g g ƒ} t t | ƒ j	 ƒ  t
 k ƒ t t | d ƒ j	 ƒ  t
 k ƒ t t | d ƒ j	 ƒ  t
 k ƒ t t | ƒ j j	 ƒ  ƒ t t | d ƒ j j	 ƒ  ƒ t t | d ƒ j j	 ƒ  ƒ d  S(   Ni	   i   R7   i   i    (   R	   R@   R
   Rc   R   R)   R7   R«   R   t   allR   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_mask_rowcols  s:    -+++9+++9+++9c      	   C` s€  t  j d d ƒ } d d d d d d g } t | d | ƒj d d ƒ } t | d | ƒj d d ƒ } t | | d t ƒ} t | j d d g d d g g ƒ t | | d t ƒ} t | j d d d g d d d g d d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ d d d d d d g } t | d | ƒj d d ƒ } t | d | ƒj d d ƒ } t | | d t ƒ} t | j d d g d d g g ƒ t | | d t ƒ} t | j d d d g d d d g d d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | t | | ƒ ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ d d d d d d g } t | d | ƒj d d ƒ } t | d | ƒj d d ƒ } t | | ƒ } t | j t
 ƒ t | | ƒ } t | j t
 ƒ t | d d d d d d d g ƒj d d ƒ } t | d d d d d d d g ƒj d d ƒ } t | | d t ƒ} t | j d d g d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | | d t ƒ} t | j d d d g d d d g d d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | d d d d d d d g ƒj d d ƒ } t | d d d d d d d g ƒj d d ƒ } t | | d t ƒ} t | j d d g d d g g ƒ t | | ƒ } t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | | d t ƒ} t | j d d d g d d d g d d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | d d d d d d d g ƒj d d ƒ } t | d d d d d d d g ƒj d d ƒ } t | | d t ƒ} t | j d d g d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ t | | d t ƒ} t | j d d d g d d d g d d d g g ƒ t | | d t ƒ} t | t  j | j	 d ƒ | j	 d ƒ ƒ ƒ d  S(   Ni   i   i    R7   i   i   t   strict(   R@   R
   R   Rc   R   Ra   R   R7   Rb   t   filledR   (   RA   t   nR   R8   R9   R   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_dot+  sz    "4++"4++00"+4+00"+4+00"+4c         C` s›   t  j d ƒ } t | ƒ } t t t | | ƒ ƒ t k ƒ t t t | | ƒ ƒ t k ƒ t t t | | ƒ ƒ t k ƒ t t t | | ƒ ƒ t k ƒ d  S(   Ni   (   R@   t   eyeR	   R   t   typeR   R   (   RA   R8   R9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_dot_returns_maskedarrayq  s    c         C` s`   t  t j d ƒ ƒ } t  t j d ƒ ƒ } t | | d | ƒ} t | | k ƒ t | | ƒ d  S(   Ni   t   out(   i   i   (   R	   R@   R³   R   R   R   R   (   RA   R8   R¶   t   res(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_dot_outz  s
    (   R[   R\   Rª   R¬   R®   R²   Rµ   R¸   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR¤   …  s   	p		 	F		t   TestApplyAlongAxisc           B` s   e  Z d  „  Z d „  Z RS(   c         C` sY   t  d ƒ j d d d ƒ } d „  } t | d | ƒ } t | d d g d d g g ƒ d  S(	   Ng      (@i   i   c         S` s   |  d S(   Ni   (    (   R9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   myfunc‡  s    i   i   i   i
   (   R
   Rc   R&   R   (   RA   R8   Rº   t   xa(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_3d„  s    	c         C` sb   t  d ƒ j d d d ƒ } d d „ } t | d | d d ƒ} t | d d g d	 d
 g g ƒ d  S(   Ni   i   i   i    c         S` s   |  d | S(   Ni   (    (   R9   t   offset(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyRº   ‘  s    R½   i   i   i   i   (   R
   Rc   R&   R   (   RA   R8   Rº   R»   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_3d_kwargsŽ  s    (   R[   R\   R¼   R¾   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR¹   ‚  s   	
t   TestApplyOverAxesc           B` s   e  Z d  „  Z RS(   c         C` sÎ   t  d ƒ j d d d ƒ } t t j | d d g ƒ } t j d g d g d g g g ƒ } t | | ƒ t | | d j t	 ƒ <t t j | d d g ƒ } t j d	 g d
 g d g g g ƒ } t | | ƒ d  S(   Ni   i   i   i   i    i<   i\   i|   i   i,   (
   R
   Rc   R%   R@   RL   R	   R   R   R¦   RY   (   RA   R8   RB   t   ctrl(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
   test_basicš  s    $$(   R[   R\   RÁ   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR¿   ˜  s   t
   TestMedianc           B` sÎ   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 d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` sJ   t  j j t  j t  j g t  j t  j g g d d ƒ} t | t  j ƒ d  S(   NR^   iÿÿÿÿ(   R@   R   R   t   infR   (   RA   t   r(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_pytype¦  s    6c         C` sX  t  j j t  j j t  j t  j g t  j t  j g g ƒ d d ƒ} t | t  j ƒ t  j j t  j j t  j t  j g t  j t  j g g ƒ d d  ƒ} t | t  j ƒ t  j j t  j j t  j t  j g t  j t  j g g d t ƒd d ƒ} t | j t ƒ t  j j t  j j t  j t  j g t  j t  j g g d t ƒd d  ƒ} t | j t ƒ d  S(   NR^   iÿÿÿÿR7   (	   R@   R   R   R   RÃ   R   Ri   Ra   R7   (   RA   RÄ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_infª  s    !!!!!	!	c         C` s  t  j d ƒ } t t  j j | ƒ d ƒ t t t  j j | ƒ ƒ t k	 ƒ t d ƒ } t t  j j | ƒ d ƒ t t t  j j | ƒ ƒ t k	 ƒ d } t t  j j | ƒ d ƒ t t t  j j | ƒ ƒ t k	 ƒ t  j d ƒ j	 d d ƒ } t t  j j | d d ƒt  j | d d ƒƒ t t  j j | d d	 ƒt  j | d d	 ƒƒ t t  j j | d d	 ƒt k	 ƒ t  j d ƒ j	 d d ƒ } t t  j j | d d ƒt  j | d d ƒƒ t t  j j | d d	 ƒt  j | d d	 ƒƒ t t  j j | d d	 ƒt k	 ƒ d  S(   Ni	   g      @i   g      @i   g      @R^   i    i   g       @iH   g      R@(
   R@   R
   R   R   R   R   R´   R   RH   Rc   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_non_masked¼  s"    """.."..c         C` sx  t  t j d ƒ d d g d d g d ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t t t j j | ƒ ƒ t	 k	 ƒ t  t j d ƒ j
 d	 d
 ƒ d d g d d g d ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t t t j j | ƒ ƒ t	 k	 ƒ t j j | d d d t ƒ} t | d d g ƒ t | j d d ƒ t t | ƒ t	 k ƒ d S(   s5   test the examples given in the docstring of ma.mediani   R7   i    i   i   g      ø?s   shape mismatchi
   i   i   i   g      @R^   iÿÿÿÿt   overwrite_inputg       @g      @N(    (    (   i   (   R	   R@   R
   R   R   R   R   R   R´   R   Rc   Ra   (   RA   Rn   t   ma_x(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_docstring_examplesÑ  s    -"9"c      	   C` sf  d } xYt  d ƒ D]K} xBt t g D]4} t d d | d | ƒ} t j t  | | ƒ t t g ƒ } xd | D]\ \ } } y  t j j | d | d | ƒWqm t	 k
 rÈ t
 | | | | | f ƒ ‚ qm Xqm Wt j | d | g t t g ƒ } xg | D]_ \ } } y  t j j | d | d | ƒWn t j k
 r9q÷ Xt
 | | | | | f ƒ ‚ q÷ Wq& Wq Wd  S(   Ns5   mask = %s, ndim = %s, axis = %s, overwrite_input = %si   i   t   ndminR7   R^   RÈ   (   RH   Rb   Ra   R	   t	   itertoolst   productR@   R   R   t	   Exceptiont   AssertionErrort	   AxisError(   RA   t   msgRË   R7   Rn   t   argsR^   t   over(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_axis_argument_errorsà  s"    " $# c         C` s`   t  d d t ƒ} t t j j | ƒ d ƒ t  d d t ƒ} t t j j | ƒ t j j ƒ d  S(   Ni   R7   (   R	   Rb   R   R@   R   R   Ra   R   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_masked_0dø  s    c      	   C` s  t  t j d ƒ d t ƒ} t t j j | ƒ t j j ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t j j j k ƒ t  t j d ƒ d t ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ t  t j d ƒ d d d d d d g ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ t  t j d ƒ d d d d d d g ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ t  t j d ƒ d d d d d d g ƒ} t t j j | ƒ d	 ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ t  t j d
 ƒ d d d d d d g ƒ} t t j j | ƒ d	 ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ t  t j d ƒ d d d d d d d g ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ t  t j d ƒ d d d d d d d g ƒ} t t j j | ƒ d ƒ t t j j | ƒ j d d ƒ t	 t
 t j j | ƒ ƒ t k	 ƒ d  S(   Ni   R7   s   shape mismatchg       @i    i   g      @g        g      @g      @i   g      @(    (    (    (    (    (    (    (    (   R	   R@   R
   Ra   R   R   R   R   R   R   R´   t   coret   MaskedConstantRb   R   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_masked_1dÿ  s@    +"*"*"*"*"-"-c      	   C` sk   t  t j j t d d d g d d d d g ƒƒ j t j j t d d d g d d d d g ƒƒ j ƒ d  S(   Ni   i   i   R7   i    (   R   R@   R   R   R	   R   (   RA   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_1d_shape_consistency%  s    3c         C` sK  d \ } } t  t j d d | ƒ ƒ } t | d *| d )t  t j | | f d t ƒƒ } | | d  d  … d f <t j t | ƒ ƒ } xA t d	 | ƒ D]0 } t j	 j
 | ƒ | | | d  d  … | f <q– Wt t | d  d  … d f ƒ d ƒ t t | ƒ d ƒ t t | d
 d ƒt j | ƒ ƒ t t | j d
 d	 ƒt j | ƒ ƒ d  S(   Nie   i   g      ð¿g      ð?i
   iöÿÿÿR6   i    i   R^   (   ie   i   (   R   R@   t   linspaceR   t   emptyR?   R
   t   lenRH   R’   t   shuffleR   R   R   t   T(   RA   R±   t   pRn   Ru   t   idxRN   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyRœ   )  s    !#"c         C` sO  t  t j d ƒ j d d ƒ ƒ } t | d *| d )t t | ƒ d ƒ t t t j	 j | ƒ ƒ t
 k	 ƒ t t | d d ƒd d d	 g ƒ t t t j	 j | d d ƒƒ t
 k ƒ t t | d d
 ƒd d d d d d d d d d g
 ƒ t t t j	 j | d d
 ƒƒ t
 k ƒ t t | d d
 ƒj d
 d
 d
 d d d d d
 d
 d
 g
 ƒ d  S(   Ni   i
   i   iýÿÿÿg      -@R^   i    g      +@g      /@i   i   i   i   (   R   R@   R
   Rc   R   R   R   R   R´   R   R   R7   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_2d_waxis9  s    !""(7(c         C` s  t  j j d ƒ j d d d ƒ } t | | d d k <t t | d ƒ d d g d d	 g d d g d
 d	 g g ƒ d | _ t t | d ƒ d d g d d g d d g g ƒ t  j j d ƒ j d d d ƒ } t | | d d k <t t | d ƒ d d g d d g d d g g ƒ d  S(   Ni   i   i   i   i    i   i	   i   i   i   ic   i
   i   i   i   i   i   i   i   (   i   i   i   (   R@   R   R
   Rc   R   R   R   R   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR¼   E  s    !:	1!c         C` s\   t  t j d ƒ j d d ƒ ƒ } t | d *| d )t t | d d ƒt | d d ƒƒ d  S(   Ni   i
   i   iýÿÿÿR^   iÿÿÿÿi   (   R   R@   R
   Rc   R   R   R   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_neg_axisP  s    !c         C` s²   x« d
 D]£ } t  t j | ƒ ƒ } t | d *| d )t  t j d ƒ ƒ } t | d | ƒ} | d k rw t | d ƒ n t | d	 ƒ t | | k ƒ t t | ƒ t	 k ƒ q Wd  S(   Ni   g      >@i   g      ?@i   iýÿÿÿR¶   g      -@g      .@(   i   g      >@i   g      ?@(    (
   R   R@   R
   R   R   R   R   R   R´   R   (   RA   t   vRn   R¶   RÄ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_out_1dU  s    c         C` sY  xRd D]J} t  t j | ƒ j d d ƒ ƒ } t | d *| d )t  t j d ƒ ƒ } t | d	 d
 d | ƒ} | d k rÍ t  d g d d d d d g d g d d t g d t g d t g d ƒ} nQ t  d g d d d d d g d g d d t g d t g d t g d ƒ} t	 | | ƒ t
 | | k ƒ t
 t | ƒ t k ƒ q Wd  S(   Ni(   g      D@i   g      >@i
   iÿÿÿÿi   iýÿÿÿR^   i   R¶   g        i   i   i   R7   i   g      +@g     €1@g     €5@g     €9@(   i(   g      D@i   g      >@(   R   R@   R
   Rc   R   R   R   Ra   Rb   R   R   R´   R   (   RA   Rã   Rn   R¶   RÄ   t   e(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_outc  s    !+)+&c         C` se   d d g d d g d d g g } t  j j | d ƒ } d d g } t t  j j | d d ƒ| ƒ d  S(   Ng      ð?g        g      @i    R^   (   R@   R   t   masked_equalR   R   (   RA   RK   t
   masked_arrR‚   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt$   test_single_non_masked_value_on_axist  s    		c      
   C` s²  t  ƒ  £} | j t ƒ x›t t j d d t ƒf D]~} t j j d t j	 d g d d d g g ƒ } | | _
 t j j | d d  ƒ} t t j | ƒ ƒ t | t j	 ƒ t j j | j ƒ  d d ƒ} t t j | ƒ ƒ t | t j	 ƒ t j j | d d ƒ} t t | ƒ t ƒ t | d t j	 d g ƒ t j j | d d ƒ} t t | ƒ t ƒ t | t j	 d g ƒ t j j | d d ƒ} t t | ƒ t ƒ t | t j	 d g ƒ q5 Wt j j d t j	 d g d d d g g ƒ } t j j | d  d  … d f <t t j j | d d  ƒt j	 ƒ t t j j | d d ƒd t j	 d g ƒ t t j j | d d ƒt j	 d	 g ƒ t g  | j D] } | j t k ^ q‚t g d
 ƒ Wd  QXd  S(   Ni   R6   i   i   i   R^   i    iÿÿÿÿg      ø?i   (   R   t   recordt   RuntimeWarningRb   R@   R   RY   R   R	   t   nanR7   R   Ri   R   t   isscalarR   t   ravelR   R´   R   R   t   logt   categoryRa   (   RA   t   wR7   t   dmRÄ   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_nan}  s6    "-	-"+(%c      
   C` s  t  j d t ƒ t  j d d t ƒ t j j t j d
 ƒ ƒ } t j j t j	 d ƒ ƒ } t j
 | d <t j j | d <t t j j | d d	 | ƒ| ƒ t j j t j d ƒ ƒ } t t j j | d d	 | ƒ| ƒ t j j t j d ƒ ƒ } t t j j | d	 | ƒ| ƒ Wd  QXd  S(   NRê   t   alwayst    i   i   i   i   i    R¶   (   i   (   i   i   (   i   i   (   i   i   (   i   (    (   t   warningst   catch_warningsRa   t   filterwarningsRë   R@   R   R   R   R   Rì   R   R   R   (   RA   t   oRŽ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_out_nanž  s    ""c         C` s…  t  j j t  j d d t ƒƒ } t  j j | d  d  d … <t  j | d <t ƒ  † } | j t	 ƒ t
 t  j j | ƒ t  j ƒ t
 t  j j | d d ƒt  j ƒ t | j d j t	 k ƒ t | j d j t	 k ƒ Wd  QXt  j j t  j d d t ƒj d d d ƒ ƒ } t  j | j ƒ d d k | _ | j ƒ  } t  j | d <t  j | d <t ƒ  | } | j t	 ƒ t j d	 d
 t	 ƒ t
 t  j j | ƒ t  j ƒ t t  j t  j j | ƒ ƒ ƒ t | j d j t	 k ƒ Wd  QXt  j j | d d ƒ} t  j | d <t  j | d <t j d t ƒ I } t j d	 d
 t	 ƒ t t  j j | d ƒ | ƒ t t | ƒ d ƒ Wd  QXt  j j | d d ƒ} t  j | d <t  j | d <t j d t ƒ I } t j d	 d
 t	 ƒ t t  j j | d ƒ | ƒ t t | ƒ d ƒ Wd  QXt  j j | d d ƒ} t  j | d <t  j | d <t j d t ƒ I } t j d	 d
 t	 ƒ t t  j j | d ƒ | ƒ t t | ƒ d ƒ Wd  QXd  S(   Ni   R6   i   i   R^   i    i   i   Rô   Rõ   Rê   (   i   i   i   (   i   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i    i   (   i    i   (   R@   R   R   R
   R?   R   Rì   R   Rê   Rë   R   R   R   Rï   Rð   Rc   RX   R7   t   copyRö   Rø   Rí   R÷   Ra   R   RÜ   (   RA   R8   Rñ   t   aorigR9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_nan_behavior«  sR    !" 0 c         C` sÏ   t  j d d d g d d d g g d t  j ƒ} t  j j | d | d k ƒ} t t  j j | d d ƒd ƒ t t  j j | d d ƒj t ƒ t t  j j | d d ƒ| d ƒ t t  j j | ƒ d ƒ d  S(   Ni   iÿ   R6   R7   R^   i   i    (	   R@   R	   t   uint8R   R   R   R   R7   Rb   (   RA   R8   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_ambigous_fillß  s    0"#c         C` s?  x8t  j t  j g D]#} t  j | t  j g t  j t  j g g ƒ } t  j j | d t  j | ƒ ƒ} t t  j j | d d ƒ| t  j g ƒ t t  j j | d d ƒ| t  j g ƒ t t  j j | ƒ | ƒ t  j t  j t  j | g t  j t  j | g g ƒ } t  j j | d t  j | ƒ ƒ} t	 t  j j | d d ƒ| ƒ t	 t  j j | d d ƒj
 t ƒ t	 t  j j | d d ƒ| d ƒ t	 t  j j | ƒ | ƒ t  j | | g | | g g ƒ } t t  j j | ƒ | ƒ t t  j j | d d ƒ| ƒ t t  j j | d d ƒ| ƒ t  j | d | d g d t  j t  j d g d	 t  j t  j | g g d
 t  j ƒ} t  j j | d t  j | ƒ ƒ} | d k rÞt t  j j | d d ƒd d | d g ƒ t t  j j | ƒ d ƒ nE t t  j j | d d ƒd d | d g ƒ t t  j j | ƒ d ƒ t t  j j | d d ƒd d | g ƒ xé t d d ƒ D]Ø } xÏ t d d ƒ D]¾ } t  j t  j g | | g | g d ƒ } t  j j | d t  j | ƒ ƒ} t t  j j | ƒ | ƒ t t  j j | d d ƒ| ƒ t t  j j | d d ƒt  j g | | g | ƒ qqWq[Wq Wd  S(   NR7   R^   i    i   i   i÷ÿÿÿiöÿÿÿi   i   R6   g      @g      @g      @g      @g      $Àg      "Àg      Àg      ð¿i
   i   (   R@   RÃ   R	   Rì   R   R   t   isnanR   R   R   R7   Rb   t   float32RH   (   RA   RÃ   R8   RN   RO   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_specialè  sF    *!((3!"#!!,,(+!c         C` sü  t  j j t  j g  d t ƒƒ } t ƒ  J } | j t ƒ t t  j j	 | ƒ t  j
 ƒ t | j d j t k ƒ Wd  QXt  j j t  j g  d t d d ƒƒ } t ƒ  ] } | j t ƒ t j d d t ƒ t t  j j	 | ƒ t  j
 ƒ t | j d j t k ƒ Wd  QXt  j j t  j g  d t d d ƒƒ } t t  j j	 | d d ƒ| ƒ t t  j j	 | d d	 ƒ| ƒ t  j j t  j t  j
 d t d d ƒƒ } t j d
 t ƒ P } t j d d t ƒ t t  j j	 | d d ƒ| ƒ t | d j t k ƒ Wd  QXd  S(   NR6   i    RË   i   Rô   Rõ   i   R^   i   Rê   (   R@   R   R   R	   R?   R   Rê   Rë   R   R   Rì   R   Rï   Rð   Rö   Rø   R   R÷   Ra   (   RA   R8   Rñ   R9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
   test_empty  s&    ! ' '*c         C` s|   t  j j t  j d ƒ ƒ } t t t  j j | j t ƒ ƒ ƒ t	 ƒ t  j
 | d <t t t  j j | j t ƒ ƒ ƒ t	 ƒ d  S(   Ng      @i   (   R@   R   R   R
   R   R´   R   R¦   t   objectR?   Rì   (   RA   Rù   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_object0  s    ((   R[   R\   RÅ   RÆ   RÇ   RÊ   RÔ   RÕ   RØ   RÙ   Rœ   Rá   R¼   Râ   Rä   Ræ   Ré   Ró   Rú   Rý   Rÿ   R  R  R  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyRÂ   ¥  s,   							&										!		4			+	t   TestCovc           B` s5   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` s   t  t j j d ƒ ƒ |  _ d  S(   Ni   (   R	   R@   R’   R“   RK   (   RA   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   setup9  s    c         C` s…   |  j  } t t j | ƒ t | ƒ ƒ t t j | d t ƒt | d t ƒƒ t t j | d t d t ƒt | d t d t ƒƒ d  S(   Nt   rowvart   bias(   RK   R   R@   R   Rb   Ra   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_1d_without_missing<  s
    	(c         C` s‘   |  j  j d d ƒ } t t j | ƒ t | ƒ ƒ t t j | d t ƒt | d t ƒƒ t t j | d t d t ƒt | d t d t ƒƒ d  S(   Ni   i   R  R	  (   RK   Rc   R   R@   R   Rb   Ra   (   RA   Rn   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_2d_without_missingD  s
    (c      	   C` s¸  |  j  } t | d <| | j ƒ  8} | j ƒ  } t t j | ƒ t | ƒ ƒ t t j | d t ƒt | d t ƒƒ t t j | d t d t ƒt | d t d t ƒƒ y t | d t ƒWn t	 k
 rÎ n X| d d !} t t j | | d  d  d … ƒ t | | d  d  d … ƒ ƒ t t j | | d  d  d … d t ƒt | | d  d  d … d t ƒƒ t t j | | d  d  d … d t d t ƒt | | d  d  d … d t d t ƒƒ d  S(   NiÿÿÿÿR  R	  t   allow_maskedi   (
   RK   R   t   meanRM   R   R@   R   Rb   Ra   t
   ValueError(   RA   Rn   t   nx(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_1d_with_missingL  s$    	
(<%#+c         C` s¥  |  j  } t | d <| j d d ƒ } t j t | ƒ ƒ j t ƒ } t j | | j	 ƒ } | | j
 d ƒ d  d  … d  f j d ƒ } t t | ƒ t j | ƒ | j d d | d ƒ t t | d t ƒt j | d t ƒ| j d | ƒ t j | j	 | ƒ } | | j
 d ƒ j d ƒ } t t | d t ƒt j | d t ƒ| j d d | d ƒ t t | d t d t ƒt j | d t d t ƒ| j d | ƒ d  S(	   Niÿÿÿÿi   i   i   i    g      ð?R	  R  (   RK   R   Rc   R@   t   logical_notR   R¦   RI   R   RÞ   R  Ri   R°   R   R   R   Ra   Rb   (   RA   Rn   t   validt   fract   xf(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_2d_with_missingd  s     	
,'%-(   R[   R\   R  R
  R  R  R  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR  7  s
   				t   TestCorrcoefc           B` sG   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` s:   t  t j j d ƒ ƒ |  _ t  t j j d ƒ ƒ |  _ d  S(   Ni   (   R	   R@   R’   R“   RK   t   data2(   RA   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR  |  s    c      
   C` s  |  j  |  j } } t j | ƒ } t j | | ƒ } t ƒ  Ì } t j d ƒ t t t | d d ƒ| j	 t d ƒ t
 t j | d d ƒt | d d ƒƒ t
 t | d d ƒ| ƒ t
 t | | d d ƒ| ƒ t
 t | d d ƒ| ƒ t
 t | | d d ƒ| ƒ Wd  QXd  S(   NRô   t   ddofiÿÿÿÿs   bias and ddof have no effecti    i   (   RK   R  R@   R   R   Rö   t   simplefilterR   t   DeprecationWarningt   filterR   (   RA   Rn   Ro   R‚   t	   expected2t   sup(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt	   test_ddof€  s    (c         C` s¶   |  j  |  j } } t j | ƒ } t ƒ  … } t j d ƒ t t t | | t	 t
 ƒ t t t | | t	 t	 ƒ t t t | d t
 ƒ| j t d ƒ t t | d d ƒ| ƒ Wd  QXd  S(   NRô   R	  s   bias and ddof have no effecti   (   RK   R  R@   R   R   Rö   R  R   R  Ra   Rb   R  R   (   RA   Rn   Ro   R‚   R  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt	   test_bias  s    c         C` s§   |  j  } t t j | ƒ t | ƒ ƒ t t j | d t ƒt | d t ƒƒ t ƒ  K } | j t d ƒ t t j | d t d t ƒt | d t d t ƒƒ Wd  QXd  S(   NR  s   bias and ddof have no effectR	  (	   RK   R   R@   R   Rb   R   R  R  Ra   (   RA   Rn   R  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR
    s    	c         C` s³   |  j  j d d ƒ } t t j | ƒ t | ƒ ƒ t t j | d t ƒt | d t ƒƒ t ƒ  K } | j t d ƒ t t j | d t d t	 ƒt | d t d t	 ƒƒ Wd  QXd  S(   Ni   i   R  s   bias and ddof have no effectR	  (
   RK   Rc   R   R@   R   Rb   R   R  R  Ra   (   RA   Rn   R  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR  ¨  s    c         C` s,  |  j  } t | d <| | j ƒ  8} | j ƒ  } t t j | ƒ t | ƒ ƒ t t j | d t ƒt | d t ƒƒ t ƒ  K } | j	 t
 d ƒ t t j | d t d t ƒt | d t d t ƒƒ Wd  QXy t | d t ƒWn t k
 rð n X| d d !} t t j | | d  d  d … ƒ t | | d  d  d … ƒ ƒ t t j | | d  d  d … d t ƒt | | d  d  d … d t ƒƒ t ƒ  › } | j	 t
 d ƒ t t j | | d  d  d … ƒ t | | d  d  d … d d ƒƒ t t j | | d  d  d … ƒ t | | d  d  d … d d ƒƒ Wd  QXd  S(	   NiÿÿÿÿR  s   bias and ddof have no effectR	  R  i   R  i   (   RK   R   R  RM   R   R@   R   Rb   R   R  R  Ra   R  (   RA   Rn   R  R  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR  ³  s2    	
<%##c      
   C` sn  |  j  } t | d <| j d d ƒ } t | ƒ } t j | ƒ } t | d  d … d  d … f | d  d … d  d … f ƒ t ƒ  æ } | j t d ƒ t t | d d ƒd  d … d  d … f | d  d … d  d … f ƒ t t | d d ƒd  d … d  d … f | d  d … d  d … f ƒ t t | d d ƒd  d … d  d … f | d  d … d  d … f ƒ Wd  QXd  S(	   Niÿÿÿÿi   i   s   bias and ddof have no effectR  iþÿÿÿR	  i   (	   RK   R   Rc   R   R@   R   R   R  R  (   RA   Rn   RB   RC   R  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR  Ñ  s    	
9((((	   R[   R\   R  R  R  R
  R  R  R  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR  z  s   						t   TestPolynomialc           B` s   e  Z d  „  Z d „  Z RS(   c         C` sÖ  t  j j d ƒ } t  j j d ƒ j d d ƒ } t t | | d ƒ t  j | | d ƒ ƒ | j t ƒ } t | d <| j t ƒ } t | d
 <| d <t | | d  d  … d f d d t	 ƒ\ } } } } } t  j | d | d d  … d f j
 ƒ  d d t	 ƒ\ } }	 }
 } } xH t | | | | | f | |	 |
 | | f ƒ D] \ } } t | | ƒ q;Wt | | d  d  … d f d d t	 ƒ\ } } } } } t  j | d d !| d d … d f d d t	 ƒ\ } }	 }
 } } xH t | | | | | f | |	 |
 | | f ƒ D] \ } } t | | ƒ qþWt | | d d t	 ƒ\ } } } } } t  j | d d !| d d … d  d  … f d d t	 ƒ\ } }	 }
 } } xH t | | | | | f | |	 |
 | | f ƒ D] \ } } t | | ƒ q·Wt  j j d ƒ d } | j ƒ  } | d d !} | d d !} | d d !} t | | d d t	 d	 | ƒ\ } } } } } t  j | | d d t	 d	 | ƒ\ } }	 }
 } } t | | ƒ xH t | | | | | f | |	 |
 | | f ƒ D] \ } } t | | ƒ qµWd  S(   Ni
   i   iÿÿÿÿi   i   i    RJ   i   Rñ   (   i    i    (   iÿÿÿÿiÿÿÿÿ(   R@   R’   R“   Rc   R   R   t   viewR   R   Ra   RM   t   zipRû   R   (   RA   Rn   Ro   t   Ct   Rt   Kt   St   DR   RÄ   RP   RR   RŽ   R8   t   a_Rñ   t   wot   xst   yst   ws(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_polyfitç  s<    (
7,:7A:'G:-0:c         C` s9  t  j j d ƒ } t  j j d ƒ j d d ƒ } t  j | d <t  j | d	 <| j t ƒ } | j t ƒ } t | d <t | d
 <t | | d d t	 ƒ\ } } } } } t  j | d d !| d d … d  d  … f d d t	 ƒ\ } }	 }
 } } xH t
 | | | | | f | |	 |
 | | f ƒ D] \ } } t | | ƒ qWd  S(   Ni
   i   iÿÿÿÿi   i    i   RJ   i   (   iÿÿÿÿiÿÿÿÿ(   iÿÿÿÿiÿÿÿÿ(   R@   R’   R“   Rc   Rì   R!  R   R   R   Ra   R"  R   (   RA   Rn   Ro   R#  R$  R%  R&  R'  R   RÄ   RP   RR   RŽ   R8   R(  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_polyfit_with_masked_NaNs  s    

'G:(   R[   R\   R-  R.  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR   å  s   	't   TestArraySetOpsc           B` s˜   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 d „  Z d „  Z d „  Z d „  Z RS(   c         C` s·   d d d d d d g } t  | d t d t ƒ} t t | d t ƒ ƒ t | d t d d d g d d d d g ƒƒ t | d d d d g ƒ t | d d d d d d d g ƒ d  S(	   Ni   i   i   t   return_indext   return_inversei    R7   i   (   R   Ra   R   RŒ   R   R   R   (   RA   RK   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_unique_onlist!  s    /c      
   C` sœ  t  d d d d d d g d d d d d d d g ƒ} t | d t d t ƒ} t | d t  d d d d g d d d d d g ƒƒ t | d d d d	 d g ƒ t | d d d d d d d g ƒ d | _ t  d
 d d d d d d g d d d d d d d g d d ƒ } t | d t d t ƒ} t | d t  d d d d g d d d d d g ƒƒ t | d d d d	 d g ƒ t | d d d d d d d g ƒ d  S(   Ni   i   i   R7   i    R0  R1  iÿÿÿÿi   RK   t
   fill_value(   R   R   Ra   R   R3  (   RA   RK   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_unique_onmaskedarray*  s    65#	!5c         C` så   t  d d d g d t ƒ} t | d t d t ƒ} t | d t  d g d t g ƒƒ t | d d g ƒ t | d d d d g ƒ t } t | d t d t ƒ} t | d t  t ƒ ƒ t | d d g ƒ t | d d g ƒ d  S(   Ni   R7   R0  R1  i    i   (   R   Ra   R   R   R   (   RA   RK   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_unique_allmasked:  s    #c         C` s£   t  t j d ƒ d d d d d d g ƒ} t d d d d g d d d d d g ƒ} t | ƒ } t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ d  S(   Ni   R7   i   i    i   (   R   R@   R
   R	   R$   R   R°   R7   (   RA   Rn   RC   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_ediff1dI  s    **c      
   C` sE  t  t j d ƒ d d d d d d g ƒ} t | d t ƒ} t d d d d d g d d d d d d g ƒ} t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ t | d d d d g ƒ} t d d d d d d d g d d d d d d d d g ƒ} t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ d  S(	   Ni   R7   i   i    t   to_begini   i   i   (	   R   R@   R
   R$   R   R	   R   R°   R7   (   RA   Rn   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_ediff1d_tobeginR  s    *0<c      
   C` sE  t  t j d ƒ d d d d d d g ƒ} t | d t ƒ} t d d d d d g d d d d d d g ƒ} t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ t | d d d d g ƒ} t d d d d d d d g d d d d d d d d g ƒ} t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ d  S(	   Ni   R7   i   i    t   to_endi   i   i   (	   R   R@   R
   R$   R   R	   R   R°   R7   (   RA   Rn   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_ediff1d_toenda  s    *0<c         C` s]  t  t j d ƒ d d d d d d g ƒ} t | d t d t ƒ} t d d d d d d g d d d d d d d g ƒ} t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ t | d d d d	 g d t ƒ} t d d d d d d d d	 g d d d d d d d d d g ƒ} t | | ƒ t | j d ƒ | j d ƒ ƒ t | j | j ƒ d  S(
   Ni   R7   i   i    R9  R7  i   i   i   (	   R   R@   R
   R$   R   R	   R   R°   R7   (   RA   Rn   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_ediff1d_tobegin_toendp  s    *6!!!c      	   C` s.  t  j d ƒ } t | ƒ } t d d d d g d d d d d g ƒ} t | | ƒ t t | t ƒ ƒ t | j d ƒ | j d ƒ ƒ t | j	 | j	 ƒ t | d t
 d t
 ƒ} t d d d d d d g d d d d d d d g ƒ} t t | t ƒ ƒ t | j d ƒ | j d ƒ ƒ t | j	 | j	 ƒ d  S(   Ni   i   R7   i    R9  R7  (   R@   R
   R$   R	   R   R   RŒ   R   R°   R7   R   (   RA   Rn   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_ediff1d_ndarray€  s    *6c         C` s˜   t  d d d d g d d d d d g ƒ} t  d d d d g d d d d d g ƒ} t | | ƒ } t  d d d g d d d d g ƒ} t | | ƒ d  S(   Ni   i   R7   i    iÿÿÿÿ(   R	   R"   R   (   RA   Rn   Ro   RB   RC   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_intersect1d  s
    **$c      	   C` s0  t  d d d d d g d d d d d d g ƒ} t  d d d d	 d d g d d d d d d d g ƒ} t | | ƒ } t | t  d d	 d g ƒ ƒ t  d d d d d g d d d d d d g ƒ} d d d d	 d g } t | | ƒ } t | t  d d	 d d g d d d d d g ƒƒ t  d d d g ƒ } t  d
 d d	 g ƒ } t | | ƒ } t t | t ƒ ƒ t | d d d d	 d d
 g ƒ t  d d d d g d d d d d g ƒ} t  d
 d d	 d g d d d d d g ƒ} t | | ƒ } t t | t ƒ ƒ t | d d d d	 d d
 g ƒ t g  t g  g  ƒ ƒ d  S(   Ni   i   i   i   iÿÿÿÿR7   i    i   i   i   i   (   R	   R   R   R   RŒ   R   R   (   RA   R8   R9   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_setxor1d˜  s&    0601**c         C` s2  t  j d ƒ j d d d g ƒ } t  j d d d g ƒ } d | d <t | d | ƒ} t d d d	 d
 d d d d d d g	 d d d d d d d d d d g	 ƒ } t d d t ƒ} t | d <t | d <t | d <t | | ƒ } t t	 | t
 ƒ ƒ t | | ƒ t  j | | | j ƒ | j @} t | | ƒ d  S(   Ni   i   i   i   i   i    R7   RK   i
   i   i   i   i   i!   R6   (   i   i   i    (   i   i   i   (   i    i    i    (   i    i    i   (   i    i   i   (   R@   R
   Rc   R   R	   RY   Ra   R1   R   RŒ   R   R   R7   (   RA   R8   R7   R9   t   ecR   RŽ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt	   test_isin²  s    !
'$


"c      	   C` s*  t  d d d d d g d d d d d d g ƒ} t  d d d d	 d d g d d d d d d d g ƒ} t | | ƒ } t | t t t t t g ƒ t  d d d d d g d d d d d d g ƒ} t  d d d g d d d d g ƒ} t | | ƒ } t | t t t t t g ƒ t g  t g  g  ƒ ƒ d  S(
   Ni   i   i   i   iÿÿÿÿR7   i    i   i   (   R	   R#   R   Ra   Rb   R   (   RA   R8   R9   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt	   test_in1dÇ  s    060$c      	   C` s6  t  d d d d d g d d d d d d g ƒ} t  d d d d	 d d g d d d d d d d g ƒ} t t j t | | ƒ ƒ t | | d
 t ƒƒ t  d d d d d g d d d d d d g ƒ} t  d d d g d d d d g ƒ} t t j t | | ƒ ƒ t | | d
 t ƒƒ t g  t g  g  d
 t ƒƒ d  S(   Ni   i   i   i   iÿÿÿÿR7   i    i   i   t   invert(   R	   R   R@   RB  R#   Ra   R   (   RA   R8   R9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_in1d_invertÕ  s    06.0$.c   	   
   C` s¢  t  d d d d d d g d d d d d d d g ƒ} t  d d d d	 d d g d d d d d d d g ƒ} t | | ƒ } t  d d d d	 d d d g d d d d d d d d g ƒ} t | | ƒ t  d d d g d d	 d g g d d d d g d d d g g ƒ} t  d d d d d	 g d d d d d d g ƒ} t  d d d d d	 d g d d d d d d d g ƒ} t | | ƒ } t | | ƒ t g  t g  g  ƒ ƒ d  S(
   Ni   i   i   i   iÿÿÿÿR7   i    i   i   (   R	   R!   R   R   (	   RA   R8   R9   RB   RC   Rn   Ro   t   ezRu   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_union1dá  s    66<B06c         C` s  t  d d d d d d d d g d d d d d d d d d g ƒ} t  d d d	 d	 d d d g ƒ } t | | ƒ } t | t  d d d
 g d d d d g ƒƒ t d ƒ } t d ƒ } t t | | ƒ t  d d g ƒ ƒ t  g  t j d g  ƒ} t t | g  ƒ j t j ƒ d  S(   Ni   i   i   i   i   i   R7   i    i   iÿÿÿÿi
   i   i	   (   R	   R    R   R
   R@   t   uint32R6   (   RA   R8   R9   RB   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_setdiff1dó  s    B!+"c         C` sV   t  j d d d g ƒ } t  j d d d g ƒ } t t | | ƒ t  j d g ƒ ƒ d  S(   NR8   R9   R   RR   (   R@   R	   R   R    (   RA   R8   R9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_setdiff1d_char_array   s    (   R[   R\   R2  R4  R5  R6  R8  R:  R;  R<  R=  R>  R@  RA  RC  RE  RG  RH  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyR/    s    																	t   TestShapeBasec           B` s   e  Z d  „  Z d „  Z RS(   c         C` sŸ   t  d d d g d d d d g ƒ} t | ƒ } t | j d ƒ t | j j | j j ƒ t | j d ƒ t | j j | j j ƒ t | j j | j j ƒ d  S(   Ni    i   i   R7   i   (   i   i   (   i   (   R   R   R   R   R7   RK   (   RA   R8   R9   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_atleast_2d	  s    $c         C` s%  t  d ƒ } t | j d ƒ t | j j | j ƒ t | j j | j ƒ t  d d ƒ } xJ | D]B } t | j d ƒ t | j j | j ƒ t | j j | j ƒ q^ Wt d ƒ } t | j d ƒ t | j j | j ƒ t | j j | j ƒ t d d ƒ } xJ | D]B } t | j d ƒ t | j j | j ƒ t | j j | j ƒ qWt d ƒ } t | j d ƒ t | j j | j ƒ t | j j | j ƒ t d d ƒ } xJ | D]B } t | j d	 ƒ t | j j | j ƒ t | j j | j ƒ q¦Wt d ƒ } t | j d
 ƒ t | j j | j j ƒ d  S(   Ng      ð?i   g       @(   i   (   i   (   i   i   (   i   i   (   i   i   i   (   i   i   i   (   i   i   (   R   R   R   R7   RK   R   R   R2   (   RA   R9   R8   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_shape_scalar  s<    (   R[   R\   RJ  RK  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyRI    s   	
t	   TestStackc           B` s#   e  Z d  „  Z d „  Z d „  Z RS(   c         C` sH  t  d d d g d d d d g ƒ} t  d d d g d d d d g ƒ} t | | g d d ƒ} t | j d
 ƒ t | j | d j ƒ t | j | d j ƒ t | | g ƒ } t | j | j ƒ t | j | j ƒ t | | g d d ƒ} t | j d ƒ t | j | d  d  … d f j ƒ t | j | d  d  … d f j ƒ d  S(   Ni    i   i   R7   i	   i   i   R^   i   (   i   i   (   i   i   (   R   R3   R   R   R   R7   R4   RK   (   RA   R8   R9   R   RŽ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_stack_1dA  s    $$#c         C` s6  t  d d d g d t ƒ} t  d d d g d t ƒ} t | | g d d ƒ} t | j d
 ƒ t | j | d j ƒ t | j | d j ƒ t | | g ƒ } t | j	 | j	 ƒ t | j | j ƒ t | | g d d ƒ} t | j d ƒ t | j | d  d  … d f j ƒ t | j | d  d  … d f j ƒ d  S(   Ni    i   i   R7   i	   i   i   R^   i   (   i   i   (   i   i   (
   R   Ra   Rb   R3   R   R   R   R7   R4   RK   (   RA   R8   R9   R   RŽ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_stack_masksS  s    #c   
      C` s¼  d } t  j j d d | ƒ } t  j j d d | ƒ } t  j j d d | ƒ j t ƒ } t  j j d d | ƒ j t ƒ } t | d | ƒ} t | d | ƒ} t | | g d d ƒ} d | }	 t | j |	 ƒ t	 | j
 | d j
 ƒ t	 | j
 | d j
 ƒ t | | g d d ƒ} | d }	 t | j |	 ƒ t	 | j
 | d j
 ƒ t	 | j
 | d j
 ƒ d } t  j j d d | ƒ } t  j j d d | ƒ } t  j j d d | ƒ j t ƒ } t  j j d d | ƒ j t ƒ } t | d | ƒ} t | d | ƒ} t | | g d d ƒ} d | }	 t | j |	 ƒ t	 | j
 | d j
 ƒ t	 | j
 | d j
 ƒ t | | g d d ƒ} | d }	 t | j |	 ƒ t	 | j
 | d j
 ƒ t	 | j
 | d j
 ƒ d  S(   Ni   i   i    i
   R7   R^   i   iÿÿÿÿ.i   i   (   i   i   (   i   (   i   (   .i    (   .i   (   i   i   i   i   (   i   (   i   (   .i    (   .i   (   R@   R’   t   randintR¦   RY   R   R3   R   R   R   R7   (
   RA   t   shpt   d1t   d2Rp   Rq   t   a1R§   R   t   c_shp(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   test_stack_nde  sD    !!

!!

(   R[   R\   RM  RN  RU  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyRL  ?  s   		(N   t   __doc__t
   __future__R    R   R   Rö   RÌ   t   numpyR@   t   numpy.testingR   R   t   numpy.ma.testutilsR   R   R   R   t   numpy.ma.coreR	   R
   R   R   R   R   R   R   R   R   R   t   numpy.ma.extrasR   R   R   R   R   R   R   R   R   R   R   R   R    R!   R"   R#   R$   R%   R&   R'   R(   R)   R*   R+   R,   R-   R.   R/   R0   R1   R2   R3   R4   R  R5   R]   R‹   R    R¤   R¹   R¿   RÂ   R  R  R   R/  RI  RL  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_extras.pyt   <module>	   s.   "LÐ
lƒ,Fýÿ “Ck:è8