U
    ,‰d0  ã                   @   sn   d dl Z d dlmZmZ d dlmZmZ eƒ Zde_eƒ Z	de	_
dd„ ZG dd„ de jƒZed	krje  ¡  dS )
é    N)Úcompile_isolatedÚFlags)ÚtypesÚutilsTc                 C   sH   d}d}t dƒD ]2}|d |  }|j|j |j|j  dkr dS qdS )Nr   ù                éd   é   é   FT)ÚrangeÚrealÚimag)ÚcÚiÚz© r   ú?/tmp/pip-unpacked-wheel-eu7e0c37/numba/tests/test_mandelbrot.pyÚis_in_mandelbrot   s    r   c                   @   s   e Zd Zdd„ ZdS )ÚTestMandelbrotc                 C   sH   t }t|tjfƒ}|j}dddddg}|D ]}|  ||ƒ||ƒ¡ q*d S )Nr   y      ð?        y              ð?y      ð?      ð?yš™™™™™¹?š™™™™™¹?)r   r   r   Z	complex64Zentry_pointÚassertEqual)ÚselfZpyfuncZcrZcfuncZpointsÚpr   r   r   Útest_mandelbrot   s    zTestMandelbrot.test_mandelbrotN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s   r   Ú__main__)ZunittestZnumba.core.compilerr   r   Z
numba.corer   r   Zenable_pyobj_flagsZenable_pyobjectZforce_pyobj_flagsZforce_pyobjectr   ZTestCaser   r   Úmainr   r   r   r   Ú<module>   s   
