U
    »mœdW  ã                   @   s"  d dl Zd dlmZ d dlZd dlmZ ej 	de 
ddg¡e 
d dg¡fe d¡e 
dd	d
dg¡fg¡dd„ ƒZejdd„ ƒZejdd„ ƒZdd„ Zdd„ Zej 	dd e d¡e d¡ fde 
d dgd dgg¡fg¡dd„ ƒZejdd„ ƒZejdd„ ƒZdd „ Zd!d"„ Zd#d$„ Zd%d&„ ZdS )'é    N)Úassert_allclosezx, expectedéè  é   iüÿÿé   çùÙ:-‚…ÀçùÙ:-‚…Àçò³uZ÷¿çÉÏÖi,Ü¿c                 C   s   t t | ¡|dd� d S )Nç‚vIhÂ%<=©Zrtol©r   ÚscÚlog_softmax)ÚxÚexpected© r   ú]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scipy/special/tests/test_log_softmax.pyÚtest_log_softmax	   s    r   c                  C   s   t  d¡} | S )Nr   )ÚnpÚarange©r   r   r   r   Úlog_softmax_x   s    
r   c                  C   s   t  ddddg¡} | S ©Nr   r   r   r	   ©r   Úarray©r   r   r   r   Úlog_softmax_expected   s    ýr   c                 C   s$   | d }|}t t |¡|dd� d S )Néd   r
   r   r   ©r   r   r   r   r   r   r   Útest_log_softmax_translation(   s    r   c                 C   s0   |   dd¡}|  dd¡}tt |¡|dd� d S )Né   r
   r   ©Úreshaper   r   r   r   r   r   r   Útest_log_softmax_noneaxis0   s    r#   zaxis_2d, expected_2dg      à?)r    r    c                 C   s(   t tjddgddgg| d�|dd� d S )Nr   r   ©Zaxisr
   r   r   )Zaxis_2dZexpected_2dr   r   r   Ú	test_axes8   s
    ýr%   c                  C   s   t  d¡ dd¡} | S )Né   r    r   )r   r   r"   r   r   r   r   Úlog_softmax_2d_xD   s    r'   c                  C   s"   t  ddddgddddgg¡} | S r   r   r   r   r   r   Úlog_softmax_2d_expectedJ   s    ýýür(   c                 C   s$   | }|}t tj|dd�|dd� d S )Nr   r$   r
   r   r   ©r'   r(   r   r   r   r   r   Útest_log_softmax_2d_axis1Y   s    r*   c                 C   s(   | j }|j }ttj|dd�|dd� d S )Nr   r$   r
   r   )ÚTr   r   r   r)   r   r   r   Útest_log_softmax_2d_axis0_   s    r,   c                 C   s8   |   ddd¡}|  ddd¡}ttj|dd�|dd� d S )Nr    )r   r    r$   r
   r   r!   )r'   r(   Zx_3dZexpected_3dr   r   r   Útest_log_softmax_3de   s    r-   c                   C   s   t t d¡ddd� d S )Ng      ð?g        r
   r   r   r   r   r   r   Útest_log_softmax_scalarl   s    r.   )Únumpyr   Znumpy.testingr   ZpytestZscipy.specialÚspecialr   ÚmarkZparametrizer   r   r   Zfixturer   r   r   r#   ÚlogZonesr%   r'   r(   r*   r,   r-   r.   r   r   r   r   Ú<module>   s>   ýû




þ


