ó
\K]c           @` sª   d  d l  m Z m Z m Z d  d l m Z d  d l Z d  d l m	 Z	 d  d l
 m Z d d l m Z d „  Z d	 e e j f d
 „  ƒ  YZ e d k r¦ e j ƒ  n  d S(   i    (   t   print_functiont   absolute_importt   division(   t   unittest_supportN(   t   njit(   t   dufunci   (   t   MemoryLeakMixinc         C` s   |  | S(   N(    (   t   a0t   a1(    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   pyuadd   s    t
   TestDUFuncc           B` s>   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c         C` s   t  j | d t d t ƒ ƒS(   Nt   targetoptionst   nopython(   R   t   DUFunct   dictt   True(   t   selft   pyfunc(    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   nopython_dufunc   s    c         C` sœ   |  j  t ƒ } |  j | j ƒ t | _ |  j | j ƒ |  j t ƒ  t | _ Wd  QX|  j t	 ƒ 0 | t
 j d d d ƒ t
 j d d d ƒ ƒ Wd  QXd  S(   Ni    i   i
   i   (   R   R	   t   assertFalset   _frozenR   t
   assertTruet   assertRaisest
   ValueErrort   Falset	   TypeErrort   npt   linspace(   R   t   duadd(    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   test_frozen   s    	c         C` s5   |  j  t ƒ } |  j t d d ƒ | d d ƒ ƒ d  S(   Ni   i   (   R   R	   t   assertEqual(   R   R   (    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   test_scalar   s    c         ` s  |  j  t ƒ ‰  t ‡  f d †  ƒ } t j d d d ƒ } | d  } | d } t j d ƒ } | | | | ƒ t j j | | | ƒ | j d ƒ } | j d	 ƒ } t j d
 ƒ } | | | | ƒ t j j | | | ƒ | d  }	 t j d ƒ }
 | | |	 |
 ƒ t j j | |	 |
 ƒ d  S(   Nc         ` s   ˆ  |  | | ƒ d  S(   N(    (   R   R   t   o0(   R   (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   npmadd$   s    i    gffffffþ?i   i
   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (	   R   R	   R   R   R   t   zerost   testingt   assert_array_equalt   reshape(   R   R!   t   Xt   X0t   X1t   out0t   Y0t   Y1t   out1t   Y2t   out2(    (   R   s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   test_npm_call"   s"    


c         ` s
  |  j  t ƒ ‰  t ‡  f d †  ƒ } t j d d d ƒ } | d  } | d } | | | ƒ } t j j | | | ƒ | j d ƒ } | j d ƒ } | | | ƒ } t j j | | | ƒ | d  }	 | | |	 ƒ }
 t j j | |	 |
 ƒ | d d	 ƒ } |  j | d
 ƒ d  S(   Nc         ` s   ˆ  |  | ƒ S(   N(    (   R   R   (   R   (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyR!   9   s    i    gffffffþ?i   i
   i   i   g      ð?g       @g      @(   i   i   (   i   i   (	   R   R	   R   R   R   R#   R$   R%   R   (   R   R!   R&   R'   R(   R)   R*   R+   R,   R-   R.   t   out3(    (   R   s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   test_npm_call_implicit_output7   s     


c         C` sË   |  j  t ƒ } |  j | j d ƒ |  j | j d ƒ |  j | j | j | j ƒ |  j | j d ƒ |  j | j g  ƒ |  j | j d  ƒ | d d ƒ |  j | j d ƒ |  j | j t
 | j ƒ ƒ d  S(   Ni   i   i    (   R   R	   R   t   nint   noutt   nargst   ntypest   typest   identityt   Nonet   len(   R   R   (    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   test_ufunc_propsK   s    (   t   __name__t
   __module__R   R   R   R/   R1   R:   (    (    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyR
      s   		
			t   __main__(   t
   __future__R    R   R   t   numbaR   t   unittestt   numpyR   R   t   numba.npyufuncR   t   supportR   R	   t   TestCaseR
   R;   t   main(    (    (    s?   lib/python2.7/site-packages/numba/tests/npyufunc/test_dufunc.pyt   <module>   s   	H