U
    HvfW                     @   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 )NvIh%<=Zrtolr   sclog_softmax)xexpected r   H/tmp/pip-unpacked-wheel-96ln3f52/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 )Nd   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.   )Z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>   




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