
    sh                         S SK r S SKJrJr  S SKJr  \" S5       " S S\5      5       r\S:X  a  \ R                  " 5         gg)    N)CUDATestCaseskip_on_cudasim)captured_stdoutz4cudasim doesn't support cuda import at non-top-levelc                   >   ^  \ rS rSrSrU 4S jrU 4S jrS rSrU =r	$ )	TestUFunc   z
Test calling a UFunc
c                 t   > [        5       U l        U R                  R                  5         [        TU ]  5         g N)r   _captured_stdout	__enter__supersetUpself	__class__s    |/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/doc_examples/test_ufunc.pyr   TestUFunc.setUp   s+     !0 1'')    c                 \   > U R                   R                  S S S 5        [        TU ]  5         g r
   )r   __exit__r   tearDownr   s    r   r   TestUFunc.tearDown   s&    &&tT48r   c                   ^ SS K mSSKJn  UR                  U4S j5       nTR	                  STR
                  S9S-
  nTR                  U5      nUS   " XC5        TR                  R                  UTR                  U5      5        g )Nr   )cudac                 (   > TR                  X5        g r
   )sin)rxnps     r   f,TestUFunc.test_ex_cuda_ufunc_call.<locals>.f   s     FF1Lr   
   )dtype   )   r%   )
numpynumbar   jitarangefloat32
zeros_liketestingassert_allcloser   )r   r   r    r   r   r   s        @r   test_ex_cuda_ufunc_call!TestUFunc.test_ex_cuda_ufunc_call   sw     
	 
	
 IIb

I+a/MM! 	
$ 	

""1bffQi0r   )r   )
__name__
__module____qualname____firstlineno____doc__r   r   r.   __static_attributes____classcell__)r   s   @r   r   r      s    
1 1r   r   __main__)	unittestnumba.cuda.testingr   r   numba.tests.supportr   r   r0   main r   r   <module>r=      sI     < / GH%1 %1 I%1R zMMO r   