ó
\K]c           @  sŻ   d  d l  m Z d  d l Z d  d l m Z d  d l m Z d  d l m	 Z	 m
 Z
 d   Z d   Z d   Z d	   Z d
 e
 e j f d     YZ e d k rĞ e j   n  d S(   i˙˙˙˙(   t   print_functionN(   t   cuda(   t   unittest_support(   t   captured_cuda_stdoutt   SerialMixinc          C  s*   t  j d  }  t |  d  t d  d  S(   Ni   iç  iÖ˙˙˙(   R   t   gridt   print(   t   i(    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   cuhello
   s    c          C  s&   t  j d  }  t |  d d d  d  S(   Ni   i   g     `A@iA  (   R   R   R   (   R   (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt
   printfloat   s    c          C  s#   t  j d  }  t |  d d  d  S(   Ni   s   hop!iç  (   R   R   R   (   R   (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   printstring   s    c           C  s   t    d  S(   N(   R   (    (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt
   printempty   s    t	   TestPrintc           B  s,   e  Z d    Z d   Z d   Z d   Z RS(   c         C  s   t  j d d t t  } t    } | d   Wd  QX| j   } t | j t   } d g d g  t	 d  D] } d | ^ qp } |  j
 | |  d  S(	   Ns   void()t   debugi   i   s   -42
i   s   %d 999
(   i   i   (   R   t   jitt   FalseR   R   t   getvaluet   sortedt
   splitlinest   Truet   ranget   assertEqual(   t   selft   jcuhellot   stdoutt   outt   linesR   t   expected(    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   test_cuhello    s    .c         C  sT   t  j d d t t  } t    } |   Wd  QX|  j | j   d d g  d  S(   Ns   void()R   s   0 23 34.750000 321
s   0 23 34.75 321
(   R   R   R   R	   R   t   assertInR   (   R   t   jprintfloatR   (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   test_printfloat+   s
    c         C  sN   t  j d d t t  } t    } |   Wd  QX|  j | j   d  d  S(   Ns   void()R   s   
(   R   R   R   R   R   R   R   (   R   t   cufuncR   (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   test_printempty3   s    c         C  s   t  j d d t t  } t    } | d   Wd  QX| j   } t | j t   } g  t	 d  D] } d | ^ qf } |  j
 | |  d  S(   Ns   void()R   i   i   s   %d hop! 999
(   i   i   (   R   R   R   R
   R   R   R   R   R   R   R   (   R   R    R   R   R   R   R   (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   test_string9   s    #(   t   __name__t
   __module__R   R   R!   R"   (    (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyR      s   			t   __main__(   t
   __future__R    t   numpyt   npt   numbaR   R   t   unittestt   numba.cuda.testingR   R   R   R	   R
   R   t   TestCaseR   R#   t   main(    (    (    sA   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_print.pyt   <module>   s   				%