U f{j<@s&dZddlmZddlmZmZddlZddlZddlm Z ddl m Z ddl Z ddl Z ddlZddlmZddlZddlmZmZd d lmZmZd d lmZd d lmZdVddZGdddeZddZ ddZ!dWddZ"dXddZ#dZ$dZ%dZ&ee#dd Z'ee#d!d Z(ee#d"d Z)ee#d#e$d$Z*d%d&Z+ee#d'e&d$Z,ee#d(d)d$Z-ee#d*d Z.ee#d+d Z/d,d-Z0d.d/Z1Gd0d1d1e2Z3Gd2d3d3e2Z4d4d5Z5d6d7Z6d8d9Z7d:d;Z8dYdd?Z:d@dAZ;dBdCZdJdKZ?dLdMZ@dNdOZAd\dPdQZBedRdSZCd]dTdUZDdS)^zTesting functions.)contextmanager)partialwrapsN)StringIO)rmtree)SkipTest)assert_array_equalassert_allclose)warnClosingStringIO) check_version) object_diff> cCs\t\}}}tt|||}d|dt||}d|d||}|S)zExplain an exception. z: )sysexc_info traceback format_list extract_tbjoinsplitformat_exception_only)startstopprefixetypevaluetbstringr"Q/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/utils/_testing.py_explain_exceptions r$c@s(eZdZdZddZddZddZdS) _TempDiraCreate and auto-destroy temp dir. This is designed to be used with testing modules. Instances should be defined inside test functions. Instances defined at module level can not guarantee proper destruction of the temporary directory. When used at module level, the current use of the __del__() method for cleanup can fail because the rmtree function may be cleaned up before this object (an alternative could be using the atexit module instead). cCst|tjdd}|S)NZtmp_mne_tempdir_)r)str__new__tempfilemkdtemp)selfnewr"r"r#r'1sz_TempDir.__new__cCs||_dSN)__str___pathr*r"r"r#__init__5sz_TempDir.__init__cCst|jdddS)NT) ignore_errors)rr.r/r"r"r#__del__8sz_TempDir.__del__N)__name__ __module__ __qualname____doc__r'r0r2r"r"r"r#r%%s r%cCs ttddS)zEWrap to requires_module with a function call (fewer lines to change).nibabelname)rrequires_moduler"r"r"r#requires_nibabel<sr;c Csddl}z@ddlm}m}m}m}ddlm}ddlm }ddl m }Wn.t k rv}zd} t |} W5d}~XYn Xd} d } |jj| d | d S) zCheck for dipy.rN)imaffineimwarpmetrics transforms)reslice) AffineMap)DiffeomorphicMapFrTzRequires dipy >= 0.10.1, got: reason)pytestZ dipy.alignr<r=r>r?Zdipy.align.reslicer@Zdipy.align.imaffinerAZdipy.align.imwarprB Exceptionr&markskipif) rEr<r=r>r?r@rArBexcZhavewhyr"r"r# requires_dipyAs  rK0.0cCs@ddl}d|}|dkr(|d|7}|jjt|| |dS)zCheck for a library version.rNz Requires rLz version >= rC)rErGrHr )Zlibrary min_versionrErDr"r"r#requires_versionYs rNc Csddl}|dkrd|n|}d|j|f}zt|tktfWnTtk r}z6tt|dkrt|d|kr|d|f7}d}W5d}~XYnXd}|jj ||d |S) z4Skip a test if package is not available (decorator).rNz import %szTest %s skipped, requires %s.zNo module named %sz Got exception (%s)TFrC) rEr3execglobalslocalsrFlenr&rGrH)functionr9callrErDrIskipr"r"r#r:cs r:z+ if not has_mne_c(): raise ImportError z0 if not has_freesurfer(): raise ImportError z@ if 'NEUROMAG2FT_ROOT' not in os.environ: raise ImportError Zpandasr8ZpylslZsklearnzMNE-Cr9rTcCs:t|tr(d|f}ttd|f|dSt|dtdSdS)zRequire Freesurfer.z@ from . import run_subprocess run_subprocess([%r, '--version']) zFreesurfer (%s)rVZ FreesurferN) isinstancer&rr:_fs_call)argrTr"r"r#requires_freesurfers rZZ neuromag2ftzgood network connectionzJif int(os.environ.get("MNE_SKIP_NETWORK_TESTS", 0)): raise ImportErrorZnitimeZh5pycCstdd|S)z"Decorate tests that need numpydoc.Znumpydocz1.0)rN)funcr"r"r#requires_numpydocsr\cCs*tjj}|dddkr&|ddS)z%Run a given command if it's __main__.r3r__main__runN)inspect currentframef_backf_localsget) local_varsr"r"r#run_command_if_mains rec@s*eZdZdZd ddZddZdd Zd S) ArgvSetterzTemporarily set sys.argv.r"TcCs6td||_|rtntj|_|r*tntj|_dS)N)python)listargvr rstdoutstderr)r*argsZdisable_stdoutZdisable_stderrr"r"r#r0szArgvSetter.__init__cCs4tj|_|jt_tj|_|jt_tj|_|jt_|Sr,)rri orig_argvrj orig_stdoutrk orig_stderrr/r"r"r# __enter__szArgvSetter.__enter__cGs|jt_|jt_|jt_dSr,)rmrrirnrjrorkr*rlr"r"r#__exit__szArgvSetter.__exit__N)r"TTr3r4r5r6r0rprrr"r"r"r#rfs   rfc@s*eZdZdZd ddZddZddZd S) SilenceStdoutzSilence stdout.TcCs ||_dSr,)close)r*rur"r"r#r0szSilenceStdout.__init__cCstj|_tt_tjSr,)rrjrr/r"r"r#rpszSilenceStdout.__enter__cGs|jrtj|jt_dSr,)rurrjrqr"r"r#rrs zSilenceStdout.__exit__N)Trsr"r"r"r#rts rtcCs,z ddl}Wntk r"YdSXdSdS)zuDetermine if nibabel is installed. Returns ------- has : bool True if the user has nibabel. rNFT)r7 ImportError)r7r"r"r# has_nibabels  rwcCs dtjkS)zCheck for MNE-C.ZMNE_ROOTosenvironr"r"r"r# has_mne_csr{cCs dtjkS)zCheck for Freesurfer.ZFREESURFER_HOMErxr"r"r"r#has_freesurfersr|cstfdd}|S)z>Decorate tests that make calls to SVD and intermittently fail.c sZz ||WStjjk rT}z&dt|krBd}t|t|W5d}~XYnXdS)NzSVD did not convergez7Intel MKL SVD convergence error detected, skipping test)nplinalgZ LinAlgErrorr&r r)rlkwargsexpmsgrSr"r#decs  zbuggy_mkl_svd..dec)r)rSrr"rr# buggy_mkl_svds rcCs\ddlm}t||r|j}dD]6}t|jd|kd}t||ksLt| |q dS)z;Assert that there are boundary annotations and remove them.BaseRaw)ZEDGEBADz %s boundaryrN) Zio.baserrW annotationsr}where descriptionrRAssertionErrordelete)rnrkeyidxr"r"r# assert_and_remove_boundary_annots  rcCst||}|dkst|dS)zAssert two objects are equal.rN)rr)abdr"r"r#assert_object_equals rc Csddlm}m}ddlm}ddlm}|ddd}|td|dd }|dk rX||}|j j d d ||j d <W5QRX|d gdgdg|}|j |d|S)Nr) Annotations create_info)_handle_meas_date)RawArray g$@)ch_namesZsfreq)rr)datainfoZ first_sampT)Z check_after meas_dateg?g?dummy)r) rrrrriorr}emptyrZ_unlockZset_annotations) rZ orig_timerrrrrrawZannotr"r"r# _raw_annots    rcCsFddlm}ddlm}t||r.||dSt||rB|j|SdS)z#Get the (n_ch, n_times) data array.r)EvokedrrN)ZevokedrrrrWr)xZch_idxrrr"r"r# _get_data#s      rMEGcCst||}t||}ttj||dd} ||} ttj| | dd} tj| dtj| d| | } | } | |k}|dkrd|n|}|dkstd| ||t ||ft | } | |kstd || ||fd S) z#Check the SNR of a set of channels.r )Zaxisg&y3)outrz (%s)rz.SNR (worst %0.2f) < %0.2f for %s/%s channels%sz%s SNR median %0.2f < %0.2f%sN) rr}sqrtmeanZclipinfminsumrrRZmedian)actualdesiredpicksmin_tolmed_tolrkindZ actual_dataZ desired_dataZ bench_rmserrorZ error_rmsZsnrssnrZ bad_countr"r"r# _check_snr-s(       r@@c Cs8ddlm}||jdgd}||jdgd}t||dd||jddgd} ||jddgd} |d k rtt| | d dttt|jt || g} ttt|jt || g} t| | d dt| d krt t || t || d dddt ||||||dd|d k r4t| d kr4t ||| d||ddd S)zAssert channel SNR of a certain level. Mostly useful for operations like Maxwell filtering that modify MEG channels while leaving EEG and others intact. r) pick_typesT)megexcludezMEG pick mismatch)err_msgF)rZchpirNzcHPI pick mismatchzOther pick mismatchrgdy=h㈵>znon-MEG channel mismatch)atolrtolrr)rgZcHPI) Zio.pickrrrr}Z setdiff1dZarangerRrZ concatenater rr) rrrrZ chpi_med_tolrrrZ picks_desiredZchpisZ chpis_desiredZothersZothers_desiredr"r"r#assert_meg_snrCs6    rc Cs`ddlm}tjdd&|j|dd|j||dd}W5QRX||ks\td||fdS) z?Assert actual and desired arrays are within some SNR tolerance.r)r~ignore)divideZfro)ordz%f < %fN)Zscipyr~r}ZerrstateZnormr)rrZtolr~rr"r"r# assert_snras   rcCsht|j|jt|j|jt|jd|jdt|jd|jdt|j|jt|j|jdS)zCheck that two STC are equal.rr N)r timesrrZverticesZtminZtstep)Zstc1Zstc2r"r"r#assert_stcs_equaljs rcCs|d|dfS)zSort dig keys.ridentr")digr"r"r# _dig_sort_keytsrc sddlm}ddlmddlm}ddlm}||}}t||rft||sRt |d|d}}n2t||stt t||st |j |j }}d}}t|t st t|t st t |t d}t |t d}t|t|kst tt|d||d|D]`\}\} } d D] } | | | | kst | qt| d | d d d d || d | d fd qtfdd|Dr|dk r||ddd\} } }||ddd\}}}t|| ddt||d ddt|| d dddS)zAssert dig allclose.r)fit_sphere_to_headshapeFIFF)Info) DigMontagerN)r)rrZ coord_framerrzFailure on %s: %s %s)rrrc3s|]}|djkVqdS)rN)ZFIFFV_POINT_EXTRA).0rrr"r# sz&assert_dig_allclose..mr)Zunitsverbosegư>)r)rr)ZbemrZ io.constantsrZ io.meas_inforZchannels.montagerrWrrrhsortedrrR enumeratezipr any)Zinfo_pyZinfo_binlimitrrrZdig_pyZdig_biniiZd_pyZd_binrZr_binZ o_head_binZ o_dev_binZr_pyZ o_head_pyZo_dev_pyr"rr#assert_dig_allcloseysR        *   rc kstdtt}|D]J\}}t|||<|dk rPt|tsDt|tj |<q|tj krtj |=qz dVW5|D].\}}|dk r|tj |<qx|tj krxtj |=qxXdS)zUse a modified os.environ with temporarily replaced key/value pairs. Parameters ---------- **kwargs : dict The key/value pairs of environment variables to replace. zmodified_env is deprecated and will be removed in 1.1. In tests, use monkeypatch from pytest instead. In subprocess calls, pass modified environments directly.N) r DeprecationWarningdictitemsrygetenvrWr&rrz)rZorig_envrvalr"r"r# modified_envs$       rcCs`ddlm}|j|jj|}|}|j }|j }|||jj||fd|ddS)z>Click on a channel name in a raw/epochs/ICA browse-style plot.r) _fake_clickZpix)ZxformbuttonN) Z viz.utilsrZcanvasZdrawZmneZax_mainZget_yticklabelsZget_window_extentZ intervalxrZ intervaly)ZfigZch_indexrrtextZbboxryr"r"r#_click_ch_names    r)rNr)rL)N)r )r)rrN)N)rr )Er6 contextlibr functoolsrrryr_rrshutilrrr(rZunittestrnumpyr}Z numpy.testingrr Z_loggingr r checkr Znumericsrr$r&r%r;rKrNr:Z _mne_callrXZ _n2ft_callZrequires_pandasZrequires_pylslZrequires_sklearnZ requires_mnerZZrequires_neuromag2ftZrequires_good_networkZrequires_nitimeZ requires_h5pyr\reobjectrfrtrwr{r|rrrrrrrrrrrrrr"r"r"r#s|                   &