
    !Цi                        S SK r S SKrS SKJrJrJr  S SKr\ R                  R                  \ R                  R                  \5      S5      r\ R                  R                  \S5      r\R                  " \5      rS rS rS rS rS	 rg)
    N)assert_assert_allcloseassert_raisesdatazwavelab_test_signals.npzc                      [         R                  R                  5       n [        R                  " / SQ/ SQ/ SQ/5      n[        U S S2S S24   U5        g )N)   r      )         )      r      )pywtr   aeronparrayr   )r   refs     S/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/pywt/tests/test_data.pytest_data_aeror      sI    99>>D
((O##% &C D!RaRL#&    c                      [         R                  R                  5       n [        R                  " / SQ/ SQ/ SQ/5      n[        U S S2S S24   U5        g )N)S   r   r   )R   r   r   )P   Q   r   r   )r   r   ascentr   r   r   )r   r   s     r   test_data_ascentr      sK    YYF
((L  " #C F2A2rr6NC(r   c                      [         R                  R                  5       n [        R                  " / SQ/ SQ/ SQ/5      n[        U S S2S S24   U5        g )N)   r    r    )r       r!   )r!   r!   r!   r   )r   r   camerar   r   r   )r"   r   s     r   test_data_camerar#   !   sK    YYF
((O##% &C F2A2rr6NC(r   c                      [         R                  R                  5       n [        R                  " / SQ5      n[        U S S U5        g )N)ir%   r   )r   r   ecgr   r   r   )r&   r   s     r   test_data_ecgr'   +   s/    
))--/C
((?
#CCGS!r   c                     S=p[         R                  R                  S5      n[        SU;   5        U H  nUR	                  SS5      R                  5       n[        U   nUS;   a]  [        U[         R                  R                  U5      XS9  [        [        [         R                  R                  XER                  5        M  [        U[         R                  R                  X5R                  5      XS9  [        [        [         R                  R                  U5        M     [        [        [         R                  R                  SS	5        [        [        [         R                  R                  SS
5        g)z/Comparison with results generated using WaveLabg-q=listDoppler-_)gaborsineoneoverx)rtolatolunknown_signali   r   N)r   r   demo_signalr   replacelowerwavelab_result_dictr   r   
ValueErrorsize)r/   r0   available_signalssignalkeyvals         r   test_wavelab_signalsr<   3   s   D 		--f5I**+#nnS#&,,.!#&++C!6!6v!>!%2*dii&;&;S((KC!6!6vxx!H!%2 *dii&;&;SA $  *dii335EsK *dii33YBr   )osnumpyr   numpy.testingr   r   r   	pywt.datar   pathjoindirname__file__data_dirwavelab_data_fileloadr5   r   r   r#   r'   r<    r   r   <module>rI      su    	  A A 77<<16:GGLL+EF gg/0 '))"Cr   