U \ a[K@sJddlmZddlmZddlmZddlm Z ddl m Z m Z m Z ddlmZddlZddZddZd d Zd d Zd dZddZddZddZddZe dgdddddZe dgdddddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Z ej!"d*d+d,gd-d.Z#d/d0Z$d1d2Z%d3d4Z&ej!"d5d6d7d8gd9d:Z'dS);N)image_comparison)do_eventget_ax mock_event)assert_allclosec stfdd}tj|f|}t|dddddt|dddddt|d d d dd|d dd krt|jd ddddgdddddgg|jdjstdS)NcsBd_|jdkst|jdks"t|jdks0t|jdks>tdS)NTd) _got_onselectxdataAssertionErrorydataZepressZereleaseaxA/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/tests/test_widgets.pyonselects z!check_rectangle..onselectpressrr r buttononmoverreleasedrawtype)linenonegY@)err_msg) rwidgetsRectangleSelectorrgetrgeometryr r kwargsrtoolrrrcheck_rectangle s   r%cCsNttdddtdddtdddd tddd d ttdd d dS)NrF)ruseblitTrr&rr )rminspanxminspanyZpixels)r)r*Z spancoordsfillZ rectprops)r%dictrrrrtest_rectangle_selector&s   r/cCst}dd}tj||ddd}d|_t|ddddd d t|d d d dd t|dd d dd |jdksltt|dddddd t|ddddd t|d dddd t|ddddd t|dddddd |jdkstt|dddddd t|ddddd t|d dd dd t|ddd dd t|dddddd dd|jD}|ddddgks\tt|dddddd t|ddddd t|d dddd t|ddddd t|dddddd dd|jD}|ddddgkst|jjd kstt|jd!d!d"fd#dgd!S)$z'For ellipse, test out the key modifierscSsdSNrr rrrr3sztest_ellipse..onselectr(T)rmaxdist interactiverrr4rr )r r rkeyrrr)xr8r9 on_key_pressrcontrol}on_key_release)Kr<r>r<shift#cSsg|] }t|qSrint.0errr Tsz test_ellipse.."z ctrl+shiftcSsg|] }t|qSrrArCrrrrF_sF)INrgQ@) rrZEllipseSelectorextentsrr r!shaper)rrr$rMrrr test_ellipse/s^       rOcCsZt}dd}tj||dddddd}d |_|jd ks.onselectr(Trb)ZmarkerfacecolorZmarkeredgecolor)rr1r2Z marker_propsr3))rr4r4r)rrr4r4))r@_@r4rR)rRrrRr4rrr r rr8r)r8r4r8r4)lrUrV)r(rr(r)rrrrMZcornersr Z edge_centersrmcolorsZ same_colorZ_corner_handlesZartistZget_markerfacecolorZget_markeredgecolor)rrr$rrrtest_rectangle_handlesfsD   rXcstfdd}fdd}d|kr.||d<tj|f||}t|ddddd t|d d d dd t|d d d dd jstd|krjstdS)Ncs"d_|dkst|dkstdS)NTrr4r r ZvminZvmaxrrrrs zcheck_span..onselectcs"|dks t|dkstd_dS)Nrr<T)r _got_on_moverZrrrrs  zcheck_span..onmoveonmove_callbackrrrrrr<rr4)rrZ SpanSelectorrr r r[)argsr#rrr$rrr check_spans   r^cCs2tddddtddddtdtddd dS) N horizontalr(T)Zminspanr&verticalr)r\rr+r-)r^r.rrrrtest_span_selectorsrac sftfdd}tj|f|}t|dddddt|dddddt|d d d ddjsbtdS) Ncsd_|dddgkstdS)NT)rr)r<r<)r4r4rY)Zvertsrrrrsz&check_lasso_selector..onselectrrrrrr<rr4)rrZ LassoSelectorrr r r"rrrcheck_lasso_selectors rbcCs(ttdtdddtddddS)NFZred)color)r&Z linepropsTrr')rbr.rrrrtest_lasso_selectorsrdcCsft}t|dd}|dddgks*t|d|dddgksJt|dd}||dS)N)arQc)TFTTFrcSsdSr0rrrrrz#test_CheckButtons..)rr CheckButtonsZ get_statusr Z set_activeZ on_clickedZ disconnect)rcheckZcidrrrtest_CheckButtonss rkzcheck_radio_buttons.pngZmpl20T)styleZ remove_textcCs^dtjd<ttjddtddddg}tddddg}t|d t|d d dS) Nztext.kerning_factorg333333?)left皙?ffffff?333333?皙?)zRadio 1zRadio 2zRadio 3)zCheck 1zCheck 2zCheck 3)FTT)pltZrcParamsrZsubplots_adjustaxesr RadioButtonsri)Zrax1Zrax2rrrtest_check_radio_buttons_images   rvz check_bunch_of_radio_buttons.pngcCs"tddddg}t|ddS)Nro皙?rqrp)ZB1ZB2ZB3ZB4ZB5ZB6ZB7ZB8ZB9ZB10ZB11ZB12ZB13ZB14ZB15)rsrtrru)Zraxrrr!test_check_bunch_of_radio_buttonssrxc Csdt\}}tttj|dddddW5QRXtttj|dddddW5QRXdS)N8@$@)rlabelvalminvalmax slidermin)rr}r~r slidermax)rssubplotspytestZraises ValueErrorrSlider)figrrrr'test_slider_slidermin_slidermax_invalids     rcCspt\}}tj|ddddd}tj|dddd|d}|j|jksFttj|dddd|d }|j|jksltdS) Nryrzr{g@rr}r~rvalinit?)rr}r~rrrr|)rr}r~rrrrsrrrvalr )rrZslider_sliderrrrtest_slider_slidermin_slidermaxs    rcCsXt\}}tj|ddddd}|j|jks0ttj|ddddd}|j|jksTtdS)Nryrzr{g$rg9@)rsrrrrr~r rrrrrrrtest_slider_valmin_valmaxs   rc Cs`t\}}tj|dddddd}|jdks0ttj|ddddddd d gd}|jd ks\tdS) Nryrzr{g&@r)rr}r~rrZvalstep rg@g333333@rrrrrtest_slider_valstep_snappings    rcCst\}}tj|dddddd}|d|jdks:t|j |j }t |j dddd gt\}}tj|ddddd d}|d|jdkst|j |j }t |j ddd dgdS) Nryr r_)rr}r~rr orientationr(g?rr`)rsrrrset_valrr poly get_extents transformed transAxesinvertedrZbounds)rrrboxrrrtest_slider_horizontal_verticals$      rrr_r`cCs|dkrddddg}n ddddg}t\}}tj|ddd|d d gd }|j|j}t | |d dd dgt |j d | d t |j d |j|j}t | |ddddg| dt |j d| dt |j ddS)Nr`rrrKryrzrrw(\?)rr}r~rrr)rwr)rr333333?rrr)rrrw)rwrr)r()rr)rsrrZ RangeSliderrrrrrrZ get_pointsflattenrr)ridxrrrrrrrtest_range_slider-s.        rcs`td_fdd}t|}|D]\}}t||f|q(j|ksNtj|ks\tdS)a Helper function to test Polygon Selector. Parameters ---------- event_sequence : list of tuples (etype, dict()) A sequence of events to perform. The sequence is a list of tuples where the first element of the tuple is an etype (e.g., 'onmove', 'press', etc.), and the second element of the tuple is a dictionary of the arguments for the event (e.g., xdata=5, key='shift', etc.). expected_result : list of vertices (xdata, ydata) The list of vertices that are expected to result from the event sequence. selections_count : int Wait for the tool to call its `onselect` function `selections_count` times, before comparing the result to the `expected_result` rcsjd7_|_dS)Nr)_selections_count_current_result)Zverticesrrrrbsz(check_polygon_selector..onselectN)rrrZPolygonSelectorrr r)event_sequenceexpected_resultZselections_countrr$etypeZ event_argsrrrcheck_polygon_selectorLs   rcCs.dt||dfdt||dfdt||dfgS)NrrSrr)r.rSrrrpolygon_place_vertexosrcCsdddg}tddtddtddtdd}t||ddddg}tddtdddtd d fd tddd fd tddd fd tddd fdtddd fdtd d fgtddtdd}t||ddddg}tddtdddtdd fd tddd fd tddd fd tddd fdtddd fdtdd fgtddtdd}t||ddddg}tddtddtddtddd tddd fd tddd fd tddd fdtddd fg}t||ddddg}tddtddtddtdddtdd fd tddd fd tddd fd tddd fdtddd fdtdd fg}t||ddddg}dtd d fd tddd fd tddd fd tddd fdtddd fdtd d fdtdd fd tddd fd tddd fd tddd fdtddd fdtdd fg tddtddtddtdd}t||ddddg}tddtdddtdd fdtdd fgtddtddtddtdd}t||ddS)N)2r)r4r)rr4rr4r)r>rr:r;)r6rrSrr>rr=)rr>)r4r>r?rr<rK)r>r>)r>)r>rrescape)rrr.)rrrrrtest_polygon_selectorus                             rzhorizOn, vertOn)TT)TF)FTc Cstjddd\}\}}}tj|j||fd||d}|rHt|jdksHt|r^t|jdks^tt |ddd }| ||jD]}| d ks|tq||jD]}| d kstqt |d d d }| ||jD]}| d kstq|jD]}| d kstqdS) NrT)ZsharexF)r&horizOnvertOnrK??rS)rr)rrg?) rsrrZ MultiCursorZcanvaslenZvlinesr ZhlinesrrZ get_xdataZ get_ydata) rrrZax1Zax2Zax3Zmultieventlrrrtest_MultiCursors0      r)(Zmatplotlib.colorscolorsrWZmatplotlib.widgetsrZmatplotlib.pyplotZpyplotrsZmatplotlib.testing.decoratorsrZmatplotlib.testing.widgetsrrrZ numpy.testingrrr%r/rOrXr^rarbrdrkrvrxrrrrrmarkZ parametrizerrrrrrrrrsJ      7,       #`