ķ
\K]c           @@ sw   d  d l  m Z m Z d  d l Z d  d l m Z d  d l j Z	 d e	 j
 f d     YZ e d k rs e	 j   n  d S(   i    (   t   print_functiont   absolute_importN(   t   roct   TestPositioningc           B@ s   e  Z d    Z RS(   c         C@ s  t  j d    } t j d d t j } | d |  t j j d d d g d d d	 g g | d  t j j d
 | d  t j j d d d g g d | d  t j j d | d  t j j d d d g d d d g g | d  t j j d | d	  t j j d | d
  d  S(   Nc         S@ sī   t  j d  } t  j d  } t  j d  } t  j d  } t  j d  } t  j   } t  j d  } | |  d | | f <| |  d | | f <| |  d | | f <| |  d | | f <| |  d | | f <| |  d | | f <| |  d | | f <d  S(   Ni    i   i   i   i   i   i   (   R   t   get_global_idt   get_global_sizet   get_local_idt   get_group_idt   get_num_groupst   get_work_dimt   get_local_size(   t   outputt	   global_idt   global_sizet   local_idt   group_idt
   num_groupst   workdimt
   local_size(    (    sE   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_positioning.pyt   udt   s    i   i   i   t   dtypei    i   i   i   i   (   i   i   i   (   i   i   (   R   t   jitt   npt   zerost   intpt   testingt   assert_equal(   t   selfR   t   out(    (    sE   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_positioning.pyt   test_kernel_jit
   s    /'/(   t   __name__t
   __module__R   (    (    (    sE   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_positioning.pyR      s   t   __main__(   t
   __future__R    R   t   numpyR   t   numbaR   t   numba.unittest_supportt   unittest_supportt   unittestt   TestCaseR   R   t   main(    (    (    sE   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_positioning.pyt   <module>   s   !