U
    ?h0                     @   sR   d dl Zd dlmZ d dlmZ d dlmZm	Z	m
Z
mZmZmZ G dd dZdS )    N)assert_array_equal)raises)
axis_sliceaxis_reverseodd_exteven_ext	const_extzero_extc                   @   s<   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd ZdS )TestArrayToolsc                 C   s  t ddd}t|dddd}t||ddd d f  t|ddd}t||dd d d f  t|dddd}t||d d ddf  t|ddd}t||d d dd f  t|dd	dd
}t||d d d	d d f  t|dd	dd
}t||d d d d d	f  d S )N         r      )startstopaxis)r   r      )r   stepr   )nparangereshaper   r   )selfas r   U/var/www/html/venv/lib/python3.8/site-packages/scipy/signal/tests/test_array_tools.pytest_axis_slice   s    zTestArrayTools.test_axis_slicec                 C   sf   t ddd}t|dd}t||d d dd d f  t|dd}t||d d d d df  d S )Nr   r   r   r   r   r   r   )r   r   r   r   r   )r   r   rr   r   r   test_axis_reverse!   s
    z TestArrayTools.test_axis_reversec                 C   s   t dddddgdddd	dgg}t|ddd
}t dddddddd	dg	dddddd	dddg	g}t|| t|ddd
}t dddddgdddddgdddd	dgdddddgg}t|| ttt|ddd
 ttt|ddd
 d S )Nr   r   r   r      	            r   r   r      
   i      )r   arrayr   r   assert_raises
ValueError)r   r   Zoddexpectedr   r   r   test_odd_ext*   s"    

zTestArrayTools.test_odd_extc                 C   s   t dddddgdddd	dgg}t|ddd
}t dddddddddg	dddddd	dd	dg	g}t|| t|ddd
}t dddd	dgdddddgdddd	dgdddddgg}t|| ttt|ddd
 ttt|ddd
 d S Nr   r   r   r   r!   r"   r#   r$   r%   r   r   )r   r+   r   r   r,   r-   )r   r   Zevenr.   r   r   r   test_even_ext=   s"    

zTestArrayTools.test_even_extc                 C   s   t dddddgdddd	dgg}t|ddd
}t dddddddddg	dddddd	dddg	g}t|| t|ddd
}t dddddgdddddgdddd	dgdddd	dgg}t|| d S r0   )r   r+   r   r   )r   r   constr.   r   r   r   test_const_extP   s    
zTestArrayTools.test_const_extc                 C   s   t dddddgdddd	dgg}t|ddd
}t dddddddddg	dddddd	dddg	g}t|| t|ddd
}t dddddgdddddgdddd	dgdddddgg}t|| d S r0   )r   r+   r	   r   )r   r   zeror.   r   r   r   test_zero_ext`   s    
zTestArrayTools.test_zero_extN)	__name__
__module____qualname__r   r    r/   r1   r3   r5   r   r   r   r   r
   
   s   	r
   )numpyr   Znumpy.testingr   Zpytestr   r,   Zscipy.signal._arraytoolsr   r   r   r   r   r	   r
   r   r   r   r   <module>   s    