U
    »mœdp
  ã                   @   sx   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„ Zdd„ Zdd	„ Zd
d„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )é    N)Úassert_equalÚassert_almost_equalÚassert_allclose)ÚboxcoxÚboxcox1pÚ
inv_boxcoxÚinv_boxcox1pc                  C   s�   t  ddddg¡} t| dƒ}t|t  | ¡ƒ t| dƒ}t|| d ƒ t| dƒ}t|d| d d  ƒ t  dddg¡}td|ƒ}t|d| ƒ d S )Nç      à?é   é   é   r   ç      ð¿)ÚnpÚarrayr   r   Úlog©ÚxÚyÚlam© r   úX/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scipy/special/tests/test_boxcox.pyÚtest_boxcox_basic   s    



r   c                  C   s*   d} d}t | |ƒ}t|t | ¡dd� d S )Ng     ð?ç‘÷P7žxf ç›+¡†›„=©Zrtol)r   r   r   r   ©r   Zlmbdar   r   r   r   Útest_boxcox_underflow   s    
r   c                  C   sn   t  dddg¡} t| dddgƒ}t|t  t jt jt jg¡ƒ d} t| ddgƒ}t|t  t j t j g¡ƒ d S )Néÿÿÿÿg      à¿r	   ç       @ç      ø¿r   ç      À)r   r   r   r   ÚnanÚinf©r   r   r   r   r   Útest_boxcox_nonfinite#   s    r$   c               	   C   s’   t  dddddddg¡} t| dƒ}t|t  | ¡ƒ t| dƒ}t|| ƒ t| dƒ}t|d	|  d|   ƒ t  d	ddg¡}td
|ƒ}t|d| ƒ d S )Ng      Ð¿g#B’¡œÇ»r   g#B’¡œÇ;g      Ð?r
   é   r   r	   r   r   )r   r   r   r   Úlog1pr   r   r   r   Útest_boxcox1p_basic/   s    




r'   c                  C   s>   t  ddg¡} t  ddg¡}t| |ƒ}t|t  | ¡dd� d S )NçVçž¯Ò<r   g¬CÒÑ]r2<r   r   )r   r   r   r   r&   r   r   r   r   Útest_boxcox1p_underflowD   s    
r)   c                  C   sn   t  dddg¡} t| dddgƒ}t|t  t jt jt jg¡ƒ d} t| ddgƒ}t|t  t j t j g¡ƒ d S )Néþÿÿÿr   r	   r   r   r    r   )r   r   r   r   r!   r"   r#   r   r   r   Útest_boxcox1p_nonfiniteK   s    r+   c                  C   s€   t  dddg¡} t  dddg¡}t| |ƒ}t||ƒ}t| |ƒ t  dddg¡} t  dddg¡}t| |ƒ}t||ƒ}t| |ƒ d S )Ng        g      ð?r   )r   r   r   r   r   r   r   )r   r   r   Zx2r   r   r   Útest_inv_boxcoxW   s    




r,   c                  C   s$   d} d}t | |ƒ}t|| dd� d S )Nr(   r   r   r   )r   r   )r   r   r   r   r   r   Útest_inv_boxcox1p_underflowe   s    
r-   )Únumpyr   Znumpy.testingr   r   r   Zscipy.specialr   r   r   r   r   r   r$   r'   r)   r+   r,   r-   r   r   r   r   Ú<module>   s   