ó
\K]c           @   sĶ   d  d l  j Z d  d l Z d  d l m Z m Z m Z d  d l	 m
 Z
 d  d l m Z d e j f d     YZ d e j f d     YZ e d	 k rĒ e j   n  d S(
   iĸĸĸĸN(   t   configt   jitt   types(   t   compile_isolated(   t   override_configt   TestJITMethodc           B   s   e  Z d    Z d   Z RS(   c         C   sŠ   d t  f d     Y} | d  } t j j | j d  t j d d d d d g d t j  | j j j   \ } | j	 d } | j j   \ } |  j
 | j j  d  S(   Nt	   Somethingc           B   s    e  Z d    Z e d    Z RS(   c         S   s   | |  _  d  S(   N(   t   x0(   t   selfR   (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt   __init__   s    c         S   sT   t  j d d d t  j  } |  j } x) t | j d  D] } | | | | <q8 W| S(   Nt   shapei   t   dtypei    (   t   npt   emptyt   float32R   t   rangeR
   (   R   t   xt   aR   t   i(    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt   method   s
    	(   t   __name__t
   __module__R	   R   R   (    (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR      s   	i   i   i   R   i    (   t   objectR   t   testingt   assert_array_equalR   t   arrayR   t	   overloadst   valuest   liftedt
   assertTruet   fndesct   native(   R   R   t	   somethingt   crest   jitloopt   loopcres(    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt$   test_bound_jit_method_with_loop_lift   s    (c         C   s?   d t  f d     Y} | d  } |  j | j |  d  d  S(   NR   c           B   s    e  Z d    Z e d    Z RS(   c         S   s   | |  _  d  S(   N(   R   (   R   R   (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR	   &   s    c         S   s   |  j  S(   N(   R   (   R   (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR   )   s    (   R   R   R	   R   R   (    (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR   %   s   	i   (   R   t   assertEqualsR   (   R   R   R    (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt   test_unbound_jit_method$   s    (   R   R   R$   R&   (    (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR   
   s   	t   TestDisabledJITc           B   s   e  Z d    Z d   Z RS(   c         C   sk   t  d t   d   } t |  } Wd  QX|  j | |  |  j d | d   |  j d | d   d  S(   Nt   DISABLE_JITc         S   s   |  S(   N(    (   R   (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR   4   s    i
   (   R   t   TrueR   t   assertEqual(   R   R   t   jitted(    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt   test_decorated_function2   s    	c         C   st   t  d t  # d   } t d t  |  } Wd  QX|  j | |  |  j d | d   |  j d | d   d  S(   NR(   c         S   s   |  S(   N(    (   R   (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR   >   s    t   nopythoni
   (   R   R)   R   R*   (   R   R   R+   (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt#   test_decorated_function_with_kwargs<   s    	(   R   R   R,   R.   (    (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyR'   1   s   	
t   __main__(   t   numba.unittest_supportt   unittest_supportt   unittestt   numpyR   t   numbaR    R   R   t   numba.compilerR   t   numba.tests.supportR   t   TestCaseR   R'   R   t   main(    (    (    s9   lib/python2.7/site-packages/numba/tests/test_jitmethod.pyt   <module>   s   '