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Zd dl	m
Z
 e e e¡ddddd¡Ze ed¡Zej d	d
¡dd„ ƒZdS )é    N)Úassert_array_equal)Úannotate_nanz..ÚioZtestsÚdataztest_raw.fifÚ	meas_date)NÚorigc                 C   s$  t j t¡}d}| |¡ | dkr,| d¡ t |j¡ 	¡ r@t
‚t|ƒ}t|ƒdksXt
‚|j|jd kslt
‚d}tj|j|d| d| …f< t|ƒ}t dg¡}|jd r´||j7 }t|j|ƒ t|jt dg¡ƒ t|jt d	g¡ƒ t|jƒdksút
‚|jd |j| fk�st
‚| |¡ dS )
z*Tests automatic NaN annotation generation.éd   Nr   r   é   é   g      ð?é   ZBAD_NAN)Úmner   Zread_raw_fifÚ	raw_fnameZresampleZset_meas_dateÚnpÚisnanÚ_dataÚanyÚAssertionErrorr   ÚlenZ	orig_timeÚinfoÚnanÚarrayZ
first_timer   ÚonsetÚdurationÚdescriptionZch_namesZset_annotations)r   ÚrawZsfreqZ	annot_nanZ
nan_ch_idxr   © r   úh/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/tests/test_annotate_nan.pyÚtest_annotate_nan   s*    



r   )Úos.pathÚpathÚopÚnumpyr   Znumpy.testingr   Zpytestr   Zmne.preprocessingr   ÚjoinÚdirnameÚ__file__Úbase_dirr   ÚmarkZparametrizer   r   r   r   r   Ú<module>   s   