U
    ,dI                     @   s>   d Z ddlmZ ddlZG dd dejZedkr:e  dS )z&
Test setting/overriding error models
    )jitNc                   @   s   e Zd Zdd Zdd ZdS )TestErrorModelc              	   C   s.   t dd }| t |d W 5 Q R X d S )Nc                 S   s   d|  S N    valr   r   @/tmp/pip-unpacked-wheel-eu7e0c37/numba/tests/test_errormodels.pymodel_python   s    z<TestErrorModel.test_div_by_zero_python.<locals>.model_pythonr   )r   assertRaisesZeroDivisionError)selfr
   r   r   r	   test_div_by_zero_python   s    
z&TestErrorModel.test_div_by_zero_pythonc                 C   s*   t dddd }| |dtd d S )NZnumpy)Zerror_modelc                 S   s   d|  S r   r   r   r   r   r	   model_numpy   s    z:TestErrorModel.test_div_by_zero_numpy.<locals>.model_numpyr   inf)r   assertEqualfloat)r   r   r   r   r	   test_div_by_zero_numpy   s    
z%TestErrorModel.test_div_by_zero_numpyN)__name__
__module____qualname__r   r   r   r   r   r	   r   	   s   r   __main__)__doc__Znumbar   ZunittestZTestCaser   r   mainr   r   r   r	   <module>   s
   