U f{j0D@sddlmZddlZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZddZdd Zejd ed d d ddd fddZddZddZddZddZddZddZddZejjdd Zd!d"Zd#d$Z d%d&Z!e j"d'd(Z#ejje j"d)d*Z$d+d,Z%dS)-N)Epochs create_info EpochsArray)testing)make_fixed_length_events) plot_drop_logcCs(|jdkst||jdks$tdS)z#Test plotting non-preloaded epochs.N)_dataAssertionErrorplot)epochs_unloadedbrowser_backendr X/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/viz/tests/test_epochs.pytest_plot_epochs_not_preloadedsrc CsLt|jdkst|jd|jd<W5QRX|jddd}|d}|dgksXt|t|d d kstt|d gkst|jd gksttj t d d |j|dW5QRX||j dg|jd <tj t d d |j|dW5QRX||j dg|d <tj t d d |j|dW5QRX||d dd|d <|ddddd|d<tj t d d |j|dW5QRX||djdddgdd}| dtj tdd |jgdW5QRXt|dkst|j dg|jd <|tj tdd |jddW5QRX|jddddS)zTest epoch plotting.g$@ZlowpassNr)scalingstitlex2namesinbadsZ projectionmatch)Z noise_covri2data)picksrescapezNo appropriate channels found)rz#title must be None or a string, got)r selection)group_byr)leneventsr infoZ_unlockr _get_ticklabels _close_allpytestwarnsRuntimeWarningch_names_fake_keypressraises ValueError readouterr TypeError)epochs epochs_fullZ noise_cov_iocapsysr figZticksr r rtest_plot_epochs_basicsH   r4rg-q=gdy=auto)magZgradstimcCs|j|ddS)z$Test the valid options for scalings.)rN)r )r0rr r r rtest_plot_epochs_scalingsLsr8c sfddttjD}j|dtjtddjdgdggdW5QRXtjtddjdggdW5QRXjdd d S) z5Test epoch_colors, for compatibility with autoreject.csg|]}dgtjqS)r)r"r*).0_r0r r Usz+test_plot_epochs_colors..) epoch_colorsz!length equal to the number of eporr9bzepoch colors for epoch \d+ has)Z event_colorN)ranger"r#r r'r,r-)r0r r>r r<rtest_plot_epochs_colorsSs   rAcCs^|}|}|jdkr0t|dks*td}n|jdkrNt|dksJtd}||ksZtdS)zTest scale bar for epochs.Z pyqtgraphr) 800.0 fT/cm 2000.0 fT matplotlib)rB0.55 secrCrFN)r Z_get_scale_bar_textsnamer"r )r0r r3ZtextsZwantsr r rtest_plot_epochs_scale_bar`s  rHc Cs6|j|jd}|jjdd}|jjdd}t|}|jjjd}|j ||fdd||jj ksnt |j ||fdd||jj kst |j ||fdd||jj kst | d| |dt|kst |jdd}|d d} | |jjd kst |jddd | |jjd ks,t |j d |jjd |j d |jjd | d|jjdd}|jjdd}|j ||fdd| d|\} } d| kst d| kst | d| t|dkst |}|jddd t|jjdkst |ddd|ddddS)z#Test plot_epochs mouse interaction.)r#rrrxformrrn_epochsyr)Zch_indexbutton)?rO)axrightleftz out of boundsrON)r r#mnetraces get_xdata get_ydatar"instr _fake_clickZ bad_epochsr r+Z _close_eventr%r$Z_click_ch_nameZ ax_vscrollZ ax_hscrollr.Z child_figs _fake_scroll) r0r1r2r r3rrMrLZ epoch_numZfirst_chouterrr r rtest_plot_epochs_clicksnsJ      r_cCs|tdd|jdd}t|jd}|jjd|}|jjd||jjd|d}|j ||gdd d }|dD]}| |q|j ||gddd d S) z&Test plot_epochs keypress interaction.g-=)r6rrKrrrrrI)pagedowndownuprarQrS-+=drfpageuphomez shift+rightendz shift+leftzrjsrk?hjr?)rJrNN) drop_baddictr r"timesrVrWrXrYr[r+)r1r r3Z sample_idxrrMkeyskeyr r rtest_plot_epochs_keypressess   rtcCstjdd}d}tdd|d}tddd gd d d gd d d gf}t||d|d}|j|dd}|jdkrzt|jj }n t|jj }|dkst dS)z2Test drawing of event lines in overlapping epochs.)rrd)shaperu)ar?)miscrx)r*Zch_typessfreq2<FrrrrrU)tminr#rx)r#rrD N) npZzerosrZ column_stackrr rGr"rVZ event_linesZ get_segmentsr )r rryr$r#r0r3Z n_event_linesr r r(test_plot_overlapping_epochs_with_eventss "  rcCs |dS)zTest sensor plotting.N)Z plot_sensorsr<r r rtest_epochs_plot_sensorssrc CsRtjdddd}tddd}t||}tjtdd| W5QRXd S) z:Test plotting of epochs when no data channels are present.r rig@@r7z!consider passing picks explicitlyrN) rrandomZ RandomStateZrandnrrr'r,r-r )r rr$r0r r rtest_plot_epochs_nodatas   rc Cs|}t|dkstttdks,t|}t|dksDtttdksXt|jddd|jddgdtd d d |jdgdgdgd d dtd|jdgdgdddtd|jdgdddtdd\}}|jdg|d|jdg|dd d dtd|jddgdddtj t dd|jdgddW5QRXtj t dd|jdd W5QRXtj t d!d |jtdd"tdd#d$W5QRXtd|jtd%d&d'tj t d(d|jtdddgd)d W5QRXtj t d*d|jdd+d,W5QRXtj t d-d|jtd.W5QRXtj t d/d8tdd\}}|j|tddgd0d1gd2d3~~W5QRXtdtj t d4d"tdd\}}|j|d5W5QRXdtgdtd d6gdtd d7gdtd d d8gfD]l\}}td|d\}}|jfd|d9|d:|tj t d;d|jfd|d:|W5QRXqtdtj t dW5QRXtj t d?d|jdd@W5QRXtj t dAd|jdgddBgdCW5QRXtj t dDd|jdEgdFW5QRXtj t dGd|jdgddgdW5QRXtdd9S)HaxTest plotting of epochs image. Note that some of these tests that should pass are triggering MPL UserWarnings about tight_layout not being applied ("tight_layout cannot make axes width small enough to accommodate all axes decorations"). Calling `plt.close('all')` just before the offending test seems to prevent this warning, though it's unclear why. rrEr6g?)rsigmarrZmeanF)Z show_sensors)rcombinets_args{Gz?test)rorder overlay_timesvminrallgMbPgMbP?)rrrvmaxcSs|S)N)min)rr r rz(test_plot_epochs_image..)rrr)rr3)raxesevokedcolorbarcSstt|dddS)N)rZaranger")rqrr r rrr)rrzOnly one channel in groupr)rrz dict or Nonefoo)r!z Scalings and units must have the)hi)Zho)ZunitsrT)Zinvert_y)rz Cannot combine sensors of differ)rzone of "fig" or "axes" must bebar)r3rz "fig" must contain . axes, got .)r3z""group_by" can only have one grouprT)rr)r3r!z"is a list, can only plot one group)r)r)r)rrN)rrz"axes" must be length ., got .z!must match the keys in "group_by"z4each value in "axes" must be a list of . axes, got .)rr!z""combine" must be None, a callable)rzsize of overlay_times parameter皙?)rrz fall outsideg8@)rz!must match the length of the data)Z plot_imager"r pltZ get_fignumsrpcloseZsubplotsr'r(r)r,r/r-Zfigure)r0Zfigsr3Zaxslengthkwargsrr r rtest_plot_epochs_images      $ "    $ rc Cstjtdd|W5QRX||tdtdtdggfdfD]&}tjtddt|W5QRXqVtdd S) zTest plotting a drop log.zbad epochs have not yet beenr)Oner r )r)Twor )r)rrr )rztuple of tuple of strrN)r'r,r-rror/rr)r argr r rtest_plot_drop_log4src s\|jddd|jddd|jdddtjtdd|jdgdW5QRXtd|}t|jd ksvt |jd d }|jd j d j j |jd j d j j tfd d|jddDst tfdd|jddDst |jddgd}d|jd}d |d dddf<dD],}tjt|d|j|dW5QRXq*dS)z$Test plotting epochs psd (+topomap).TF)averagespatial_colorszNo frequencies in bandrrrrbandsrrZjoint)Zvlimrc3s |]}|jdjjkVqdSrN)imagesnormrr:rP)vmin_0r r Ssz'test_plot_psd_epochs..rrc3s |]}|jdjjkVqdSr)rrrr)vmax_0r rrTs)z20 Hz)z15-25 Hzzfor channel %srNTFdB)plot_psdr'r, RuntimeErrorplot_psd_topomaprrr"rr rrrrrr*get_datar( UserWarning)r0r3Zerr_strrr )rrrtest_plot_psd_epochsCs&  $$rcCs,dddg}|j|d}t|jdks(tdS)z.Test plotting of PSD topography for nirs data.)rz0.2 Hz)g?z0.4 Hz)g?z0.8 HzrrTN)rr"rr )Z fnirs_epochsrr3r r rtest_plot_psdtopo_nirs_s  rc Cs|ddddddddgt|}t||dd }|||jdd }d }|D]}||qT|d d d |d d d |d|ddS)zTest of basic CTF plotting.ZUDIO001ZUPPT001z SCLK01-177zBG1-4304z MLC11-4304ZEEG058z UADC007-4302Tpreload) butterfly)r?r?r`rarbrarQrSrcrdrerfrfrgrhrirjrjrkrkrlrmrnrOrUrrN)Z pick_channelsrrr r&r+r\Z_resize_by_factor)raw_ctfr evtsr0r3rrrsr r rtest_plot_epochs_ctffs&    rc Cst|}t||dd}tjt|jdgd|dD]*}tjtdd|j|dW5QRXq6| d g|jd d d d S) z(Test plotting CTF epochs psd (+topomap).Trrrrzfor channel EEG060rrZEEG060F)rrN) rrr'r,rrr(rrZ drop_channels)rrr0rr r rtest_plot_psd_epochs_ctfs  rcCsFt|dd}t||ddddd}t|dks4t|jddddS) z.Test that using selection and butterfly works.NrrrOT)r}ZtmaxrZbaseliner)r!r)rrr"r r )rawr r#r0r r r$test_plot_epochs_selection_butterflysr)&Zmatplotlib.pyplotZpyplotrnumpyrr'rVrrrZ mne.datasetsrZ mne.eventrZmne.vizrrr4markZ parametrizerpr8rArHr_rtrrrZslowtestrrrrZrequires_testing_datarrrr r r r s>    1  2  `