a
    QeW                  
   @   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
g d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   g:-g:-guZgi,ܿc                 C   s   t t| |dd d S )NvIh%<=Zrtolr   sclog_softmax)xexpected r   X/var/www/sunrise/env/lib/python3.9/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 g d} | S Nr   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 g dg d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.specialZspecialr
   markZparametrizer   r   r   Zfixturer   r   r   r    logZonesr"   r$   r%   r'   r)   r*   r+   r   r   r   r   <module>   s6   




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