o
    	hW                  
   @   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%<=rtolr   sclog_softmax)xexpected r   X/var/www/vscode/kcb/lib/python3.10/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   axisr   r   r
   )axis_2d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)   x_3dexpected_3dr   r   r   test_log_softmax_3de   s   r0   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   r1   )numpyr   numpy.testingr   pytestscipy.specialspecialr   markparametrizer   r   r   fixturer   r   r   r!   logonesr&   r(   r)   r+   r-   r0   r1   r   r   r   r   <module>   s8    

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