B
      ›\g  ã               @   s†   d dl mZ d dlmZ d dlmZmZ d dlm	Z	m
Z
 eƒ Ze d¡ eƒ Ze d¡ dd„ ZG d	d
„ d
ejƒZedkr‚e ¡  dS )é    )Úprint_functionN)Úcompile_isolatedÚFlags)ÚtypesÚutilsZenable_pyobjectZforce_pyobjectc             C   sJ   d}d}x<t dƒD ]0}|d |  }|j|j |j|j  dkrdS qW dS )Nr   y                éd   é   é   FT)ÚrangeÚrealÚimag)ÚcÚiÚz© r   ú:lib/python3.7/site-packages/numba/tests/test_mandelbrot.pyÚis_in_mandelbrot   s    r   c               @   s   e Zd Zdd„ ZdS )ÚTestMandelbrotc             C   sL   t }t|tjfƒ}|j}dddddg}x |D ]}|  ||ƒ||ƒ¡ q,W d S )Ny                y      ð?        y              ð?y      ð?      ð?yš™™™™™¹?š™™™™™¹?)r   r   r   Z	complex64Zentry_pointZ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
__future__r   Znumba.unittest_supportZunittest_supportZunittestZnumba.compilerr   r   Znumbar   r   Zenable_pyobj_flagsÚsetZforce_pyobj_flagsr   ZTestCaser   r   Úmainr   r   r   r   Ú<module>   s   


