o
    g0                     @   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   a/home/ubuntu/cloudmapper/venv/lib/python3.10/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 g dg dg}t|ddd}t g dg dg}t|| t|ddd}t g d	g dg dg d
g}t|| ttt|ddd ttt|ddd d S )Nr   r   r   r      	            r"   r   r   r   )	r   r   r   r   r   r   r"   r'   r&   )	   
   r$   r%   r&   r'   r"   r   r   r   )ir   r   r"   )      r(   r%   r"   r"   )r   arrayr   r   assert_raises
ValueError)r   r   oddexpectedr   r   r   test_odd_ext*   "   




zTestArrayTools.test_odd_extc                 C   s   t g dg dg}t|ddd}t g dg dg}t|| t|ddd}t g dg dg dg d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&   r'   r"   r'   r&   r   r"   )r   r-   r   r   r.   r/   )r   r   evenr1   r   r   r   test_even_ext=   r3   zTestArrayTools.test_even_extc                 C   s   t g dg dg}t|ddd}t g dg dg}t|| t|ddd}t g dg dg dg dg}t|| d S )	Nr!   r#   r   r   r   )	r   r   r   r   r   r   r"   r"   r"   )	r$   r$   r$   r%   r&   r'   r"   r"   r"   r   )r   r-   r   r   )r   r   constr1   r   r   r   test_const_extP      



zTestArrayTools.test_const_extc                 C   s   t g dg dg}t|ddd}t g dg dg}t|| t|ddd}t g d	g dg dg d	g}t|| d S )
Nr!   r#   r   r   r   )	r   r   r   r   r   r   r"   r   r   )	r   r   r$   r%   r&   r'   r"   r   r   r   )r   r   r   r   r   )r   r-   r	   r   )r   r   zeror1   r   r   r   test_zero_ext`   r8   zTestArrayTools.test_zero_extN)	__name__
__module____qualname__r   r    r2   r5   r7   r:   r   r   r   r   r
   
   s    	r
   )numpyr   numpy.testingr   pytestr   r.   scipy.signal._arraytoolsr   r   r   r   r   r	   r
   r   r   r   r   <module>   s
     