o
    iwh                     @   s   d dl Z d dlZd dlmZmZmZ d dlZe j	
e j	edZe j	
edZeeZdd Zdd Zd	d
 Zdd Zdd ZdS )    N)assert_assert_allcloseassert_raisesdatazwavelab_test_signals.npzc                  C   D   t j } tg dg dg dg}t| d dd df | d S )N)   r      )         )      r      )pywtr   aeronparrayr   )r   ref r   V/var/www/html/ecg_monitoring/venv/lib/python3.10/site-packages/pywt/tests/test_data.pytest_data_aero      

r   c                  C   r   )N)S   r   r   )R   r   r   )P   Q   r   r   )r   r   ascentr   r   r   )r   r   r   r   r   test_data_ascent   r   r   c                  C   r   )N)   r   r   )r      r   )r   r   r   r   )r   r   camerar   r   r   )r    r   r   r   r   test_data_camera!   r   r!   c                  C   s.   t j } tg d}t| d d | d S )N)ir"   r   )r   r   ecgr   r   r   )r#   r   r   r   r   test_data_ecg+   s   
r$   c                  C   s   d } }t jd}td|v  |D ]?}|dd }t| }|dv r;t|t j|| |d tt	t jj||j
 qt|t j||j
| |d tt	t jj| qtt	t jjdd	 tt	t jjdd
 dS )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+   r,   available_signalssignalkeyvalr   r   r   test_wavelab_signals3   s"   r8   )osnumpyr   numpy.testingr   r   r   	pywt.datar   pathjoindirname__file__data_dirwavelab_data_fileloadr1   r   r   r!   r$   r8   r   r   r   r   <module>   s    



