
    (pha                     B    S SK rS SKJrJr  S SKJs  Jr   " S S5      r	g)    N)assert_array_equalassert_array_almost_equalc                        \ rS rSrS rS rSrg)TestWavelets   c                 L   [         R                  " SS5      nS[        R                  " S5      [        R                  S-  -  -  n[        X5        / SQnU HK  n[         R                  " US5      n[        U5      U:X  d   e[        R                  " U5      nXTS-  :X  a  MK   e   Sn[         R                  " US5      n[        R                  " S	US-  5      n[        X   XS-   *    5        / S
QnSnU H  n	[         R                  " Xi5      n[        R                  " S	U5      US-
  S-  -
  n
[        R                  " [        R                  " X-
  5      5      n[        R                  " [        R                  " X-   5      5      n[        X   S	5        [        X   S	5        M     g )N      ?      g      @g      ?)         3   e   d   g       @r   )r   
   r         c   )wavelets_rickernpsqrtpir   lenargmaxaranger   argminabs)selfwexpectedlengthslengthmax_locpointshalf_vecaasavec	exp_zero1	exp_zero2s                S/var/www/html/venv/lib/python3.13/site-packages/scipy/signal/tests/test_wavelets.pytest_rickerTestWavelets.test_ricker   sZ   S!$(BEETM:;1'&F  -Aq6V###iilG{+++	  VS)99Q!,!!+qa</AB "A  +A))Av&&3,!);;C		"&&/2I		"&&/2I%alA6%alA6     c                 :   S/nS nSn[         R                  " [         R                  [         R                  " SU5      -  S-  5      n[        R
                  " XBU5      nUR                  [        U5      U4:X  d   e[        XER                  5       5        / SQn[        R
                  " U[        R                  U5      nUR                  [        U5      U4:X  d   eUS-  /nS n[        R
                  " XFU5      n[        U[         R                  " U5      5        g )	Nr	   c                 0    [         R                  " S/5      $ Nr
   )r   array)sts     r-   delta_wavelet,TestWavelets.test_cwt.<locals>.delta_wavelet'   s    88QC= r0   r   r   g      $@)r
         r   r   r   c                 6    [         R                  " USU-  5      $ r3   )r   full)lr!   s     r-   flat_wavelet+TestWavelets.test_cwt.<locals>.flat_wavelet8   s    771a!e$$r0   )r   sinr   r   r   _cwtshaper   r   flattenr   mean)r    widthsr7   len_data	test_datacwt_datr>   s          r-   test_cwtTestWavelets.test_cwt%   s    	!FF255299Q#99D@A	 --	&A}}Vh 7777!)__->? "--	8+;+;VD}}Vh 7777R-	%--	@!'2779+=>r0    N)__name__
__module____qualname____firstlineno__r.   rI   __static_attributes__rK   r0   r-   r   r      s    7:?r0   r   )
numpyr   numpy.testingr   r   scipy.signal._waveletssignal	_waveletsr   r   rK   r0   r-   <module>rV      s     G ) )4? 4?r0   