Aluode/PerceptionLabPortable
0
1o
2 '�Yi�6 � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl3m Z ddl Z
ddlmZmZ ddlmZ dd lmZmZ dd4lmZ dd lmZ dd lmZ dGdd�Z G dd� de �Z dd� Z dd� Z dd� Z!dd� Z"dd � Z#d d � Z$G d d � d �Z%d!d"� Z&d#d$� Z'd%d&� Z(dHd'd(�Z)d)d*d+�d,d-�Z*d.d/� Z+d0d1� Z,dId3d4�Z- dJd6d7�Z.d8d9� Z/d:d;� Z0d<d=� Z1dKd>d?�Z2dLd@dA�Z3dBdC� Z4dMdEdF�Z5dS )NzTesting functions.� N��wraps)�rmtree)�SkipTest)�assert_allclose�assert_array_equal)�linalg� )�ClosingStringIO�warn)�
check_version)�run_subprocess�� object_diff������> c C s\ t �� \}}}t�t�|�| |� �}d�|��d�t�||� }d| d| �|� }|S )zExplain an exception.� �5�:6)�sys�exc_info� traceback� format_list�7extract_tb�join�split�format_exception_only)�start�stop�prefix�etype�value�tb�string� r$ �|E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\mne/utils/_testing.py�_explain_exception s8 r& c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_TempDira� Create and auto-destroy temp dir.9 10 This is designed to be used with testing modules. Instances should be11 defined inside test functions. Instances defined at module level can not12 guarantee proper destruction of the temporary directory.13 14 When used at module level, the current use of the __del__() method for15 cleanup can fail because the rmtree function may be cleaned up before this16 object (an alternative could be using the atexit module instead).17 c C s t �| tjdd��}|S )NZtmp_mne_tempdir_)r )�str�__new__�tempfile�mkdtemp)�self�newr$ r$ r% r) 0 s z_TempDir.__new__c C s | � � | _d S �N)�__str__�_path�r, r$ r$ r% �__init__4 s z_TempDir.__init__c C s t | jdd� d S )NT)�
ignore_errors)r r0 r1 r$ r$ r% �__del__7 s z_TempDir.__del__N)�__name__�18__module__� __qualname__�__doc__r) r2 r4 r$ r$ r$ r% r' $ s19 r' c C s20 t � | �S )z%Decorate a function as requiring MNE.)�requires_mne_mark)�funcr$ r$ r% � requires_mne; � 21r; c C s ddl } | jjt� dd�S )z%Mark pytest tests that require MNE-C.r NzRequires MNE-C��reason)�pytest�mark�skipif� has_mne_c�r? r$ r$ r% r9 @ s r9 c C s ddl } | jjtdd� dd�S )z(Mark pytest tests that require OpenMEEG.r NZopenmeegz2.5.6zRequires OpenMEEG >= 2.5.6r= )r? r@ rA r rC r$ r$ r% �requires_openmeeg_markG s �rD c C sx ddl }d}t| t�r0|d| � �7 }zt| dg� W n ty% d}Y nw d}|jj||d�S |jjt� dd�| �S ) zRequire Freesurfer.r NzRequires Freesurferz22 command: z --versionTFr= )r? �23isinstancer( r
� Exceptionr@ rA �has_freesurfer)�argr? r> �skipr$ r$ r% �requires_freesurferP s 24 ��rJ c C s* dd l }|jjttj�dd��dd�| �S )Nr ZMNE_SKIP_NETWORK_TESTSz MNE_SKIP_NETWORK_TESTS is setr= )r? r@ rA �int�os�environ�get)r: r? r$ r$ r% �requires_good_networkh s ��rO c C s. t �� jj} | �dd�dkr| d � dS dS )z%Run a given command if it's __main__.r5 r �__main__�runN)�inspect� currentframe�f_back�f_localsrN )�25local_varsr$ r$ r% �run_command_if_mainq s �rW c @ s* e Zd ZdZd dd�Zdd� Zdd � Zd26S ) �27ArgvSetterzTemporarily set sys.argv.r$ Tc C s< t d| �| _|r t� ntj| _|rt� | _d S tj| _d S )N)�python)�list�argvr28 r �stdout�stderr)r, �argsZdisable_stdoutZdisable_stderrr$ r$ r% r2 { s zArgvSetter.__init__c C s4 t j| _| jt _t j| _| jt _t j| _| jt _| S r. )r r[ � orig_argvr\ � orig_stdoutr] � orig_stderrr1 r$ r$ r% � __enter__� s zArgvSetter.__enter__c G s | j t_| jt_| jt_d S r. )r_ r r[ r` r\ ra r] )r, r^ r$ r$ r% �__exit__� s zArgvSetter.__exit__N)r$ TT)r5 r6 r7 r8 r2 rb rc r$ r$ r$ r% rX x s29 30 rX c C �31 dt jv S )zCheck for MNE-C.ZMNE_ROOT�rL rM r$ r$ r$ r% rB � r< rB c C rd )zCheck for Freesurfer.ZFREESURFER_HOMEre r$ r$ r$ r% rG � r< rG c s t � �� fdd��}|S )z>Decorate tests that make calls to SVD and intermittently fail.c 32 sP z� | i |��W S t jjy' } zdt|�v r"d}t|� t|��� d }~ww )NzSVD did not convergez7Intel MKL SVD convergence error detected, skipping test)�npr Z LinAlgErrorr( r r )r^ �kwargs�exp�msg��functionr$ r% �dec� s ��zbuggy_mkl_svd.<locals>.decr )rk rl r$ rj r% �
buggy_mkl_svd� s 33rm c C s| d}ddl m} t| |�r| j} dD ])}t�| j|� d�k�d }t|�|ks6J dt|�� d|� d |� ���| �|� qd34S ) z;Assert that there are boundary annotations and remove them.T� ��BaseRaw)ZEDGE�BADz boundaryr zGot z 'z! boundary' annotations, expected N) �iorp rE � annotationsrf �where� description�len�delete)rs �n�__tracebackhide__rp �key�idxr$ r$ r% � assert_and_remove_boundary_annot� s 35� �r| zObject mismatchF)�err_msg�allclosec C s0 d}t | ||d�}|dksJ |� d|� ���dS )z Assert two objects are equal.T)r~ r r Nr )�a�br} r~ ry �dr$ r$ r% �assert_object_equal� s r� c C s� ddl m} ddlm}m} ddlm} |ddd�}|t�d�|dd �}| d ur,|| �} |j j36d37d ��
| |j d < W d � n1 sCw Y |d
gdgdg|�}|j |d� |S )Nrn )� create_info)� Annotations�_handle_meas_date)�RawArray�38 g $@)�ch_namesZsfreq)r� r� )�data�infoZ39first_sampT)Z check_after� meas_dateg �?g�������?�dummy)rs ) �_fiff.meas_infor� rs r� r� rr r� rf �emptyr� Z_unlockZset_annotations) r� Z orig_timer� r� r� r� r� �rawZannotr$ r$ r% �40_raw_annot� s � r� c C sF ddl m} ddlm} t| |�r| | d S t| |�r!| j| S dS )z#Get the (n_ch, n_times) data array.rn )�Evokedro r N)Zevokedr� rr rp rE r� )�xZch_idxr� rp r$ r$ r% � _get_data� s 41 4243�r� �MEGc C s d}t | |�}t ||�} t�tj| | dd��}44|| } t�tj| | dd��} tj| dtj| d� |45| }
|
�� }|
|k �� }|dkrJd|� d�n|}|d ksgJ d46|d �d |d �d
|� dt|�� d|� �47��t� |
�}||ks�J |� d|d �d|d �|� ���dS )z#Check the SNR of a set of channels.Tr )Zaxisg&����y3)�outr z (�)r z SNR (worst z0.2fz) < z for �/z channelsz SNR median z < N)48r� rf �sqrt�meanZclip�inf�min�sumrv Zmedian)�actual�desired�picks�min_tol�med_tolri �kindry Z actual_dataZ desired_dataZ bench_rms�errorZ error_rmsZsnrs�snrZ bad_countr$ r$ r% �49_check_snr� s, 5051 52����53,r� � @@c C s@ d}ddl m} ||jdg d�}||jdg d�} t|| dd� || jddg d�}54||jddg d�} |d ur<t|55| d56d� t�t�t| j��t� ||57g��} t�t�t|j��t� | | g��}
t| |
d d� t| �d krzt58t | | �t || �d
ddd� t | |||||dd� |d ur�t|59�d kr�t | ||60d||dd� d S d S d S )z�Assert channel SNR of a certain level.61 62 Mostly useful for operations like Maxwell filtering that modify63 MEG channels while leaving EEG and others intact.64 Trn )�65pick_types)�meg�excludezMEG pick mismatch�r} F)r� Zchpir� NzcHPI pick mismatchzOther pick mismatchr g�dy���=��h㈵��>znon-MEG channel mismatch)�atol�rtolr} r� )r� � ZcHPI)
Z66_fiff.pickr� r� r rf Z setdiff1dZarangerv r� Z concatenater r� r� )r� r� r� r� Z chpi_med_tolri ry r� r� Z
picks_desiredZchpisZ
chpis_desiredZothersZothers_desiredr$ r$ r% �assert_meg_snr� s8 �� ��r� c C sp d}t jdd�� tj|dd�tj|| dd� }W d � n1 s#w Y ||ks6J d|�d|����dS ) z?Assert actual and desired arrays are within some SNR tolerance.T�ignore)�divideZfro)�ordNzsnr=z < tol=)rf Zerrstater Znorm)r� r� Ztolry r� r$ r$ r% �67assert_snr s68 " � r� c C s� d}t | j|jdd� t | j|jdd� t| jd |jd dd� t| jd |jd dd� t | j|jd d� t | j|jd69d� d S ) z Check that two STC are equal.TzTimes mismatchr� z
Data mismatchr zLeft vertices mismatchr zRight vertices mismatchz
tmin mismatchztstep mismatchN)r �timesr� r ZverticesZtminZtstep)Zstc1Zstc2ry r$ r$ r% �assert_stcs_equal% s ��r� c C s | d | d fS )zSort dig keys.r� �identr$ )�digr$ r$ r% �
_dig_sort_key9 s r� c
s d}ddl m� ddlm} ddlm} ddlm} | |}}t||�r7t||�s-J d��|d |d }}n t||�s@J d��t||�sIJ d��|j |j }}d } }t|t70�s]J d71��t|t72�sfJ d ��t |t d �}t |t d �}t
|�t
|�ks~J d
��tt|d |� |d |� ��D ]7\} \}73} dD ]} |74| | | ks�J d| �d| � ���q�t|75d | d ddd| � d|76d � d| d � �d� q�t� fdd�|D ���r| d u�r||ddd�\}
}}|| ddd�\}}}t||
d d d � t||dd d d� t||dd d d� d S d S d S )!zAssert dig allclose.Trn ��FIFF)�Info)�fit_sphere_to_headshape)�77DigMontagez Both must be Info or DigMontager� Nzdig_py must be a listzdig_bin must be a list)rz z Different number of dig points)r� r� Z coord_framezkey=z mismatch on point �rr� z Failure on r r )r� r� r} c 3 s � | ]78}|d � j kV qdS )r� N)ZFIFFV_POINT_EXTRA)�.0r� r� r$ r% � <genexpr>` s � z&assert_dig_allclose.<locals>.<genexpr>�mr� )Zunits�verboseg�����ư>zSphere radius mismatch)r� r} z Sphere device origin mismatchzSphere origin mismatch)Z_fiff.constantsr� r� r� Zbemr� Zchannels.montager� rE r� rZ �sortedr� rv � enumerate�zipr �any)Zinfo_pyZinfo_bin�limitry r� r� r� Zdig_pyZdig_bin�iiZd_pyZd_binrz Zr_binZ79o_head_binZ o_dev_binZr_pyZ o_head_pyZo_dev_pyr$ r� r% �assert_dig_allclose> sf 8081 *& �" � ��82��r� c C s` ddl m} | j�� | jj�� | }|�� }|j� � }|j83� � }|| | jj||fd|d� dS )z>Click on a channel name in a raw/epochs/ICA browse-style plot.rn )� _fake_clickZpix)Zxform�buttonN) Z viz.utilsr� ZcanvasZdrawZmneZax_mainZget_yticklabelsZget_window_extentZ intervalxr� Z intervaly)�figZch_indexr� r� �textZbboxr� �yr$ r$ r% �_click_ch_namex s 848586 r� c C s t dd�r | �� S | jd �� S )z/Get fig suptitle (shim for matplotlib < 3.8.0).Z87matplotlibz3.8r )r Z get_suptitleZtextsZget_text)r� r$ r$ r% �
_get_suptitle� s 88r� r� c C s d}ddl m}m} t| |�r8t||�sJ d��| d |d ks$J d��| d |d ks0J d��| d } |d }t| tj�sBJ d89��t|tj�sLJ d ��| jd ksUJ d
��|jd ks^J d��|| |ddd�\}}||ks|J d| d�dd| d�d���||ks�J |d�d|d�d���d S )NTrn )� Transform� angle_distance_between_rigidz!Both must be Transform or ndarray�fromz'from' frame mismatch�toz'to' frame mismatchZtranszactual should be ndarrayzdesired should be ndarray)� r� z actual.shape should be (4, 4)z desired.shape should be (4, 4)�degr� )Z angle_unitsZdistance_unitsi� z0.3fz > z mm translationu ° rotation)Z90transformsr� r� rE rf Zndarray�shape) r� r� Zdist_tolZ angle_tolry r� r� Zangle�distr$ r$ r% �assert_trans_allclose� s& 9192�93�$r� )r Nr )r )r� )r� r� Nr. )r r )r� r� )6r8 rR rL r r* r � functoolsr �shutilr Zunittestr �numpyrf Z
numpy.testingr r Zscipyr Z_loggingr94 r �checkr �miscr
Znumericsr r& r( r' r; r9 rD rJ rO rW rX rB rG rm r| r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r% �<module> sT 9596 9798 99�' 100101: 102