o
    Z۷i                     @   sb   d dl Z d dlmZ d dlmZ ddiZddiZdd ZG d	d
 d
e jZ	e
dkr/e   dS dS )    N)jit)typesforceobjTnopythonc                 C   s   | | S N )abr   r   Q/home/ubuntu/vllm_env/lib/python3.10/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fi ||}|| }|| }| t|t| | || d S r   )r   r   assertIstypeassertEqual)selfflagsint_typeoperandspyfunccfuncexpectedgotr   r   r
   test_int_return_type   s   z"TestPythonInt.test_int_return_typec                 C      | j td d S N)r   )r   no_pyobj_flagsr   r   r   r
   test_int_return_type_npm      z&TestPythonInt.test_int_return_type_npmc                 C   s   | j tj|d d S )N)r   r   )r   r   uint64r   r   r   r   r
   test_unsigned_int_return_type   s   z+TestPythonInt.test_unsigned_int_return_typec                 C   r   r   )r#   r   r   r   r   r
   !test_unsigned_int_return_type_npm"   r    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   r   r   )r'   r   r   r   r   r
   test_long_int_return_type_npm*   r    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_type-   s   z)TestPythonInt.test_longer_int_return_typeN)__name__
__module____qualname__force_pyobj_flagsr   int64r   r   r#   r$   r'   r(   r,   r   r   r   r
   r      s    
	r   __main__)unittestnumbar   
numba.corer   r0   r   r   TestCaser   r-   mainr   r   r   r
   <module>   s    $