§
    µ³ÙgW  ã            
       ó\  — 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ùÙ:-‚…Àgò³uZ÷¿gÉÏÖi,Ü¿c                 óN   — t          t          j        | ¦  «        |d¬¦  «         d S )Nç‚vIhÂ%<=©Úrtol©r   ÚscÚlog_softmax)ÚxÚexpecteds     úZ/granary/edev/quantum/lib/python3.11/site-packages/scipy/special/tests/test_log_softmax.pyÚtest_log_softmaxr   	   s)   € õ •B”N 1Ñ%Ô% x°eÐ<Ñ<Ô<Ð<Ð<Ð<ó    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ô 0€Hð €Or   c                 ó\   — | dz   }|}t          t          j        |¦  «        |d¬¦  «         d S )Néd   r	   r
   r   ©r   r   r   r   s       r   Útest_log_softmax_translationr"   (   s8   € ð 	˜Ñ€AØ#€HÝ•B”N 1Ñ%Ô% x°eÐ<Ñ<Ô<Ð<Ð<Ð<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Ø#×+Ò+¨A¨qÑ1Ô1€HÝ•B”N 1Ñ%Ô% x°eÐ<Ñ<Ô<Ð<Ð<Ð<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ô 1€Hð €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Ý•B”N 1¨1Ð-Ñ-Ô-¨x¸eÐ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Ý•B”N 1¨1Ð-Ñ-Ô-¨x¸eÐ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   € à×#Ò# A q¨!Ñ,Ô,€DØ)×1Ò1°!°Q¸Ñ:Ô:€KÝ•B”N 4¨fÐ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'   € Ý•B”N 3Ñ'Ô'¨°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+  ðØ Ð Ð Ð Ø )Ð )Ð )Ð )Ð )Ð )à €€€à Ð Ð Ð Ð Ð ð „×Ò˜Ø€R„Xˆt�QˆiÑÔ˜(˜"œ( A t 9Ñ-Ô-Ð.ð €R„Yˆq�\„\�8�2”8ð 3ð 3ð 3ñ 4ô 4ð 5ð	)ñ 	ô 	ð=ð =ñ	ô 	ð=ð „ðð ñ „ðð
 „ðð ñ „ðð=ð =ð =ð=ð =ð =ð „×ÒÐ/ØˆˆŒˆs‰Œ�g�b”g˜f‘o”oÑ%Ð&ØˆˆŒ�1�d�)˜a ˜YÐ'Ñ(Ô(Ð)ð2ñ ô ðð ñ	ô ðð „ðð ñ „ðð
 „ðð ñ „ððEð Eð EðEð Eð EðPð Pð Pð:ð :ð :ð :ð :r   