
    Ug                         d dl Zd dlmZmZmZ d dlmZmZm	Z	  G d d          Z
 G d d          Z G d d	          ZdS )
    N)assert_equalassert_almost_equalassert_allclose)logitexpit	log_expitc                   &    e Zd Zd Zd Zd Zd ZdS )	TestLogitc                 D   t          j        ddd          }t          j        ||          }t          j        d          5  t	          |          }d d d            n# 1 swxY w Y   t          ||           t          |j        t          j        |                     d S )Nr      
   dtypeignore)divide)nplinspacearrayerrstater   r   r   r   selfr   expectedaactuals        ]/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/scipy/special/tests/test_logit.pycheck_logit_outzTestLogit.check_logit_out   s    K1b!!HQe$$$[))) 	 	1XXF	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	FH---V\28E??33333s   AA"%A"c                     t          j        t           j         ddddddddt           j        g
t           j        	          }|                     d
|           d S )NgcY@ gwQDB.gԙ̿gc?gC?C.?gN7?Q?gcY@ @r   f4)r   r   inffloat32r   r   r   s     r   test_float32zTestLogit.test_float32   sc    8bfWk''%z&	0 8:z	C C C
 	T8,,,,,    c                     t          j        t           j         ddddddddt           j        g
          }|                     d	|           d S )
Ngu? g$/Qr   g̿g?gDB.?g$/Q?gu? @f8)r   r   r    r   r"   s     r   test_float64zTestLogit.test_float64   sT    8bfWk''&
&	0 1 1
 	T8,,,,,r$   c                 
   t          j        t           j        gdz            }t          j        d          5  t	          t          j        g d                    }d d d            n# 1 swxY w Y   t          ||           d S )N   r   )invalid)g      g       g       @g      @)r   r   nanr   r   r   )r   r   r   s      r   test_nanzTestLogit.test_nan"   s    8RVHQJ''[*** 	9 	928$6$6$67788F	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	9 	Xv&&&&&s   $A((A,/A,N)__name__
__module____qualname__r   r#   r'   r,    r$   r   r
   r
      sP        4 4 4- - -- - -' ' ' ' 'r$   r
   c                   &    e Zd Zd Zd Zd Zd ZdS )	TestExpitc                     t          j        ddd          }t          j        ||          }t          |          }t	          ||           t          |j        t          j        |                     d S )Nr)   r   r   )r   r   r   r   r   r   r   r   s        r   check_expit_outzTestExpit.check_expit_out+   sd    KAr""HQe$$$qFH---V\28E??33333r$   c                 v    t          j        g dt           j                  }|                     d|           d S )N)
{j?\wb֥?g~ܟ?g5?s4 ?ǻe?g+?>S?w?gݙ?gl?r   r   )r   r   r!   r5   r"   s     r   r#   zTestExpit.test_float322   sO    8 4 4 4 <>:	G G G
 	T8,,,,,r$   c                 ^    t          j        g d          }|                     d|           d S )N)
r7   r8   gIq?g!/?r9   r:   gw?4S?r;   g4ؙ?g!/l?r&   )r   r   r5   r"   s     r   r'   zTestExpit.test_float64:   s@    8 4 4 4 5 5
 	T8,,,,,r$   c                    t           j        t           j        t           j        fD ]}dD ]}t          j        ||          }t          t          |          dd           t          t          |           dd           t          t          |          j        |           t          t          |           j        |           d S )N)X   Y   i    i\,  i],  r   g      ?g#B;)atolg        )	r   r!   float64
longdoubler   r   r   r   r   )r   r   ns      r   
test_largezTestExpit.test_largeB   s    j"*bm< 	5 	5E5 5 5HQe,,,a#E::::qb		3U;;;;U1XX^U333UA2YY_e44445	5 	5r$   N)r-   r.   r/   r5   r#   r'   rE   r0   r$   r   r2   r2   *   sP        4 4 4- - -- - -5 5 5 5 5r$   r2   c                   &    e Zd Zd Zd Zd Zd ZdS )TestLogExpitc                 p    t          j        g d          }t          |          }t          ||           d S )N)g     g     pg     @g     Ar   r   r   r   r   xys      r   test_large_negativez TestLogExpit.test_large_negativeN   s9    H66677aLLQr$   c                     t          j        g d          }t          |          }t          |t          j        g d                     d S )N)g     p@g     @@g     @)       rO   rO   rI   rJ   s      r   test_large_positivez TestLogExpit.test_large_positiveS   sL    H---..aLL
 	Q!3!3!34455555r$   c                 |    t          j        g d          }t          |          }g d}t          ||d           d S )N)皙&.r   &.>皙?r   r   d   i  r@   i  i  )g     @gC   4g(U $g\ÁcgIgOg>B.g9B.gB.g]g$zԿgLVg]gCx^&ْgó(8g   g'1     gV瞯<rtol)r   r   r   r   r   rK   rL   r   s       r   test_basic_float64zTestLogExpit.test_basic_float64\   s`    H D D D E EaLL! ! ! 	8%000000r$   c                     t          j        g dt           j                  }t          |          }t          j        g dt           j                  }t	          ||d           d S )N)rR   rS   rT   rU   rV   rW   rX   r   rY   rZ   r   r   r[   r   )g      @g      4g $gcgm{qg8j	C.ra   ra   g@g %zԿgcAVgα!gƠ>r\   )r   r   r!   r   r   r^   s       r   test_basic_float32zTestLogExpit.test_basic_float32{   s    H 0 0 079zC C CaLL 8 C C C #%*	. . . 	8$//////r$   N)r-   r.   r/   rM   rP   r_   rb   r0   r$   r   rG   rG   L   sP          
6 6 61 1 1>0 0 0 0 0r$   rG   )numpyr   numpy.testingr   r   r   scipy.specialr   r   r   r
   r2   rG   r0   r$   r   <module>rf      s       , , , , , , , , , , 1 1 1 1 1 1 1 1 1 1 '  '  '  '  '  '  '  'F5 5 5 5 5 5 5 5DE0 E0 E0 E0 E0 E0 E0 E0 E0 E0r$   