U
    ?h                     @   sH   d dl Zd dlmZ d dlmZmZ G dd deZedkrDe	  dS )    N)cuda)unittestCUDATestCasec                   @   s   e Zd Zdd ZdS )TestCudaComplexc                 C   sX   t ddd }tjdtjd}| }|d|jf |d | t||d  d S )Nzvoid(complex128[:], complex128)c                 S   s   t d}| |  |7  < d S )N   )r   grid)abi r   ]/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_complex_kernel.pyfoo   s    
z2TestCudaComplex.test_cuda_complex_arg.<locals>.foo   )Zdtyper   y               @)	r   ZjitnpZarangeZ
complex128copyshapeZ
assertTrueZallclose)selfr   r   Za0r   r   r   test_cuda_complex_arg   s    
z%TestCudaComplex.test_cuda_complex_argN)__name__
__module____qualname__r   r   r   r   r   r      s   r   __main__)
numpyr   Znumbar   Znumba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s
   