
    tKgW              
       ~   d dl Zd dlmZ d dlZd dlmZ ej                  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                  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y)    N)assert_allclosezx, expected     i   g:-g:-guZgi,ܿc                 F    t        t        j                  |       |d       y )NvIh%<=rtolr   sclog_softmax)xexpecteds     h/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/special/tests/test_log_softmax.pytest_log_softmaxr   	   s     BNN1%xe<    c                  0    t        j                  d      } | S )Nr   )nparanger   s    r   log_softmax_xr      s    
		!AHr   c                  4    t        j                  g d      } | S Nr   r   arrayr   s    r   log_softmax_expectedr      s     xx / 0H Or   c                 T    | dz   }|}t        t        j                  |      |d       y )Nd   r	   r
   r   r   r   r   r   s       r   test_log_softmax_translationr"   (   s(     	A#HBNN1%xe<r   c                     | j                  dd      }|j                  dd      }t        t        j                  |      |d       y )N   r	   r
   reshaper   r   r   r!   s       r   test_log_softmax_noneaxisr'   0   s=     	a#A#++Aq1HBNN1%xe<r   zaxis_2d, expected_2dg      ?)r$   r$   c                 V    t        t        j                  ddgddgg|       |d       y )Nr   r   axisr	   r
   r   )axis_2dexpected_2ds     r   	test_axesr-   8   s-    
 
q	D!9-G<r   c                  P    t        j                  d      j                  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(     xx //	0 1H Or   c                 R    | }|}t        t        j                  |d      |d       y )Nr   r)   r	   r
   r   r0   r2   r   r   s       r   test_log_softmax_2d_axis1r5   Y   s#    A&HBNN11-xeDr   c                 z    | j                   }|j                   }t        t        j                  |d      |d       y )Nr   r)   r	   r
   )Tr   r   r   r4   s       r   test_log_softmax_2d_axis0r8   _   s/    A&((HBNN11-xeDr   c                     | j                  ddd      }|j                  ddd      }t        t        j                  |d      |d       y )Nr$   )r   r$   r)   r	   r
   r%   )r0   r2   x_3dexpected_3ds       r   test_log_softmax_3dr<   e   sA    ##Aq!,D)11!Q:KBNN4f5{Or   c                  F    t        t        j                  d      dd       y )Ng      ?g        r	   r
   r    r   r   test_log_softmax_scalarr?   l   s    BNN3'5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      st    )   RXXtQi("((At9-. RYYq\8288 3 4 5	) 	=	=  
  == /sgbggfo%&1d)aY'()2 	  
  EEP:r   