ó
\K]c           @   sđ   d  d l  Z d  d l m Z d  d l m Z d  d l m Z d  d l m	 Z	 m
 Z
 m Z d  d l m Z m Z e d  e	 d e e j f d	     Y  Z e d
 k rĩ e j   n  d S(   iĸĸĸĸN(   t   cuda(   t	   transpose(   t   unittest(   t   ddtt   datat   unpack(   t   skip_on_cudasimt   SerialMixins-   Device Array API unsupported in the simulatort   Testc           B   sM   e  Z e d  d e j f d d e j f d d e j f  e d     Z RS(   i   i   i   i  i   c         C   sŪ   t  j | | d | j | |  } t  j | | d | j | |  } t j |  } t j j j |  } t	 | |  | j
 |  |  j t  j | j	   | k   d  S(   Nt   dtype(   t   npt   aranget   reshapet   zerosR    t	   to_devicet   cudadrvt   devicearrayt   from_array_likeR   t   copy_to_hostt
   assertTruet   all(   t   selft   rowst   colsR	   t   xt   yt   dxt   dy(    (    sE   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_transpose.pyt   test_transpose   s    %%(   t   __name__t
   __module__R   R
   t   float64t
   complex128R   R   (    (    (    sE   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_transpose.pyR   	   s   t   __main__(   t   numpyR
   t   numbaR    t   numba.cuda.kernels.transposeR   t   numba.cuda.testingR   t   numba.testing.ddtR   R   R   R   R   t   TestCaseR   R   t   main(    (    (    sE   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_transpose.pyt   <module>   s   	