o
    e'	                     @   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_allcloseassert_raisesassert_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   DD:\Projects\ConvertPro\env\Lib\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=listZDoppler-_)ZgaborZsineoneoverx)rtolatolZunknown_signali   r   N)r   r   Zdemo_signalr   replacelowerwavelab_result_dictr   r   
ValueErrorsize)r(   r)   Zavailable_signalssignalkeyvalr   r   r   test_wavelab_signals2   s"   r2   )osnumpyr   Znumpy.testingr   r   r   Z	pywt.datar   pathjoindirname__file__data_dirZwavelab_data_fileloadr,   r   r   r!   r$   r2   r   r   r   r   <module>   s    



