ó
ÐEe]c           @` s¸  d  d l  m Z m Z m Z d  d l m Z d  d l Z d  d l j	 j
 Z
 d  d l j	 j Z d  d l 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# 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> m? Z? m@ Z@ mA ZA mB ZB mC ZC mD ZD mE ZE mF ZF mG ZG mH ZH mI ZI mJ ZJ mK ZK mL ZL mM ZM mN ZN mO ZO mP ZP mQ ZQ mR ZR mS ZS mT ZT mU ZU mV ZV mW ZW mX ZX mY ZY mZ ZZ m[ Z[ m\ Z\ d  d l] m^ Z^ e j_ Z_ d d „ Z` d	 ea f d
 „  ƒ  YZb d ea f d „  ƒ  YZc d ea f d „  ƒ  YZd d „  Ze d S(   i    (   t   divisiont   absolute_importt   print_function(   t   reduceN(   t   assert_t   assert_raisest   assert_equal(K   t   MaskTypet   MaskedArrayt   absolutet   addt   allt   allcloset   allequalt   alltruet   aranget   arccost   arcsint   arctant   arctan2t   arrayt   averaget   chooset   concatenatet	   conjugatet   cost   cosht   countt   dividet   equalt   expt   filledt   getmaskt   greatert   greater_equalt   innert   isMaskedArrayt   lesst
   less_equalt   logt   log10t	   make_maskt   maskedt   masked_arrayt   masked_equalt   masked_greatert   masked_greater_equalt   masked_insidet   masked_lesst   masked_less_equalt   masked_not_equalt   masked_outsidet   masked_print_optiont   masked_valuest   masked_wheret   maximumt   minimumt   multiplyt   nomaskt   nonzerot	   not_equalt   onest   outert   productt   putt   ravelt   repeatt   resizet   shapet   sint   sinht   sometruet   sortt   sqrtt   subtractt   sumt   taket   tant   tanht	   transposet   wheret   zeros(   t   picklet    c         C` s?   t  |  | ƒ } | s; t d | t |  ƒ t | ƒ f ƒ n  | S(   Ns   Not eq:%s
%s
----%s(   R   t   printt   str(   t   vt   wt   msgt   result(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   eq   s    &t   TestMac           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 d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` sz  t  j d d d d t d d d d d d d d g ƒ } t  j d d	 d d d
 d d	 d d d d	 d g ƒ } d } d d d d d d d d d d d d g } d d d d d d d d d d d d g } t | d | ƒ} t | d | ƒ} t  j d d	 d d g ƒ } t | d d d d d g ƒ}	 t  j | d | ƒ }
 | j } | j d ƒ | | | | | | | | |	 |
 | f |  _ d  S(   Ng      ð?g       Àg       @g      @g      @g      $Àg      $@g      @g        g      ð¿g      Ài   i    t   maskg      à¿g      à?gš™™™™™é?g@Œµx¯D(   t   npR   t   piRP   RD   t   set_fill_valuet   d(   t   selft   xt   yt   a10t   m1t   m2t   xmt   ymt   zt   zmt   xft   s(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   setup&   s    73**	c         C` s  |  j  \ } } } } } } } } }	 }
 } t t | ƒ ƒ t t | ƒ ƒ t t | ƒ | ƒ t | j | ƒ t | j | j ƒ t | j t d „  | ƒ ƒ t t | ƒ t	 | ƒ t d „  | ƒ ƒ t t
 | |
 ƒ ƒ t t
 t | d ƒ |
 ƒ ƒ t t
 | | ƒ ƒ d  S(   Nc         S` s   |  | S(   N(    (   Rb   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   <lambda>=   RS   c         S` s   |  | S(   N(    (   Rb   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyRn   >   RS   g@Œµx¯D(   R`   R   R$   R   RD   t   dtypet   sizeR   R   t   lenRZ   R   (   Ra   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   Rl   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testBasic1d5   s    *)c         C` sG  x@d d	 g D]2} |  j  \ } } } } } } } }	 }
 } } | | _ | | _ | | _ | | _ | | _ t t | ƒ ƒ t t | ƒ ƒ t t | ƒ | ƒ t | j | ƒ t | j t d „  | ƒ ƒ t t | ƒ t | ƒ t d „  | ƒ ƒ t t	 | | ƒ ƒ t t	 t
 | d ƒ | ƒ ƒ t t	 | | ƒ ƒ |  j ƒ  q Wd  S(
   Ni   i   i   i   c         S` s   |  | S(   N(    (   Rb   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyRn   Q   RS   c         S` s   |  | S(   N(    (   Rb   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyRn   S   RS   g@Œµx¯D(   i   i   (   i   i   (   R`   RD   R   R$   R   Rp   R   R   Rq   RZ   R   Rm   (   Ra   Rl   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testBasic2dC   s$    *					c         C` sÖ  |  j  \ } } } } } } } } }	 }
 } t d d g d d g g ƒ } t | d d g d d g g ƒ } t t | | | | ƒ ƒ t t | | | | ƒ ƒ t t | | | | ƒ ƒ xd d d g D]} | j | ƒ } | j | ƒ } | j | ƒ } | j | ƒ } |
 j | ƒ }
 t t | | ƒ ƒ t t | | | | ƒ ƒ t t | | | | ƒ ƒ t t | | | | ƒ ƒ t j d d	 d
 d	 ƒ   t t | | | | ƒ ƒ Wd  QXt t | | | | ƒ ƒ t t | | | | ƒ ƒ t t | | | | ƒ ƒ t j d d	 d
 d	 ƒ   t t | | | | ƒ ƒ Wd  QXt t | | | | ƒ ƒ t t | | | | ƒ ƒ t t | | | | ƒ ƒ t t | | | | ƒ ƒ t t | d | d ƒ ƒ t t t | ƒ d t | ƒ d ƒ ƒ t t | | | | ƒ ƒ t t t j	 | | ƒ t	 | | ƒ ƒ ƒ t t t j
 | | ƒ t
 | | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t j d d	 d
 d	 ƒ - t t t j | | ƒ t | | ƒ ƒ ƒ Wd  QXqÊ Wd  S(   Ni   i   i    i   i   i   i   R   t   ignoret   invalidg      @(   i   (   i   i   (   i   i   (   R`   R   R+   R   RZ   t   reshapeR]   t   errstatet   absR
   RJ   R9   R   (   Ra   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   Rl   t   a2dt   a2dm(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testArithmeticY   sF    *!!!'(((c         C` sS   t  j d g ƒ } t d g ƒ } t t | | t ƒ ƒ t t | | t ƒ ƒ d  S(   Ni   (   R]   R   R   t
   isinstanceR   (   Ra   t   nat   ma(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testMixedArithmetic   s    c         C` s3  |  j  \ } } } } } } } } }	 }
 } t t t j | ƒ t | ƒ ƒ ƒ t t t j | ƒ t | ƒ ƒ ƒ t t t j | ƒ t | ƒ ƒ ƒ t t t j | ƒ t | ƒ ƒ ƒ t t t j | ƒ t | ƒ ƒ ƒ t t t j	 | ƒ t	 | ƒ ƒ ƒ t j
 d d d d ƒ } t t t j t | ƒ ƒ t | ƒ ƒ ƒ t t t j t | ƒ ƒ t | ƒ ƒ ƒ t t t j t | ƒ ƒ t | ƒ ƒ ƒ Wd  QXt t t j | ƒ t | ƒ ƒ ƒ t t t j | ƒ t |	 ƒ ƒ ƒ t t t j | ƒ t |	 ƒ ƒ ƒ t t t j | ƒ t |	 ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | ƒ t | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | | ƒ t | | ƒ ƒ ƒ t t t j | ƒ t | ƒ ƒ ƒ t t t j | | f ƒ t | | f ƒ ƒ ƒ t t t j | | f ƒ t | | f ƒ ƒ ƒ t t t j | | f ƒ t | | f ƒ ƒ ƒ t t t j | | | f ƒ t | | | f ƒ ƒ ƒ d  S(   NR   Rt   Ru   (   R`   R   RZ   R]   R   R   RE   RF   RM   RN   Rw   RI   Rx   R'   R(   R   R   R   R   R   R	   R   R<   R%   R!   R&   R"   R   R   (   Ra   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   Rl   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testUfuncs1…   s8    *""""""((.""""("(((((("...c         C` s[  t  d d d d g d d d d d g ƒ} t t | ƒ j j t j k ƒ t d t | ƒ ƒ t d t d ƒ ƒ t t d t  d d d g ƒƒ ƒ | j	 d
 ƒ } t t | ƒ j j t j k ƒ t t
 t | d ƒ t j ƒ ƒ t t | ƒ j j t j k ƒ t t d t | ƒ ƒ ƒ t t t | d ƒ ƒ t k ƒ t t d d	 g t | d ƒ ƒ ƒ d  S(   Ng        g      ð?g       @g      @R\   i   i    i   i   (   i   i   (   R   R   R   Ro   t   typeR]   t   intpR   RZ   Rv   R|   t   ndarrayR    R:   (   Ra   t   ott(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_xtestCount¤   s    *"c         C` s   |  j  \ } } } } } } } } }	 }
 } t j | ƒ } t | ƒ } t t t | ƒ t j | ƒ ƒ ƒ t t t | ƒ t	 j | ƒ ƒ ƒ d  S(   N(
   R`   R]   RA   R   RZ   t   maxR7   R   t   minR8   (   Ra   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   Rl   t   xrt   xmr(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testMinMax³   s
    *"c         C` s©  |  j  \ } } } } } } } } }	 }
 } t t t j j | ƒ t j | ƒ ƒ ƒ t t t j j | ƒ t j | ƒ ƒ ƒ t t d t t d ƒ d d ƒƒ ƒ t t d t t d ƒ d d ƒƒ ƒ t t t j | d d ƒt | d d ƒƒ ƒ t t t j t	 | d ƒ d d ƒt | d d ƒƒ ƒ t t t j | d ƒ t | d ƒ ƒ ƒ t t t j
 | d d ƒt
 | d d ƒƒ ƒ t t t j
 | d ƒ t
 | d ƒ ƒ ƒ t t t j
 t	 | d ƒ d d ƒt
 | d d ƒƒ ƒ t | ƒ d k r¥t t t j | | f d ƒ t | | f d ƒ ƒ ƒ t t t j j | d ƒ t j | d ƒ ƒ ƒ t t t j | d ƒ t | d ƒ ƒ ƒ t t t j
 | d ƒ t
 | d ƒ ƒ ƒ n  d  S(   Ni   t   axisi    i   (   R`   R   RZ   R]   R
   R   t
   accumulateRK   R   R   R?   Rq   R   (   Ra   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   Rl   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testAddSumProd½   s$    *((%%.7(.(!.(c         C` s
  t  j d d d d g ƒ } t | d d d d d g ƒ} t | d d d d d g ƒ} t | ƒ } t | ƒ t | ƒ t t t  j | ƒ t | d d ƒƒ ƒ t t | d ƒ t | d ƒ k ƒ t | d | d k ƒ t | d t k ƒ t t | d | d ƒ ƒ t t | d d !| d d !ƒ ƒ t t | | ƒ ƒ t t | d | d ƒ ƒ d	 | d <d	 | d <t t | | ƒ ƒ d
 | d d +d
 | d d +t t | | ƒ ƒ t | d <t t | | ƒ ƒ t | d d +t t | | ƒ ƒ | | (t | d <t t	 t
 | ƒ t d d d d g ƒ ƒ ƒ t d d d d g d d d d g ƒ | (t t	 t
 | ƒ t d d d d g ƒ ƒ ƒ t d d d d g d d d d g ƒ | (t t	 t
 | ƒ t d d d d g ƒ ƒ ƒ t t	 | t d d d d g ƒ ƒ ƒ t  j d ƒ d } t | d ƒ } t t | | ƒ ƒ t t	 t d d d d d g t ƒ | j ƒ ƒ t t d | j ƒ ƒ t d d d d g t ƒ } t  j d d d d g t ƒ } | d } | d } t t | ƒ t ƒ t t | ƒ t ƒ t | | ƒ t | d d !j d k ƒ d  S(   Ni   i   i   i   R\   i    t
   fill_valuei   i	   ic   g      ð?g      @t   hello(   i    (   R]   R   RU   t   reprR   RZ   RH   R   R*   R   R    R+   R   R5   R   R\   RŽ   t   objectR   RD   (   Ra   t   x1t   x2t   x3t   x4t   s1t   s2(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testCIÒ   sZ    

($!



+(+(+%.

c         C` s  d d d d d g } t  | ƒ } t  | ƒ } t | | k ƒ t  | d d ƒ} t | | k	 ƒ t j d ƒ } t | d | ƒ} t | j | k	 ƒ t t | | j ƒ ƒ t | j | k ƒ t | d d ƒ} t | j j	 | j j	 k ƒ t | d | d d ƒ} t | j | k ƒ t | d t
 k ƒ d | d <t | d t
 k	 ƒ t | j | k ƒ t t | j d ƒ ƒ t | d | d d ƒ}	 t |	 j | k	 ƒ t |	 d t
 k ƒ d |	 d <t |	 d t
 k	 ƒ t |	 j | k	 ƒ t t |	 j d ƒ ƒ t | d d | ƒ}
 t t |
 ƒ j | d j k ƒ t d	 ƒ } t
 | d <t | d ƒ } t t t | | g ƒ | ƒ ƒ t t t | ƒ d d d d d d d d g ƒ ƒ t | d d d ƒ} t t | d d d d d d d d g ƒ ƒ t | d d d ƒ} t t | | ƒ ƒ d  S(   Ni    i   t   copyi   R\   i   i	   g      ð?i   i   R‹   i   (   i   (   i   i   i   i   (   R)   R   R]   R   R   t   _dataR   R\   t   _maskt   __array_interface__R*   R   Ro   RC   RZ   R   R    RB   (   Ra   t   nt   mRf   t   m3R’   t   y1t   y1at   y2t   y2at   y3R•   t   y4t   y5t   y6(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testCopySize  sN    

 
1+c         C` s  t  d ƒ } d d d d d g } t | ƒ } | j ƒ  } t | d | ƒ} t | d t k ƒ t | d t k ƒ d d g | d d g <t | j | k ƒ t | d t k ƒ t | d t k	 ƒ t t | d d d	 d
 d g ƒ ƒ t | d | d t ƒ} | j	 d d d	 g d
 d d g ƒ t | j | k	 ƒ t | d t k ƒ t | d t k ƒ t t | d
 d d d d g ƒ ƒ d  S(   Ni   i    i   R\   i   i   i
   i(   i   iÿÿÿÿR™   id   iÈ   (
   R   R)   R™   R   R   R*   R\   RZ   t   TrueR@   (   Ra   R`   R   Rž   Rf   Rb   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testPut7  s$    ""c         C` sà  t  d ƒ } t | d d d d d d g ƒ} t d d g d d d g ƒ} t | d t k	 ƒ t | d t k	 ƒ | | d d	 +t | d t k ƒ t | d t k	 ƒ t t | d d d d d	 g ƒ ƒ t  d ƒ } t | d d d d d d g ƒ} | d d	 !} t d d g d d d g ƒ} t | d t k	 ƒ t | d t k	 ƒ | | (t | d t k ƒ t | d t k	 ƒ t t | d d g ƒ ƒ t | d t k ƒ t | d t k	 ƒ t t | d d d d d	 g ƒ ƒ d  S(
   Ni   R\   i    i
   i(   i   i   i   i   (   R   R   R   R*   RZ   (   Ra   R`   Rb   Ri   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testPut2M  s,    !"!c         C` s    |  j  \ } } } } } } } } }	 }
 } d d d d d d d d d d d d g } t j | ƒ d } t | | |	 ƒ t t t | | d d ƒ|	 k ƒ ƒ d  S(   Ni   i    R‹   (   R`   R]   R;   R@   R   R   RL   (   Ra   Rb   Rc   Rd   Re   Rf   Rg   Rh   Ri   Rj   Rk   Rl   Rž   t   i(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testMaPutg  s
    **c   
      C` sd	  t  d ƒ } | j d d ƒ } d | j d <t | d  d k ƒ | d | } t t | j | ƒ ƒ t t | j d | ƒ ƒ t t | t | ƒ j d	 | | ƒ ƒ d
 | j d <t  d ƒ } t | d <t t	 | d ƒ t	 t ƒ k ƒ | d k } t t
 t | t t ƒ ƒ d k ƒ t t t | t t ƒ ƒ | j k ƒ t | | t ƒ } t | j | j k ƒ t | d t k ƒ t | d t k ƒ t | d t k ƒ t | d t k	 ƒ t | d t k	 ƒ t t | | ƒ ƒ t | t | ƒ } t | j | j k ƒ t | d t k ƒ t | d t k	 ƒ t | d t k	 ƒ t | d t k ƒ t | d t k ƒ t | | ƒ } t | j | j k ƒ t | d t k ƒ t | d t k	 ƒ t | d t k	 ƒ t | d t k ƒ t | d t k ƒ t t | | ƒ ƒ t d d d d d g ƒ } t d d d d d g ƒ } t | d <t | | | ƒ } t t | d d d
 d d g ƒ ƒ t | d <t | | | ƒ } t t | d d d
 d d g ƒ ƒ t | d t k ƒ t | d t k	 ƒ t | d t k ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t | d ƒ | ƒ t | d ƒ ƒ ƒ t t t t t d ƒ ƒ d d ƒ d d d d d g ƒ ƒ t t t t t d ƒ ƒ d d ƒ d d d d d g ƒ ƒ t t t t t t d ƒ ƒ d d d d d d g ƒd d ƒ j  d d d d d g ƒ ƒ t t t t t t d ƒ ƒ d d d d d d g ƒd d ƒ j  d d d d d g ƒ ƒ t t t t t t d ƒ ƒ d d d d d d g ƒd ƒ j  d d d d d g ƒ ƒ t t t t d d d d d g d d d d d d g ƒd ƒ j  d d d d d g ƒ ƒ t t t d d d d d g d d d d d g ƒ d d d d d g ƒ ƒ t! d! d t" j# ƒ} t$ | j t% ƒ } t | | ƒ } t t | | ƒ ƒ t& | | | f ƒ } t t | d d d
 d d g ƒ ƒ t | d t k ƒ t | d t k	 ƒ t | d t k ƒ t  d ƒ } t | d <t  d ƒ d } t | d <t d d d d d d g d d d d d d d g ƒ} | j' d ƒ } t | | | ƒ } t | | | ƒ }	 t t | |	 ƒ ƒ t t( |	 ƒ t) k ƒ t t |	 d d d d d d g ƒ ƒ t | t d ƒ } t t | d d d d d d g ƒ ƒ t | d t ƒ } t t | d d d d d d g ƒ ƒ d  S("   Ni   i   i   i   i   i    y              $@i
   ie   g        .i   i   i   i	   g      ð?g       @g      @g      @g      @i   g      ÀiûÿÿÿiÇ   R\   ic   Ro   i   i   i(   i2   (   i   i    (   i
   i
   i
   (*   R   Rv   t   flatR   RZ   t   realt   imagR   R*   RU   R   RP   RD   Ro   R6   R   R!   R-   R"   R.   R%   R0   R&   R1   R<   R2   R   R,   R/   t   listt   rangeR3   R\   R=   R]   t   float32RQ   R   R   R   R    R:   (
   Ra   Rb   Ri   t   ct   atestt   btestt   ctestRc   t   cmRj   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testOddFeaturesn  sÂ    (
 "%
"
"......::!!!0"

6%%c         C` s6  t  t t d d d g d d d g ƒ d d d g ƒ ƒ t  t t d d d g d d d g ƒ d d d g ƒ ƒ t d ƒ } t d ƒ d } t | d <t | d <t  t t | | ƒ t t | | ƒ | | ƒ ƒ ƒ t  t t | | ƒ t t | | ƒ | | ƒ ƒ ƒ t  t j	 | ƒ d k ƒ t  t j	 | ƒ d k ƒ d  S(   Ni   i   i   i   i    i	   i   (
   R   RZ   R8   R7   R   R*   RP   R%   R!   R   (   Ra   Rb   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testMinMax2Ó  s    77

11c         C` sœ  t  d ƒ } t j  d ƒ } t | d d +| j d d d ƒ } | j d d d ƒ } t t t j | d ƒ t | d ƒ ƒ ƒ t t t j | d d ƒ t | d d ƒ ƒ ƒ t t t j t	 | d ƒ t	 | d ƒ ƒ t | | ƒ ƒ ƒ t t t j
 t	 | d ƒ t	 | d ƒ ƒ t
 | | ƒ ƒ ƒ t d	 d d
 d d g t ƒ } t | d <t | d d d g ƒ } t | d d	 k ƒ t | d d k ƒ t | d d k ƒ d  S(   Ni   i   i   i   i   i   i    i   t   abct   def(   i   i    i   (   i   i    i   (   i   i    i   (   i   i    i   (   R   R]   R*   Rv   R   RZ   RO   RL   R#   R   R>   R   R‘   (   Ra   Rb   Rc   t   t(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt    test_testTakeTransposeInnerOuterà  s"    (.''
c         C` sg  t  d ƒ } t  d ƒ } t  d ƒ } t | d <| d 7} t t | | d ƒ ƒ | d 7} t t | | d ƒ ƒ t  d ƒ } t  d ƒ } t | d <| d 8} t t | | d ƒ ƒ | d 8} t t | | d ƒ ƒ t  d ƒ d } t  d ƒ d } t | d <| d 9} t t | | d ƒ ƒ | d 9} t t | | d ƒ ƒ t  d ƒ d } t  d ƒ } t | d <| d } t t | | ƒ ƒ | d } t t | | ƒ ƒ t  d ƒ d } t  d ƒ d } t | d <| d } t t | | d ƒ ƒ | t  d ƒ } t t | t d ƒ ƒ ƒ t  d ƒ j t j ƒ } t  d ƒ } t | d <| d 7} t t | | d ƒ ƒ d  S(   Ni
   i   i   g      ð?g       @(   i
   (   R   R*   R   RZ   R=   t   astypeR]   R³   (   Ra   Rc   Rb   Rg   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testInplaceô  sR    















c         C` s   t  d ƒ } t | d d d … <| j d d ƒ } xU t d t j d ƒ D]= } t j | d | ƒ} t j | ƒ } t t	 | | ƒ ƒ qH Wd  S(   Ni   i   i
   i   i   i   t   protocol(
   R   R*   Rv   R²   RR   t   HIGHEST_PROTOCOLt   dumpst   loadsR   RZ   (   Ra   Rb   t   protoRl   Rc   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testPickle&  s    c         C` s^   t  d ƒ } t | d <t t t ƒ d k ƒ t | d t k ƒ t t | d d ƒ d ƒ d  S(   Ni   i   s   --i    (   R   R*   R   RU   R   R   (   Ra   t   xx(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testMasked0  s
    
c      
   C` sÛ  t  d d d d g d d d d d g ƒ} t t d t | d d ƒƒ ƒ t t d t | d	 d d d d g ƒƒ ƒ t | d	 d d d d g d
 d ƒ\ } } t t d | ƒ ƒ t | d k ƒ t | (t t | d d ƒt k ƒ t  d d d d g d d d d d g ƒ} | j d d ƒ } t | d  d  … d f <t t t | d d ƒd d g ƒ ƒ t t | d d ƒd t k ƒ t t d d g t | d d ƒƒ ƒ t | d d d
 d ƒ\ } } t t | d d g ƒ ƒ d  S(   Ng        g      ð?g       @g      @R\   i   i    R‹   t   weightst   returnedg      @i   (   R   R   RZ   R   R*   Rv   (   Ra   R„   RY   t   wts(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testAverage18  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 ƒ } t t t | d d ƒd ƒ ƒ t t t | d d d | ƒd ƒ ƒ t t  d ƒ d t  d ƒ g ƒ } t t t | d  ƒ t j j t j  d ƒ ƒ d d	 ƒ ƒ t t t | d d ƒt j  d ƒ d d ƒ ƒ t t t | d d ƒt | d d ƒt | d d ƒd g ƒ ƒ t t t | d  d | ƒd ƒ ƒ t t t | d d d | ƒd d d d d d g ƒ ƒ t 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 t | | ƒ d d ƒd ƒ ƒ t t t t | | ƒ d d ƒd ƒ ƒ t t t | | ƒ d d ƒt k ƒ t t t | |	 ƒ d d ƒd ƒ t t t t | | ƒ d d ƒƒ d ƒ t | | ƒ }
 t t t |
 d  ƒ d ƒ ƒ t t t |
 d d ƒd d d d d d g ƒ ƒ t t t |
 d d ƒd d g ƒ ƒ t t t |
 d d d | ƒd d d d d d g ƒ ƒ 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 | ƒ k ƒ t d d g d d g g t ƒ } t | d d g d d g g ƒ } t | d d ƒ} t t | d d g ƒ ƒ t | d d ƒ} t t | d d g ƒ ƒ t | d d  ƒ} t t | d  ƒ ƒ t | d d ƒ} t t | d d g ƒ ƒ d  S(!   Ni    i   i   R‹   g      @RÉ   g       @g      @g      (@g      4@g      @g        g      ð?g      @g      $@g     ÀX@g      @g      @i   RÊ   i   i   g      à?g      @g      ø?g«ªªªªª
@g«ªªªªª
@(   i   i   i   (   i   i   i   (   i   i   i   (   i   i   i   g«ªªªªª@(   R   R   R   R   R   t   NoneR]   R
   R   RQ   R=   R+   R*   R   R   RD   t   floatRZ   (   Ra   t   w1t   w2Rb   Rc   Re   Rf   RŸ   t   m4t   m5Ri   t   at   bt   r1t   r2Ry   Rz   t   a2dat   a2dma(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testAverage2K  sl    0%"'0,",0((%"(%00*!!c         C` s×   t  d t t d ƒ ƒ ƒ t  d t t d ƒ ƒ ƒ t  d t t d g g g ƒ ƒ ƒ t  d t t d g g ƒ ƒ ƒ t t t t d d g ƒ ƒ t t t t d d g ƒ ƒ t t t t d d g d d d g ƒƒ d  S(   Ni   g      ð?i    R\   (   R   t   intR   RÎ   R   t	   TypeErrort
   ValueErrort   bool(   Ra   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testToPython…  s    "c         C` s  t  d d d ƒ} t j d d ƒ  t d t  d ƒ j ƒ Wd  QXt d | j ƒ t | j ƒ t | j ƒ t t | | ƒ j ƒ t t | | ƒ j ƒ t | j ƒ  j | j	 j k ƒ t  d d d ƒ} t | j ƒ  | j	 k ƒ t
 t | ƒ t t ƒ ƒ d  S(   Ni    R\   i   R   Rt   (   R   R]   Rw   R   R\   R7   R8   R   Ro   Rš   R   RU   R4   (   Ra   Rg   Rb   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testScalarArithmeticŽ  s    c      	   C` s  t  d d d g ƒ } t t | j ƒ  | j j ƒ  ƒ ƒ t t | j ƒ  | j j ƒ  ƒ ƒ t t | j ƒ  | j j ƒ  ƒ ƒ t t | j ƒ  | j j ƒ  ƒ ƒ t t | j d d d d d ƒ | j j d d d d d ƒ ƒ ƒ t t | j	 d d d g ƒ | j j	 d d d g ƒ ƒ ƒ t t | j
 ƒ  | j j
 ƒ  ƒ ƒ t t | j ƒ  | j j ƒ  ƒ ƒ t  d d g d d g g ƒ } t t | j ƒ  | j j ƒ  ƒ ƒ t t | j ƒ  | j j ƒ  ƒ ƒ t t | j d d g ƒ | j j d d g ƒ ƒ ƒ t t | j ƒ  | j j ƒ  ƒ ƒ d  S(   Ni   i   i   i    i   (   R   R   RZ   t   anyRš   R   t   argmaxt   argminR   t   compresst   conjR   t   diagonalRK   RL   RO   (   Ra   RÓ   Rž   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testArrayMethods  s    """"":""""4c         C` s)   t  d d d g ƒ } t | j d ƒ d  S(   Ni   i   i   (   R   R   t   ndim(   Ra   RÓ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testArrayAttributes®  s    c         C` sM   t  g  t t j ƒ D]. } | t t ƒ k r | j d ƒ r | ^ q ƒ d  S(   Nt   _(   R   t   dirR]   Rƒ   R   t
   startswith(   Ra   Rž   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testAPI²  s    c         C` sy   t  d d d g ƒ } t  d d d g d d d d g ƒ} t | d j d ƒ t | d j d ƒ t | d j d ƒ d  S(	   Ni   i   i   R\   i    (    (    (    (   R   R   RD   (   Ra   RÓ   RÔ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testSingleElementSubscript·  s
    $(   t   __name__t
   __module__Rm   Rr   Rs   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í   (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyR[   $   s8   				&				
		2	3				e			2	
			:						t
   TestUfuncsc           B` s5   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` su   t  d d d t d g d d d d g d g d ƒt  d d d t d g d d d d g d g d ƒf |  _ d  S(   Ng      ð?i    iÿÿÿÿi   R\   i   i   (   R   R^   R`   (   Ra   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyRm   À  s    4c      *   C` sÂ  d d d d d d d d d d	 d
 d d g } xŽd d d d d d d d d d d d d d d d d d d d d d d d d d d d	 d
 d d d  d! d" d# d$ d% d& d' d( d) d* g* D]} y t  t | ƒ } Wn  t k
 rí t  t | ƒ } n Xt  t j | ƒ } |  j | j  } t j ƒ  [ | | k r<t j	 d+ d, ƒ n  | d. k r[t j	 d d, ƒ n  | | Œ  } | | Œ  } Wd  QXt
 t | j d- ƒ | j d- ƒ | ƒ ƒ t
 t | j | j ƒ ƒ q² Wd  S(/   NRI   t   arctanhR   R   t   arccoshR'   R(   R   t   true_dividet   floor_dividet	   remaindert   fmodR   R   RE   R   RM   R   RF   R   RN   t   arcsinhR	   t   fabst   negativet   floort   ceilt   logical_notR
   RJ   R9   t   hypotR   R   R<   R&   R"   R%   R!   t   logical_andt
   logical_ort   logical_xorRu   Rt   i    (   Rñ   R'   R(   (   t   getattrt   umatht   AttributeErrort   fromnumericR]   R~   R`   t   ninRw   t   seterrR   RZ   R   t   eqmaskR\   (   Ra   t   f_invalid_ignoret   ft   uft   mft   argst   urt   mr(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_testUfuncRegressionÄ  sD    						(c         C` st   |  j  d } t t | d d ƒƒ t t | d d ƒƒ t t | d  d d ƒd ƒ t t | d d ƒd ƒ d  S(   Ni    R‹   i   (   R`   R   R   RG   R   RK   R?   (   Ra   RÓ   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_reduceé  s
    c         C` sç   t  d d ƒ j d d ƒ } t | d k  | ƒ } t | j ƒ  | j ƒ  ƒ t | j ƒ  d ƒ t | j d ƒ | j d ƒ k j ƒ  ƒ t | j d ƒ d d d d	 g k j ƒ  ƒ t | j d ƒ d j ƒ t | j d ƒ d j ƒ d  S(
   Ni   i   i   i   i   i    i   i   i   (	   R   Rv   R6   R   R†   R‡   R   R   R\   (   Ra   RÓ   t   amask(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_minmaxð  s    (+c         C` s[   xT d D]L } t  d d d d g d d d d d g ƒ} t t t | ƒ d g ƒ ƒ q Wd  S(   Ns   ?bhilqpBHILQPfdgFDGOi   i    i   R\   (   R   R   RZ   R;   (   Ra   R½   Rb   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_nonzeroú  s    *(   Rî   Rï   Rm   R  R  R  R  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyRð   ¿  s
   		%		
t   TestArrayMethodsc           B` sP   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z	 RS(   c      %   C` s™  t  j d d d d d d d d d	 d
 d d d d d d d d d d d d d d d d d d d d d d  d! d" d# d$ g$ ƒ } | j d% d% ƒ } | j d& d' d' d& ƒ } t  j d( d) d( d) d( d( d) d( d) d) d( d) d( d( d( d) d( d) d( d( d( d) d) d) d) d( d( d) d( d( d( d( d) d( d) d( g$ ƒ } t d* | d+ | ƒ } t d* | d+ | j | j ƒ ƒ } t d* | d+ | j | j ƒ ƒ } | | | | | | | f |  _ d  S(,   Ng     À @g®Gáz.@g-²ï§!@g      !@gé&1¬ü?gƒÀÊ¡E¶@g\Âõ(Ü @g…ëQ¸@g{®Gáº#@gP—nƒ@gh‘í|?õ!@g!°rh‘í@gj¼t“@gJ+‡@gáz®Ga@g!°rh‘í@gÁÊ¡E¶ó@g-²ï§@g)\Âõ(@gÃõ(\B#@g+‡Ù@gu“V@gÛù~j¼ô@gÑ"Ûù~ê@g¨ÆK7‰Á@g
×£p=J#@g®Gáz”@gòÒMbØ@g}?5^ºÉ#@g¾Ÿ/Ý$@g!°rh‘m@g-²ï§#@gJ+‡@gªñÒMb@gƒÀÊ¡Å@g-²ï§Æï?i   i   i   i    i   t   dataR\   (   R]   R   Rv   RD   R`   (   Ra   Rb   t   Xt   XXRž   t   mxt   mXt   mXX(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyRm     s$    !!c   	      C` s   |  j  \ } } } } } } } | j ƒ  } t | j ƒ  | j ƒ  j ƒ  j ƒ  ƒ t t | j ƒ  | j ƒ  t | j | j ƒ  d d ƒƒ ƒ d  S(   NR‹   i    (	   R`   Rå   R   t   tracet
   compressedRK   R   RZ   R\   (	   Ra   Rb   R  R  Rž   R  R  R  t   mXdiag(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt
   test_trace  s    %c   	      C` s”   |  j  \ } } } } } } } | j d d ƒ } t t | j | j ƒ ƒ t t | j | j d d ƒ ƒ ƒ t t | j | j j d d ƒ ƒ ƒ d  S(   Ni   i   (   R`   t   clipR   RZ   R\   Rš   (	   Ra   Rb   R  R  Rž   R  R  R  t   clipped(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt	   test_clip   s
    "c         C` s   |  j  \ } } } } } } } | j \ } } t | j ƒ  | j ƒ  j ƒ  ƒ t j | t j ƒ }	 t j | t j ƒ }
 x: t | ƒ D], } | d  d  … | f j ƒ  j ƒ  |
 | <qƒ Wx. t | ƒ D]  } | | j ƒ  j ƒ  |	 | <qÀ Wt	 t
 | j d ƒ |
 ƒ ƒ t	 t
 | j d ƒ |	 ƒ ƒ d  S(   Ni    i   (   R`   RD   R   t   ptpR  R]   RQ   t   float_R²   R   RZ   (   Ra   Rb   R  R  Rž   R  R  R  R   t   rowst   colst   k(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_ptp'  s    *c   
      C` s}   |  j  \ } } } } } } } | j d d ƒ } t t | d | d  d  … d f ƒ ƒ | j d d ƒ }	 t |	 j d ƒ d  S(   Ni    i   iÿÿÿÿi   i   (   i   i   i   i   (   R`   t   swapaxesR   RZ   R   RD   (
   Ra   Rb   R  R  Rž   R  R  R  t	   mXswappedt
   mXXswapped(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_swapaxes4  s
    'c   	      C` s   |  j  \ } } } } } } } | j d ƒ } t t | j | j d ƒ j d ƒ ƒ ƒ | j d ƒ } t t | j | j d ƒ j d ƒ ƒ ƒ d  S(   Ni    i   (   R`   t   cumprodR   RZ   Rš   R   (	   Ra   Rb   R  R  Rž   R  R  R  t   mXcp(    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_cumprod;  s
    (c   	      C` s   |  j  \ } } } } } } } | j d ƒ } t t | j | j d ƒ j d ƒ ƒ ƒ | j d ƒ } t t | j | j d ƒ j d ƒ ƒ ƒ d  S(   Ni    i   (   R`   t   cumsumR   RZ   Rš   R   (	   Ra   Rb   R  R  Rž   R  R  R  R-  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_cumsumB  s
    (c         C` s   |  j  \ } } } } } } } t t | j d d  ƒ | j ƒ  j ƒ  ƒ ƒ t t | j d d  ƒ | j ƒ  j ƒ  ƒ ƒ t t | j d d ƒ j | j d d ƒ j ƒ ƒ t t | j ƒ  j | j ƒ  j ƒ ƒ | j d d ƒ | j d d ƒ } }	 xª t d ƒ D]œ }
 t t |	 |
 | |
 j ƒ  j ƒ  ƒ ƒ t t | |
 | d  d  … |
 f j ƒ  j ƒ  ƒ ƒ t t t	 j
 | |
 ƒ | d  d  … |
 f j ƒ  j ƒ  ƒ ƒ qü Wd  S(   NR‹   i   i    i   i   (   R`   R   RZ   t   varRÍ   R  t   stdRD   R²   R]   RI   (   Ra   Rb   R  R  Rž   R  R  R  t   mXvar0t   mXvar1R&  (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   test_varstdI  s    ++1%%'3(
   Rî   Rï   Rm   R  R!  R'  R+  R.  R0  R5  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyR     s   							c         C` s<   |  t  k r | t  k S| t  k r, |  t  k S|  | k j ƒ  S(   N(   R:   R   (   Re   Rf   (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyR  W  s
    

(f   t
   __future__R    R   R   t	   functoolsR   t   numpyR]   t   numpy.core.umatht   coreR  t   numpy.core.fromnumericR  t   numpy.testingR   R   R   t   numpy.maR   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"   R#   R$   R%   R&   R'   R(   R)   R*   R+   R,   R-   R.   R/   R0   R1   R2   R3   R4   R5   R6   R7   R8   R9   R:   R;   R<   R=   R>   R?   R@   RA   RB   RC   RD   RE   RF   RG   RH   RI   RJ   RK   RL   RM   RN   RO   RP   RQ   t   numpy.core.numericRR   R^   RZ   R‘   R[   Rð   R  R  (    (    (    s9   lib/python2.7/site-packages/numpy/ma/tests/test_old_ma.pyt   <module>   s    ÿ Í	ÿ ÿ AW