U
    ?h                     @   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   sJ   t j } tdddgdddgdddgg}t| d dd df | d S )	N                        )pywtr   aeronparrayr   )r   ref r   F/var/www/html/venv/lib/python3.8/site-packages/pywt/tests/test_data.pytest_data_aero   s    
r   c                  C   sJ   t j } tdddgdddgdddgg}t| d dd df | d S )NS   R   P   Q   r   )r   r   ascentr   r   r   )r   r   r   r   r   test_data_ascent   s    
r   c                  C   sJ   t j } tdddgdddgdddgg}t| d dd df | d S )N      r   )r   r   camerar   r   r   )r   r   r   r   r   test_data_camera    s    
r   c                  C   s0   t j } tdddg}t| d d | d S )Niir   )r   r   ecgr   r   r   )r    r   r   r   r   test_data_ecg*   s    
r!   c                  C   s   d } }t jd}td|k |D ]~}|dd }t| }|dkrvt|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&      r/   )osnumpyr   Znumpy.testingr   r   r   Z	pywt.datar   pathjoindirname__file__data_dirZwavelab_data_fileloadr)   r   r   r   r!   r/   r   r   r   r   <module>   s   



