
    i                        d dl Z d dlZd dlmZmZmZ d dlZe j                  j                  e j                  j                  e      d      Ze j                  j                  ed      Z ej                  e      Zd Zd Zd Zd Zd	 Zy)
    N)assert_assert_allcloseassert_raisesdatazwavelab_test_signals.npzc                      t         j                  j                         } t        j                  g dg dg dg      }t        | d dd df   |       y )N)   r      )         )      r      )pywtr   aeronparrayr   )r   refs     P/var/www/html/BatchJob/venv/lib/python3.12/site-packages/pywt/tests/test_data.pytest_data_aeror      sI    99>>D
((O##% &C D!RaRL#&    c                      t         j                  j                         } t        j                  g dg dg dg      }t        | d dd df   |       y )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                      t         j                  j                         } t        j                  g dg dg dg      }t        | d dd df   |       y )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                      t         j                  j                         } t        j                  g d      }t        | d d |       y )N)ir%   r   )r   r   ecgr   r   r   )r&   r   s     r   test_data_ecgr'   +   s/    
))--/C
((?
#CCGS!r   c                     dx} }t         j                  j                  d      }t        d|v        |D ]  }|j	                  dd      j                         }t        |   }|dv rat        |t         j                  j                  |      | |       t        t        t         j                  j                  ||j                         t        |t         j                  j                  ||j                        | |       t        t        t         j                  j                  |        t        t        t         j                  j                  dd	       t        t        t         j                  j                  dd
       y)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   D4 		--f5I**+# BnnS#&,,.!#&++C!6!6v!>!%D2*dii&;&;S#((KC!6!6vsxx!H!%D2 *dii&;&;SAB  *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      sw    	  A A 77<<16:GGLL+EF bgg/0 '))"Cr   