o
    8ήc$                     @   s\   d dl Zd dlmZ d dlmZmZ d dlZedG dd deZe	dkr,e
  dS dS )    N)	vectorize)skip_on_cudasimCUDATestCasez&ufunc API unsupported in the simulatorc                   @   s   e Zd Zdd ZdS )TestVectorizeComplexc                 C   sL   t dgdddd }tjdtjd}||}| t|| d | d S )	Nzcomplex128(complex128)cuda)targetc                 S   s   | |  d S )N      ? )ar	   r	   U/tmp/pip-target-vg8gfxp4/lib/python/numba/cuda/tests/cudapy/test_vectorize_complex.pyvcomp
   s   z:TestVectorizeComplex.test_vectorize_complex.<locals>.vcomp   )dtyper   )r   nparange
complex128
assertTrueallclose)selfr   ABr	   r	   r   test_vectorize_complex	   s
   
z+TestVectorizeComplex.test_vectorize_complexN)__name__
__module____qualname__r   r	   r	   r	   r   r      s    r   __main__)numpyr   numbar   numba.cuda.testingr   r   unittestr   r   mainr	   r	   r	   r   <module>   s    