U f{j1 @sddlmZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZddlmZmZddlmZmZmZmZmZmZmZmZmZddd d d gZddd gZd d gZddgZe e!eeeZ"e e!eeeZ#e#e"fD]\Z$dD]Z%e$&e$'e%qe(e$D]4\Z)Z%e%dd krej*e$e)dej+j,ie$e)<qqe-e dZ.e/dZ0e j1j j2ddZ3ej+j,e j4ej+5dej+5dej+6de#ddZ7e j4ej+5dej+5dej+6de#ej+j8e9dd:dkddd d!Z;e j4ej+5dej+5dej+6de"d"d#Ze j4d(d)Z?e j4ej+6d*ed+d,Z@e j4ej+5dej+5dej+6d*eej+6d-d.d/d0gd1d2ZAe j4d3d4ZBe j4ej+5d5ej+5d6d7d8ZCe j4ej+5d5d9d:ZDdS);) nullcontextN)testing)NOINFO_WARNING_create_events)_check_pandas_installed_record_warnings) check_info_fieldsget_data_paths get_raw_data get_epochs get_evokedpandas_not_found_warning_msg get_raw_info check_dataassert_warning_in_record neuromag306CTFCNTBTIeximiaZv7v73TF))rrT)rrFZmarks)Zexpected_warningmatch pymatreader)downloadz)ignore:.*parse meas date.*:RuntimeWarningz)ignore:.*number of bytes.*:RuntimeWarningzcur_system, version, use_infoc Cst|}t|}|r$t|}t}nd}tjft}tj |d|f}|t j ||}W5QRX|j ddddf} |j } t| | |t|||dS)zBTest comparing reading an Evoked object and the FieldTrip version.Nzaveraged_%s.mat)r r rrpytestwarnsno_info_warningospathjoinmneioread_evoked_fieldtripdatarr) cur_systemversionuse_infotest_data_folder_ftZmne_avginfoctx cur_fnameZavg_ftmne_dataft_datar/d/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/fieldtrip/tests/test_fieldtrip.pytest_read_evoked5s   r1ZAZURE_CI_WINDOWSfalsetruezPandas problem on Azure CI)reasonc sftdd}|dk }t|}t|}|r6t|}t} nd}tjft} tj |d|f} |r| t j | |} W5QRXt| j|jstnJt>} t j | |} | jdksttt| |dkrtt| W5QRX|ddddddf} | }t| ||t|| |tjd fdd }|td|tjtd d t j | |W5QRXdS) zBTest comparing reading an Epochs object and the FieldTrip version.F)strictNzepoched_%s.matrcs8|||}d|dkr4t|ddd|dd<|S)NZfsampler%)nprepeat)fnameZvariable_namesZ ignore_fieldsoutread_matr/r0 modify_mats  z$test_read_epochs..modify_matr<Zmultipler)NN)rr r rrrrrrr r!r"r#read_epochs_fieldtrip isinstancemetadataZ DataFrameAssertionErrorrrr rget_datarrrr<setattrRuntimeWarning)r&r'r(Z monkeypatchZpandasZ has_pandasr)Z mne_epochedr*r+r,Z epoched_ftZ warn_recordr-r.r=r/r;r0test_read_epochsTs>     rFc Cst|}t|dd}|r@t|}|dkr8tjtdd}qPt}nd}tjft}tj |d|f}|t j ||}W5QRX|dkr|s|d d d d d dg|dkr|s|dddgt|||t|||dS)zATest comparing reading a raw fiff file and the FieldTrip version.T)Zdrop_extra_chs)rrzcannot be found inr>Nz raw_%s.matrZMzAZMxAZMyaAZMyAZMxaAZMzaArZTRIG2ZTRIG1ZGATE)r r rrrrErrrr r!r"r#read_raw_fieldtripZ drop_channelsrrCr) r&r'r(r)Z raw_fiff_mner*r+r,Z raw_fiff_ftr/r/r0test_raws6      rHc CsFtd}td}tj|d}tttj ||W5QRXdS)zTest for a meaningful exception when the data is a cell array.Z cellarrayz %s_%s.matrz/Loading of data in cell arrays is not supportedr>r\r]r^N) rr r!r rrrJrKr"r$r?r#rG)r'r[r9r*r/r/r0!test_throw_exception_on_cellarrays  r_c Csd}t|}t|}|ddd=|ttJtjt j |d|t t j |d|t t j |d|W5QRXdS) z6Test _create_info when channels are missing from info.rZchsrz raw_v7.matzaveraged_v7.matrIN)r rZ_update_redundantrrrEr"r#rGrr r!r$r?)r&r)r*r/r/r0test_with_missing_channels s$    raz0ignore: Importing FieldTrip data without an infoz%ignore: Cannot guess the correct typec Cs>tjtdd}tjtddtjj |ddW5QRXdS)zLoading data with non-uniform times per epoch is not supportedr>NrX rr r!rYrrJrKr"r#r?r9r/r/r0*test_throw_error_on_non_uniform_time_fieldsrdc Cs>tjtdd}tjtddtjj |ddW5QRXdS)zETest if an error is thrown if the data was saved with an old version.rWzold_version.matzThis file was created with an old version of FieldTrip. You can convert the data to the new version by loading it into FieldTrip and applying ft_selectdata with an empty cfg structure on it. Otherwise you can supply the Info field.r>NrXrbrcr/r/r03test_throw_error_when_importing_old_ft_version_data*s re)E contextlibrrR itertoolsos.pathrrnumpyr7r"Z mne.datasetsrZmne.io.fieldtrip.utilsrrZ mne.utilsrrZmne.io.fieldtrip.tests.helpersrr r r r r rrrZall_systems_rawZall_systems_epochsZ all_versionsr(listproductZall_test_params_rawZall_test_params_epochsobjkeypopindex enumerateZkiparammarkZslowtestrErZ importorskiprZdatasetsZ data_pathrYZrequires_testing_datafilterwarningsZ parametrizer1ZskipifgetenvlowerrFrHrMrPrVrZr_rardrer/r/r/r0s  ,    "     .   %