U
    /‰d0  ã                   @   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   úG/tmp/pip-unpacked-wheel-9gxwnfpp/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   Zzeror.   r   r   r   Útest_zero_ext`   s    ÿÿ
ýzTestArrayTools.test_zero_extN)	Ú__name__Ú
__module__Ú__qualname__r   r    r/   r1   r3   r4   r   r   r   r   r
   
   s   	r
   )Znumpyr   Znumpy.testingr   Zpytestr   r,   Zscipy.signal._arraytoolsr   r   r   r   r   r	   r
   r   r   r   r   Ú<module>   s    