
    ^Mh*                         d dl Zd dlmZmZmZ d dlmZmZm	Z	m
Z
 d dlZd Zd Zd Zd Zd Zd	 Zd
 Zd Zej                            dddgddgg          d             ZdS )    N)assert_equalassert_almost_equalassert_allclose)boxcoxboxcox1p
inv_boxcoxinv_boxcox1pc                     t          j        g d          } t          | d          }t          |t          j        |                      t          | d          }t          || dz
             t          | d          }t          |d| dz  dz
  z             t          j        g d          }t          d|          }t          |d|z             d S )N)      ?         r   r   r   r   r   r   r         )nparrayr   r   logxylams      _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/special/tests/test_boxcox.pytest_boxcox_basicr   	   s    
  A 	q!A26!99%%% 	q!A1q5!!! 	q!A31q>*** (;;;

Cq#A4#:&&&&&    c                  v    d} d}t          | |          }t          |t          j        |           d           d S )Ng     ?P7xf +=rtol)r   r   r   r   r   lmbdar   s      r   test_boxcox_underflowr"      s>    AEq%AArvayyu------r   c                  x   t          j        g d          } t          | g d          }t          |t          j        t           j        t           j        t           j        g                     d} t          | ddg          }t          |t          j        t           j         t           j         g                     d S )N)r$   g      ࿩r          @      r         )r   r   r   r   naninfr   r   s     r   test_boxcox_nonfiniter,   $   s    
  Aq"""##ABHbfbfbf566777 	
Aq4)ABHrvgw/0011111r   c                     t          j        g d          } t          | d          }t          |t          j        |                      t          | d          }t          ||            t          | d          }t          |d| z  d| z   z             t          j        g d          }t          d|          }t          |d|z             d S )	N)g      пg#Bǻr   g#B;g      ?r      r   r   r   r   r   r$   r   )r   r   r   r   log1pr   s      r   test_boxcox1p_basicr0   0   s    
66677A 	AA28A;;''' 	AA1 	AA3q5!a%=))) (;;;

CSA4#:&&&&&r   c                      t          j        ddg          } t          j        ddg          }t          | |          }t          |t          j        |           d           d S )NV瞯<r   gC]r2<r   r   )r   r   r   r   r/   r    s      r   test_boxcox1p_underflowr3   E   sZ    
%!!AHfe_%%EEAArx{{//////r   c                  x   t          j        g d          } t          | g d          }t          |t          j        t           j        t           j        t           j        g                     d} t          | ddg          }t          |t          j        t           j         t           j         g                     d S )N)r5   r'   r%   r$   r(   r   )r   r   r   r   r)   r*   r+   s     r   test_boxcox1p_nonfiniter6   L   s    
  A$$$%%ABHbfbfbf566777 	AT1IABHrvgw/0011111r   c                  v   t          j        g d          } t          j        g d          }t          | |          }t          ||          }t	          | |           t          j        g d          } t          j        g d          }t          | |          }t          ||          }t	          | |           d S )N)g        g      ?r&   )r   r   r   r   r   r   r	   )r   r   r   x2s       r   test_inv_boxcoxr9   X   s    
A
(<<<
 
 Cq#A	As		B2
A
(<<<
 
 CCA	a		B2r   c                  R    d} d}t          | |          }t          || d           d S )Nr2   r   r   r   )r	   r   )r   r   r   s      r   test_inv_boxcox1p_underflowr;   f   s6    A
CQAAqu%%%%%%r   zx, lmbd      g{Gz?iec                 *   t          | |          }t          j        |          sJ t          ||          }t	          | |           t          | dz
  |          }t          j        |          sJ t          ||          }t	          | dz
  |           d S )Nr   )r   r   isfiniter   r   r   r	   )r   lmbr   x_invy1px1p_invs         r   test_boxcox_premature_overflowrD   m   s     	q#A;q>>q#EAu 1Q3

C;s3$$GAaC!!!!!r   )numpyr   numpy.testingr   r   r   scipy.specialr   r   r   r	   pytestr   r"   r,   r0   r3   r6   r9   r;   markparametrizerD    r   r   <module>rL      s$       L L L L L L L L L L D D D D D D D D D D D D 
' ' '(. . .	2 	2 	2' ' '*0 0 0	2 	2 	2  & & & 	3Z
D\ 
" " 
" " "r   