U
    f»{ju  ã                   @   sn   d dl Zd dlZd dlmZmZmZ d dlmZ d dl	m
Z
 ejdd�Zed d Zed	 Zejd
d„ ƒZdS )é    N)Úread_evokedsÚ
pick_typesÚcompute_proj_evoked)Útesting)Úplot_projs_jointF)ÚdownloadZMEGÚsamplezsample_audvis-ave.fifc               	   C   s  t tƒd  d¡} g | jd< d\}}}d\}}}t t| jdd�d|… t| jd	d
d�d|… t| jdd�d|… g¡}|  |¡ t| j	ƒ|| | ksœt
‚| jd
d�dddgks¶t
‚t| |||d�}t|ƒdksÖt
‚tjtdd�� |  dd¡ d¡ W 5 Q R X tddd	d�}	t|| |	dd�}
|
jd  ¡ }| d¡�sBt
|ƒ‚|
jd  ¡ }| d¡�sdt
|ƒ‚|
jd  ¡ }| d¡�s†t
|ƒ‚d}|
j| }| ¡ dk�s¦t
‚t|jƒ|d |  k�sÂt
‚t|jƒ}t|
jƒd!k�sàt
‚t|| d"|	dd#�}
t|
j| jƒ|d$ k�st
‚dS )%zTest plot_projs_joint.r   )Nr   Zbads)é	   é
   é   )é   r   é   Zgrad)ÚmegNFT)r   ÚeegÚmag)Úuniquer   )Ún_magÚn_gradÚn_eegé   Zaliasing)Úmatchgš™™™™™¹¿gš™™™™™¹?r
   é   )ÚresZcontoursZsensorsÚerror)Útopomap_kwargsÚverboseZGradé   ZEEGé   ZMagÚ r   r   zMEG 0111)Zpicks_tracer   r   r   )r   Úevoked_fnameZapply_baselineÚinfoÚnpZconcatenater   ÚpickÚlenZch_namesÚAssertionErrorZget_channel_typesr   ÚpytestZwarnsÚRuntimeWarningÚcropZdecimateÚdictr   ZaxesZ
get_ylabelÚ
startswithÚlines)Zevokedr   r   r   Z
n_mag_projZn_grad_projZ
n_eeg_projZpicksZprojsr   ZfigZylabZmag_trace_ax_idxZmag_trace_axZold_len© r+   úV/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/viz/tests/test_proj.pyÚtest_plot_projs_joint   s`    


ý
   ÿ   ÿ

   þr-   )Únumpyr!   r%   Zmner   r   r   Zmne.datasetsr   Zmne.vizr   Z	data_pathÚdata_dirZ
sample_dirr   Zrequires_testing_datar-   r+   r+   r+   r,   Ú<module>   s   