
\K]c           @   sl   d  Z  d d l m Z d d l m Z d d l m Z d e j f d     YZ e	 d k rh e j
   n  d S(	   s&   
Test setting/overriding error models
i(   t   division(   t   jit(   t   unittest_supportt   TestErrorModelc           B   s   e  Z d    Z d   Z RS(   c         C   s3   t  d    } |  j t   | d  Wd  QXd  S(   Nc         S   s   d |  S(   Ni   (    (   t   val(    (    s;   lib/python2.7/site-packages/numba/tests/test_errormodels.pyt   model_python   s    i    (   R   t   assertRaisest   ZeroDivisionError(   t   selfR   (    (    s;   lib/python2.7/site-packages/numba/tests/test_errormodels.pyt   test_div_by_zero_python   s    c         C   s8   t  d d  d    } |  j | d  t d   d  S(   Nt   error_modelt   numpyc         S   s   d |  S(   Ni   (    (   R   (    (    s;   lib/python2.7/site-packages/numba/tests/test_errormodels.pyt   model_numpy   s    i    t   inf(   R   t   assertEqualt   float(   R   R   (    (    s;   lib/python2.7/site-packages/numba/tests/test_errormodels.pyt   test_div_by_zero_numpy   s    (   t   __name__t
   __module__R	   R   (    (    (    s;   lib/python2.7/site-packages/numba/tests/test_errormodels.pyR   
   s   	t   __main__N(   t   __doc__t
   __future__R    t   numbaR   R   t   unittestt   TestCaseR   R   t   main(    (    (    s;   lib/python2.7/site-packages/numba/tests/test_errormodels.pyt   <module>   s   