3
OdW              	   @   s   d dl Zd dlmZ d dlZd dljZejj	dej
ddgej
d d'gfejdej
d(d)d*d+gfgdd Zejdd Zejdd Zdd Zdd Zejj	dd ejdejd, fdej
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, expectedi     i     g:-@g:-@guZ?gi,?c             C   s   t tj| |dd d S )NgvIh%<=)rtol)r   sclog_softmax)xexpected r
   X/var/www/html/virt/lib64/python3.6/site-packages/scipy/special/tests/test_log_softmax.pytest_log_softmax	   s    r   c              C   s   t jd} | S )Nr   )nparange)r   r
   r
   r   log_softmax_x   s    
r   c              C   s   t jddddg} | S )	Ng:-@g:-@guZ?gi,?g:-g:-guZgi,ܿ)r   array)r	   r
   r
   r   log_softmax_expected   s
    r   c             C   s$   | d }|}t tj||dd d S )Nd   gvIh%<=)r   )r   r   r   )r   r   r   r	   r
   r
   r   test_log_softmax_translation(   s    r   c             C   s0   | j dd}|j dd}ttj||dd d S )N   gvIh%<=)r   )reshaper   r   r   )r   r   r   r	   r
   r
   r   test_log_softmax_noneaxis0   s    r   zaxis_2d, expected_2dg      ?r   c             C   s(   t tjddgddgg| d|dd d S )Ni  r   )axisgvIh%<=)r   )r   r   r   )Zaxis_2dZexpected_2dr
   r
   r   	test_axes8   s    r   c              C   s   t jdjdd} | S )N   r   r   )r   r   r   )r   r
   r
   r   log_softmax_2d_xD   s    r   c              C   s"   t jddddgd	d
ddgg} | S )Ng:-@g:-@guZ?gi,?g:-g:-guZgi,ܿg:-g:-guZgi,ܿ)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   gvIh%<=)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   gvIh%<=)r   )Tr   r   r   )r   r   r   r	   r
   r
   r   test_log_softmax_2d_axis0_   s    r   c             C   s8   | j ddd}|j ddd}ttj|dd|dd d S )Nr   r   )r   gvIh%<=)r   )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jdddd d S )Ng      ?g        gvIh%<=)r   )r   r   r   r
   r
   r
   r   test_log_softmax_scalarl   s    r    ig:-g:-guZgi,ܿ)r   r   ii)Z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>   s,   
&
