
ʽXc           @` s   d  d l  m Z m Z m Z d  d l Z d  d l m Z m Z m	 Z	 m
 Z
 d  d l m Z m Z d e f d     YZ d e f d     YZ d S(	   i    (   t   divisiont   print_functiont   absolute_importN(   t   TestCaset   assert_equalt   assert_almost_equalt   assert_allclose(   t   logitt   expitt	   TestLogitc           B` s,   e  Z d    Z d   Z d   Z d   Z RS(   c         C` s   t  j d d d  } t  j | d | } t  j d d  } z t |  } Wd  t  j |   Xt | |  t | j t  j |   d  S(   Ni    i   i
   t   dtypet   dividet   ignore(   t   npt   linspacet   arrayt   seterrR   R   R   R
   (   t   selfR
   t   expectedt   at   olderrt   actual(    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt   check_logit_out
   s    c         C` sQ   t  j t  j d d d d d d d d t  j g
 d	 t  j } |  j d
 |  d  S(   NgcY@ gwQgDB.gԙ̿gc?gC?C.?gN7?Q?gcY@ @R
   t   f4(   R   R   t   inft   float32R   (   R   R   (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt   test_float32   s    c         C` sH   t  j t  j d d d d d d d d t  j g
  } |  j d	 |  d  S(
   Ngu? g$/QgDB.g̿g?gDB.?g$/Q?gu? @t   f8(   R   R   R   R   (   R   R   (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt   test_float64   s    c         C` sr   t  j t  j g d  } t  j d d  } z% t t  j d d d d g   } Wd  t  j |   Xt | |  d  S(   Ni   t   invalidR   g      g       g       @g      @(   R   R   t   nanR   R   R   (   R   R   R   R   (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt   test_nan'   s    %(   t   __name__t
   __module__R   R   R   R   (    (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyR	   	   s   			t	   TestExpitc           B` s,   e  Z d    Z d   Z d   Z d   Z RS(   c         C` s`   t  j d d d  } t  j | d | } t |  } t | |  t | j t  j |   d  S(   Nii   i
   R
   (   R   R   R   R   R   R   R
   (   R   R
   R   R   R   (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt   check_expit_out3   s
    c         C` sJ   t  j d d d d d d d d d	 d
 g
 d t  j } |  j d |  d  S(   Ng{j?g\wb֥?g~ܟ?g5?gs4 ?gǻe?g+?>S?gw?gݙ?gl?R
   R   (   R   R   R   R#   (   R   R   (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyR   :   s    c         C` sA   t  j d d d d d d d d d	 d
 g
  } |  j d |  d  S(   Ng{j?g\wb֥?gIq?g!/?gs4 ?gǻe?gw?4S?gw?g4ؙ?g!/l?R   (   R   R   R#   (   R   R   (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyR   B   s    c         C` s   x t  j t  j t  j f D] } x d D]{ } t  j | d | } t t |  d d	 d
 t t |  d d	 d
 t t |  j |  t t |  j |  q& Wq Wd  S(   NiX   iY   i  i  i\,  i],  R
   g      ?t   atolg#B;g        (   iX   iY   i  i  i\,  i],  (	   R   R   t   float64t
   longdoubleR   R   R   R   R
   (   R   R
   t   n(    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt
   test_largeJ   s    (   R    R!   R#   R   R   R(   (    (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyR"   2   s   			(   t
   __future__R    R   R   t   numpyR   t   numpy.testingR   R   R   R   t   scipy.specialR   R   R	   R"   (    (    (    s=   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_logit.pyt   <module>   s
   ")