U
    iâËdü  ã                   @   sd   d dl Z d dlmZmZ d dlmZ eƒ Zde_eƒ Zdd„ Z	G dd„ de j
ƒZed	kr`e  ¡  dS )
é    N)Úcompile_isolatedÚFlags)ÚtypesTc                 C   s   | | S )N© )ÚaÚbr   r   úT/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/tests/test_python_int.pyÚ
return_int   s    r	   c                   @   sZ   e Zd Zeejdfdd„Zdd„ Zefdd„Zdd	„ Z	efd
d„Z
dd„ Zefdd„ZdS )ÚTestPythonInt)é   é   c           	      C   sP   t }t|||f|d�}|j}||Ž }||Ž }|  t|ƒt|ƒ¡ |  ||¡ d S ©N)Úflags)r	   r   Úentry_pointZassertIsÚtypeÚassertEqual)	Úselfr   Úint_typeÚoperandsZpyfuncÚcrZcfuncÚexpectedÚgotr   r   r   Útest_int_return_type   s    z"TestPythonInt.test_int_return_typec                 C   s   | j td� d S r   )r   Úno_pyobj_flags©r   r   r   r   Útest_int_return_type_npm   s    z&TestPythonInt.test_int_return_type_npmc                 C   s   | j tj|d� d S )N)r   r   )r   r   Zuint64©r   r   r   r   r   Útest_unsigned_int_return_type"   s    z+TestPythonInt.test_unsigned_int_return_typec                 C   s   | j td� d S r   )r   r   r   r   r   r   Ú!test_unsigned_int_return_type_npm%   s    z/TestPythonInt.test_unsigned_int_return_type_npmc                 C   s   | j |dd� d S )N)l        l        ©r   r   ©r   r   r   r   r   Útest_long_int_return_type(   s    z'TestPythonInt.test_long_int_return_typec                 C   s   | j td� d S r   )r!   r   r   r   r   r   Útest_long_int_return_type_npm-   s    z+TestPythonInt.test_long_int_return_type_npmc                 C   s   | j |dd dd fd� d S )Né   éF   éK   r   r    r   r   r   r   Útest_longer_int_return_type0   s    z)TestPythonInt.test_longer_int_return_typeN)Ú__name__Ú
__module__Ú__qualname__Úforce_pyobj_flagsr   Zint64r   r   r   r   r!   r"   r&   r   r   r   r   r
      s    ÿ

r
   Ú__main__)ZunittestZnumba.core.compilerr   r   Z
numba.corer   r*   Zforce_pyobjectr   r	   ZTestCaser
   r'   Úmainr   r   r   r   Ú<module>   s   %