o
    8ήc                     @   sB   d dl mZ d dlmZ G dd dejZedkre  dS dS )    )cuda)unittestc                   @   s   e Zd Zdd ZdS )TestCudaMandelc                 C   s   t jddddd }dS )z+Just make sure we can compile this
        zD(uint32, float64, float64, float64, float64, uint32, uint32, uint32)T)devicec                 S   s   || | }|| | }	| | }
| | }||
|  }|||	  }t ||}d}t|D ]}|| | }|j|j |j|j  dkrE|  S q+|S )Ny                   )complexrangerealimag)tidmin_xmax_xmin_ymax_ywidthheightiterspixel_size_xpixel_size_yxyr	   r
   czi r   J/tmp/pip-target-vg8gfxp4/lib/python/numba/cuda/tests/cudapy/test_mandel.pymandel
   s   
z*TestCudaMandel.test_mandel.<locals>.mandelN)r   jit)selfr   r   r   r   test_mandel   s   zTestCudaMandel.test_mandelN)__name__
__module____qualname__r   r   r   r   r   r      s    r   __main__N)numbar   numba.cuda.testingr   TestCaser   r    mainr   r   r   r   <module>   s    