
    ^MhW              
       \   d dl Zd dlmZ d dlZd dlmZ ej        	                    d ej
        ddg           ej
        d dg          f ej        d           ej
        g d          fg          d	             Zej        d
             Zej        d             Zd Zd Zej        	                    dd  ej        d           ej        d          z  fd ej
        d dgd dgg          fg          d             Zej        d             Zej        d             Zd Zd Zd Zd ZdS )    N)assert_allclosezx, expected     i   g:-g:-guZgi,ܿc                 N    t          t          j        |           |d           d S )NvIh%<=rtolr   sclog_softmax)xexpecteds     d/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/special/tests/test_log_softmax.pytest_log_softmaxr   	   s)     BN1%%xe<<<<<<    c                  .    t          j        d          } | S )Nr   )nparanger   s    r   log_softmax_xr      s    
	!AHr   c                  2    t          j        g d          } | S Nr   r   arrayr   s    r   log_softmax_expectedr      s+     x / / / 0 0H Or   c                 \    | dz   }|}t          t          j        |          |d           d S )Nd   r	   r
   r   r   r   r   r   s       r   test_log_softmax_translationr"   (   s8     	A#HBN1%%xe<<<<<<r   c                     |                      dd          }|                     dd          }t          t          j        |          |d           d S )N   r	   r
   reshaper   r   r   r!   s       r   test_log_softmax_noneaxisr'   0   sS     	a##A#++Aq11HBN1%%xe<<<<<<r   zaxis_2d, expected_2dg      ?)r$   r$   c                 ^    t          t          j        ddgddgg|           |d           d S )Nr   r   axisr	   r
   r   )axis_2dexpected_2ds     r   	test_axesr-   8   sJ    
 
q	D!9-G<<<     r   c                  V    t          j        d                              dd          } | S )N   r$   r   )r   r   r&   r   s    r   log_softmax_2d_xr0   D   s$    
	!Q""AHr   c                  :    t          j        g dg dg          } | S r   r   r   s    r   log_softmax_2d_expectedr2   J   s?     x / / // / /	0 1 1H Or   c                 Z    | }|}t          t          j        |d          |d           d S )Nr   r)   r	   r
   r   r0   r2   r   r   s       r   test_log_softmax_2d_axis1r5   Y   s6    A&HBN11---xeDDDDDDr   c                 n    | j         }|j         }t          t          j        |d          |d           d S )Nr   r)   r	   r
   )Tr   r   r   r4   s       r   test_log_softmax_2d_axis0r8   _   s<    A&(HBN11---xeDDDDDDr   c                     |                      ddd          }|                     ddd          }t          t          j        |d          |d           d S )Nr$   )r   r$   r)   r	   r
   r%   )r0   r2   x_3dexpected_3ds       r   test_log_softmax_3dr<   e   sZ    ##Aq!,,D)11!Q::KBN4f555{OOOOOOr   c                  N    t          t          j        d          dd           d S )Ng      ?g        r	   r
   r    r   r   test_log_softmax_scalarr?   l   s'    BN3''5999999r   )numpyr   numpy.testingr   pytestscipy.specialspecialr   markparametrizer   r   r   fixturer   r   r"   r'   logonesr-   r0   r2   r5   r8   r<   r?   r>   r   r   <module>rJ      s+       ) ) ) ) ) )        RXtQi("(At9--. RYq\\828 3 3 3 4 4 5	) 	 	= =	 	=   
   = = == = = /sgbgfoo%&1d)aY'(()2   	    
   E E EE E EP P P: : : : :r   