U
    f»{j   ã                   @   sž   d dl mZ d dlZd dlmZ d dlZd dlm	Z	 d dl
mZ d dlmZ d dlmZ d dlmZ ejdd	�Ze ed
d¡Zejdd„ ƒZejdd„ ƒZdS )é    N)Úassert_array_equal)Ú
pick_types)Útesting)Ú_test_raw_reader)Úread_raw_cnt)Úread_annotationsF)ÚdownloadZCNTzscan41_short.cntc               	   C   sŒ   t jtdd�� tttdddgd�} W 5 Q R X t| jƒdks@t‚t	| j
dg d	�}t|ƒd
ks`t‚| j
d ddgksvt‚| j
d dksˆt‚dS )zTest reading raw cnt files.znumber of bytes©ÚmatchÚautoZNA1ZLEFT_EAR)Zinput_fnameÚeogÚmiscé   T)r   Úexcludeé   ZbadsZVEOGRZ	meas_dateN)ÚpytestÚwarnsÚRuntimeWarningr   r   ÚfnameÚlenÚannotationsÚAssertionErrorr   Úinfo)ÚrawZeog_chs© r   úX/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/cnt/tests/test_cnt.pyÚ	test_data   s     ÿr   c               	   C   s¬   t jtdd�� ttƒ} W 5 Q R X t dddgdddgdddgd	ddgd
ddgg¡}ttƒ}t|ƒdkslt	‚t
|jdd… |dd…df | jd  ƒ d| jd ks¨t	‚dS )z&Test comparing annotations and events.zCould not parse meas dater	   iM  r   é   iò  i€  ém   i	  i¨  r   NéÿÿÿÿZsfreqzSTI 014Zch_names)r   r   r   r   r   ÚnpÚarrayr   r   r   r   Zonsetr   )r   ÚeventsZannotr   r   r   Ú#test_compare_events_and_annotations)   s    ü*r#   )Úos.pathÚpathÚopÚnumpyr    Znumpy.testingr   r   Zmner   Zmne.datasetsr   Zmne.io.tests.test_rawr   Z
mne.io.cntr   Zmne.annotationsr   Z	data_pathÚjoinr   Zrequires_testing_datar   r#   r   r   r   r   Ú<module>   s   
