U
    f»{j¢  ã                   @   s¢   d dl Z d dlmZ d dlmZ d dl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mZ d dlmZ ejdd	�Ze ed
¡Zejjejdd„ ƒƒZdS )é    N)Úcopyfile)Úassert_allclose)Úsparse)Útesting)Úcatch_loggingÚ_record_warnings)Úread_morph_mapF)ÚdownloadZsubjectsc              
   C   sš  t | ƒ}dD ]‚}t t ||¡¡ t t ||d¡¡ |dkrBdnd}dD ]B}|D ]8}|d|d | g}ttjtf|žŽ tj|f|žŽ ƒ qRqJqdD ]ª\}}}	tƒ �}
t||||	d	d
�}W 5 Q R X |
 	¡ }
d|
ksÖt
‚d|
ksât
‚t||t|	d�}t|ƒt|ƒk�st
‚t||ƒD ]*\}}|| j}t|t |¡ddd� �qq”tƒ � tdd|d�}W 5 Q R X |D ]0}|t |jd |jd ¡  ¡ dk�sdt
‚�qddS )z%Test reading and creating morph maps.)ÚsampleÚ	sample_dsÚfsaverage_dsZsurfr   )ÚregZ
left_right)r   )ZlhZrhz.sphere.))r   r   F)r   r   TT)ÚxhemiÚverbosezdoes not existZCreating)r   gü©ñÒMbP?r   )ZatolZrtolr
   )Úsubjects_dirN)ÚstrÚosÚmkdirÚopÚjoinr   r   r   r   ÚgetvalueÚAssertionErrorÚlenÚzipÚdatar   ÚnpZ
zeros_liker   r   ÚeyeÚshapeÚsum)Ztmp_pathÚtempdirÚsubjectÚregsZhemir   ÚargsZsubject_fromZ
subject_tor   ÚlogÚmmapZmmap2Úm1Úm2ÚdiffÚmm© r)   úW/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/tests/test_morph_map.pyÚtest_make_morph_maps   s>    ÿ
 ÿÿ
r+   )r   Úos.pathÚpathr   Úshutilr   ZpytestÚnumpyr   Znumpy.testingr   Zscipyr   Zmne.datasetsr   Z	mne.utilsr   r   Zmner   Z	data_pathr   r   ÚmarkZslowtestZrequires_testing_datar+   r)   r)   r)   r*   Ú<module>   s   