U f{j<@sZddlmZddlZddlmZddlmZmZddl Z ddl m Z m Z mZmZddlZddlZddlmZddlmZmZddlmZddlmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)dd lm*Z*dd l+m,Z,ddl-m.Z.m/Z/m0Z0e1e2e3e4dZ5dZ6dde6DZ7dde6DZ8dde6DZ9dde6DZ:e1e5dZ;e1eje1eje?fejGOd6d7d8d9ZPejAd:d;ZQdS)<)BytesION)reducepartial)assert_array_almost_equalassert_array_equalassert_allclose assert_equal)testing) read_raw_fif read_raw_bti)_make_bti_dig_points) _read_config_read_head_shape_read_bti_header_get_bti_dev_t_correct_trans _get_bti_info_loc_to_coil_trans_convert_coil_trans_check_nan_dev_head_t_rename_channels)_test_raw_reader) pick_info)FIFF) pick_types)assert_dig_allclose) Transformcombine_transformsinvert_transformdata)linuxsolariscCsg|]}ttd|qS)z test_pdf_%sopjoinbase_dir.0ar)X/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/bti/tests/test_bti.py "sr+cCsg|]}ttd|qS)ztest_config_%sr"r&r)r)r*r+#scCsg|]}ttd|qS)z test_hs_%sr"r&r)r)r*r+$scCsg|]}ttd|qS)zexported4D_%s_raw.fifr"r&r)r)r*r+%sz tmp_raw.fifF)downloadZBTiZerm_HFHzc,rfDCZ4Dsimz c,rfDC,fn50,ocCstttdddS)z#Test reading data from 2500 system.N) pdf_fnamehead_shape_fname)rr fname_2500r)r)r)r*test_read_25004sr1cs~dfdd}|tjjjd|ttdtdtdddd}tf|}|j }t |j d|d t d t jd S) z:Test that we don't set loc to None when no trans is found.ZMLzAcs8t||}dd|dD}|d|dd=|S)NcSsg|] }|dqS)namer)r'chr)r)r*r+@sz>test_no_loc_none.._read_config_bad..chsdevZ transform)r index)argskwargscfgidxch_namer)r*_read_config_bad>s z*test_no_loc_none.._read_config_badr rF)r. config_fnamer/rename_channelssort_by_ch_namer5loc N)setattrmneioZbtidict pdf_fnames config_fnames hs_fnamesr ch_namesr7rinfonpfullnan)Z monkeypatchr>r9rawr;r)r<r*test_no_loc_none:s   rQcCs0tD]&}t|}tdd|dDstqdS)zTest read bti config file.css"|]}d|ko|dkVqdS)unknownN)lower)r'blockr)r)r* Rsz#test_read_config..Z user_blocksN)rIr allAssertionError)configr:r)r)r*test_read_configMs rZc Cstttdtdtdd}|dd\}}d|dd|d}}||k||k}|||}|dd\}}|jd|kst |jd|jdkst dS)zTest crop and append raw.r)r.r?r/Ng?g?) rr rHrIrJcopycropshapesumrX) rPytt0t1maskZraw_Zy__r)r)r*test_crop_appendVsrgc Csd}tddtd|}ttttD]|\}}}t|||dd}|jd}|jd}|jd }tddtd|}tt ||dd } t| |dd } t | d |d q"d S)zTest transformations.g{Gz?g)\(?ctf_megmegriFpreload dev_ctf_t dev_head_t ctf_head_tctf_headheadtransN) rrziprHrIrJr rLrrr) Z bti_trans bti_dev_tpdfrYhsrPrnZdev_head_t_oldrprbZdev_head_t_newr)r)r*test_transformsds    rxc CstttttD]\}}}}tjtt|dddtjtt||dddt t rZt t t|dd}t|||dd}dt|kstt|jdt|jdtt|jdd |jdd d t|jd d kstt|j|jd ddd||fD\}}t||d dd||fD\}} t|| t|dtd|dtdtdd|jddtDdd|jddtDtdd|jddtDdd|jddtDt|jdt|jdtdD]V} |j| dkrn>|j| dk stdD]$} t|j| | |j| | qq|t tt } t| dD]N} t| j| tsvt| j| d } t| jdt !| t "dr\tq\t t qdS)z"Test bti conversion to Raw object.eggsFrlZspamTZRawBTiNrorsdig)i i"d)limitcSs.g|]&}tdd|jddtDqS)cSsg|]}|dqSrBflattenr'dr)r)r*r+s'test_raw...r5NrMZ concatenaterLNCHr'Zr_r)r)r*r+s ztest_raw..cSs.g|]&}tdd|jddtDqS)cSsg|]}|dqSr~rrr)r)r*r+srr5Nrrr)r)r*r+s rcSsg|] }|dqSranger)r'cr)r)r*r+sr5cSsg|] }|dqSrr)rr)r)r*r+scSsg|] }|dqScalr)rr)r)r*r+scSsg|] }|dqSrr)rr)r)r*r+s)rornrp)tofromrs)rr)#rtrHrIrJexported_fnamespytestZraises ValueErrorr r#exists tmp_raw_fnameosremover reprrXrrKrrrLlenrrrZ_calssaveprint isinstancerGr_rMallcloseeye)rvrYrwZexportedexraZcoil1Zcoil2Zloc1Zloc2keyZentreZthis_tr)r)r*test_rawzsh        "   rc sXttttD]\}}}t|||dddddddd \}}td||dddddddd \}}t|dd d |d Dtd d |dDdd dddddgtdd |dDdddddddgt|t|dddd}t|t|dddd}|ddk st|ddk st|ddk s*t|ddk s.r5cSsg|]}|dr|qSA startswithr'nr)r)r*r+s A22A2ZA104ZA241ZA138cSsg|]}|dr|qSrrrr)r)r*r+s ZA133ZA158ZA44ZA134ZA216T)rkZstimrespZsfreqZlowpassZhighpassZ meas_date)rnrorprscSsg|] }|dqSr~r)r3r)r)r*r+scSsg|] }|dqSr~r)r3r)r)r*r+s)r.r?r/rArm)r.r?r/r@rArmcsg|]}j|qSr)) bti_ch_labelsr7r3Zraw1r)r*r+s)rtrHrIrJrrrrrXrrMarrayr r_datarK) rvrYrwrLbti_infoZinfo2rraw2Zsort_idxr)rr*%test_info_no_rename_no_reorder_no_pdfs       rc Csrttdddddddd}ttttD]F\}}}||||dd\}}t|||ddd j}t|t |ddd dd t|t |ddd dd t ||}t |d d }t ||ddt |ddt dt |ddt dt|\} } } } } t| | | | | ddd\}}}t |dt dtt||d|dD]^\}\}}}t|d|dt |d|dt |d|drt|dkr>qq>dd|dD}t|dD]\}}|dsԐq||}|d|d}|d|d}t||ddt ||rt|dd }|dd!}t|d|d"tjt|d|d"tjqq$d#S)$zTest bti no-conversion option.rirhFrr)rrrrrr@rA)rT)r.r?r/rrm)rkZref_meg)r] bti_transformrrnrsrorrp)rZuse_hpir{identr cSsi|]}|d|dqS) chan_labelrBr)r3r)r)r* sz&test_no_conversion..r5rKrrBgV瞯<)ZatolzMEG 001rZ coord_frameN)rrrtrHrIrJr rLrrrrrrMrrr enumeraterrrXrrr7rZFIFFV_COORD_DEVICEZFIFFV_MNE_COORD_4D_HEAD)get_inforvrYrwZraw_inforfZ raw_info_conrrnnasionlparpahpi dig_pointsr{rbiioldnewconZch_mapZch_labelrdt2Zt3Zidx_aZidx_br)r)r*test_no_conversions        rc CsttttD]\}}}t|||ddd}t|d}t|}W5QRXt|d}t|}W5QRXt|d}t|}W5QRXt|||ddd}t|t |ddd|dddq dS)zTest bti bytes-io API.TF)rrmrbNr) rtrHrIrJr openrreadrr)rvrYrwrPZfidrr)r)r* test_bytes_io-s   rc CshtD]^}t|\}}}}}t|||||\}}}dddh} ttdddd|D} | | rtqdS) zTest reading bti headshape.kindrrcSst|t|S)N)list)xrar)r)r*Ez&test_setup_headshape..cSsg|] }|qSr))keysrr)r)r*r+Fsz(test_setup_headshape..N)rJrr setrrX) rwrrrrrr{rbrfexpectedfoundr)r)r*test_setup_headshape>s   rcCs0ttD] \}}t|t|dd}tddt|dd}d}d}d}t||}tdd |}d }d } t} |d D],} | d } | ds| d| } | | qtt | || d} t | | }t j|ddddf<t|t|D]H\}\}}|d |d}|dk r|rt|d |d}t|||}qqdS)z>Test unlikely case that the device to head transform is empty.T)rArjrqrrrirhrkrrr5r2rr)rrrsNrB)rrHrrIrrrrrgetappendrrtrMrOrrr)rr.rrnrrrrurrZ bti_ch_namesr4r=Zneuromag_ch_namesZ ch_mappingr;Zchan_4dZ chan_neuromagrBrbr)r)r*test_nan_transJsH       rfnamerm)TFcCst||ddS)zTest for gh-6048.rlN)r )rrmr)r)r*test_bti_ch_datassrcCs*ttddd}ttt|jddddS)z.Check that EOG channels can be set (gh-10092).F)ZX65ZX67ZX69ZX66ZX68)rmrT)ZeogrN)r fname_simrrrrL)rPr)r)r*test_bti_set_eog{s r)RrFrros.pathpathr# functoolsrrnumpyrMZ numpy.testingrrrrrrEZ mne.datasetsr Zmne.ior r Zmne.io._digitizationr Zmne.io.bti.btir rrrrrrrrrZmne.io.tests.test_rawrZ mne.io.pickrZmne.io.constantsrrZ mne.utilsrZmne.transformsrrrr$abspathdirname__file__r%archsrHrIrJrrZ data_pathr0rZfname_sim_filtrZrequires_testing_datar1rQrZrgrxmarkZslowtestrrrrrrZ parametrizerrr)r)r)r*sn    0         9=< )