
    _Mh                     `    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          ZdS )    N)xp_assert_equal)raises)
axis_sliceaxis_reverseodd_exteven_ext	const_extzero_extc                   2    e Zd Zd Zd Zd Zd Zd Zd ZdS )TestArrayToolsc                    t          j        d                              dd          }t          |ddd          }t	          ||ddd d f                    t          |dd          }t	          ||dd d d f                    t          |ddd          }t	          ||d d ddf                    t          |dd          }t	          ||d d dd f                    t          |dd	d
          }t	          ||d d d	d d f                    t          |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ss      c/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/signal/tests/test_array_tools.pytest_axis_slicezTestArrayTools.test_axis_slice   si   IbMM!!!Q''q2221QqS!!!V9%%%q+++1RSS!!!V9%%%q2221QQQ!V9%%%q+++1QQQV9%%%q2221SSqS!!!V9%%%q2221QQQ!V9%%%%%    c                    t          j        d                              dd          }t          |d          }t	          ||d d dd d f                    t          |d          }t	          ||d d d d df                    d S )Nr   r   r   r   r   r   r   )r   r   r   r   r   )r   r   rs      r   test_axis_reversez TestArrayTools.test_axis_reverse!   s    IbMM!!!Q''###1TTrT111W:&&&###1QQQ"W:&&&&&r    c                    t          j        g dg dg          }t          |dd          }t          j        g dg dg          }t          ||           t          |dd          }t          j        g d	g dg dg d
g          }t          ||           t	          t
          t          |dd           t	          t
          t          |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expecteds       r   test_odd_extzTestArrayTools.test_odd_ext*   s   Hooo%oo' ( ( a###8999:::< = =X&&&a###8///,__,__///1 2 2 	X&&&j'1aa8888j'1aa888888r    c                    t          j        g dg dg          }t          |dd          }t          j        g dg dg          }t          ||           t          |dd          }t          j        g dg dg dg dg          }t          ||           t	          t
          t          |dd           t	          t
          t          |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   r2   r   r   r3   r4   )r   r   evenr6   s       r   test_even_extzTestArrayTools.test_even_ext=   s   Hooo%oo' ( ( 11%%%8888888: ; ;h'''11%%%8___,__,__,__. / / 	h'''j(Aqq9999j(Aqq999999r    c                 6   t          j        g dg dg          }t          |dd          }t          j        g dg dg          }t          ||           t          |dd          }t          j        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   r2   r	   r   )r   r   constr6   s       r   test_const_extzTestArrayTools.test_const_extP   s    Hooo%oo' ( ( !QQ'''8888888: ; ;x(((!QQ'''8___,__,__,__. / / 	x(((((r    c                 6   t          j        g dg dg          }t          |dd          }t          j        g dg dg          }t          ||           t          |dd          }t          j        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   r2   r
   r   )r   r   zeror6   s       r   test_zero_extzTestArrayTools.test_zero_ext`   s    Hooo%oo' ( ( 11%%%8888888: ; ;h'''11%%%8___,__,__,__. / / 	h'''''r    N)	__name__
__module____qualname__r   r$   r7   r:   r=   r@    r    r   r   r   
   sn        & & &*' ' '9 9 9&: : :&) ) ) ( ( ( ( (r    r   )numpyr   scipy._lib._array_apir   pytestr   r3   scipy.signal._arraytoolsr   r   r   r   r	   r
   r   rD   r    r   <module>rI      s        1 1 1 1 1 1 * * * * * *- - - - - - - - - - - - - - - -d( d( d( d( d( d( d( d( d( d(r    