a EbH@s<ddlmZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZddlmZmZddlmZmZmZmZmZmZdd lmZmZdd lmZdd l m!Z!m"Z"m#Z#m$Z$ddl%Z&ddl'Z'ddl(Z)dd l*m+Z+m,Z,d dZ-edgdgddddZ.ddZ/edgdddddZ0edgdddddZ1d d!Z2ed"gdd#d$d%d&Z3ed'd(gd)d*Z4ed+ge5d,krdnd-d.d/d0Z6ed1gd2d3Z7d4d5Z8ed6gdd#d7d8id9d:d;Z9dd?d@ge'j;dGdHZ?dIdJZ@e'j;*z,test_divider_append_axes..r )pltsubplotsr append_axescanvasdraw get_rendererdpiheightpytestapproxwidthZy0y1Zx0x1)figaxdividerrendererZbboxesr.r!r$r%test_divider_append_axess0   ((((000r9Ztwin_axes_empty_and_removedZpng) extensionstolcCstjddddgd}gd}tt|dt|d}|jdddd d d tt||t|dD]\}\}}tt|dt||}t||}d |vr|j dd d d |vr| d|vr|j dd d |jdd||rd|ndd d d qht j ddddS)N)z font.sizezxtick.labelsizezytick.labelsize)ZtwinxZtwinytwin)host invisible twin removedtwin invisibleztwin removed host invisibler:?r center)ZhorizontalalignmentZverticalalignmentrBFrAr@ r?)ZwspaceZhspace)mplZrcParamsupdater lentext enumeratergetattraxis set_visibleremover(subplots_adjust) generators modifiershimodgentr!r!r% test_twin_axes_empty_and_removed:s0     rXc CsZt}t|ddddddd}dtjd d }|d j|d td }|jd  |dS) Nor:r:r:Lrsingle) nrows_ncolsZngridsZ label_modeZ cbar_locationZ cbar_modei' rnearest) interpolationZnorm) r(figurernprandomZrandimshowrZ cbar_axesZcolorbar)r5gridZimr!r!r%$test_axesgrid_colorbar_log_smoketestVsrhzinset_locator.pngdefaultT)style remove_textc Cs$tjddgd\}}tjddd}d}td}|j\}}||d d |d d |f<|j||d d d t|d dd}|j||d d d |j j dd|j j ddd\}} } } | || || | tjddtjddt||dddddt|jdddddddd} || dS)Nfigsizeaxes_grid/bivariate_normal.npyTZnp_loadrmrwr_lowerextentr`origin upper rightZzoomlocZnbinsggggffffffFvisiblerCnone0.5loc1loc2fcecrD lower centerrrr borderpadsepframeon)r(r)rget_sample_datarbzerosshaperdryaxisget_major_locator set_paramsxaxisset_xlimset_ylimxticksyticksrr transData add_artist r5r6rfr{ZZ2nynxaxinsr4Zx2r3y2Zasbr!r!r%test_inset_locatorfs:         rzinset_axes.pngc Cs*tjddgd\}}tjddd}d}td}|j\}}||d d |d d |f<|j||d d d t|d d d|j d}|j||d d d |j j dd|j j ddd\}} } } ||| || | tjddtjddt||dddddt|jdddddddd} || dS)NrlrmrnrpTrqrrrvrxr_ryrz?rZ)r2r/bbox_to_anchorbbox_transformrrrFrrCrrrrDrrr)r(r)rrrbrrrdr transAxesrrrrrrrrrrrrrr!r!r%test_inset_axess>          rc Csd}d}tj||d\}}|ddddt|dddd}|jt|jt d|dd|dd|d d|d ddft|d d dd}|jt|jt d t|d ddddd}|jt|jt d||dd||d d|d |dd|d|d ft|dddd d}t|ddd|j dd d}|jt |j|jt t t|d d dd}Wdn1s0Yt t"t|d d |j d}Wdn1s0YdS)Nd)r}rl)ror.r?g@r)r2r/rr:z40%z30%)g(\?gQ?rrrr)rru)r2r/rrrgi@gY@z35%z60%)r2r/rrz100%)rrgffffff?333333?)r2r/rrrr)皙?rD)r2r/r)r2r/r)r(r)rPrr+r,r get_positionextentsrbarrayrrr0Zraises ValueErrorZwarns UserWarning)r.ror5r6ZinsZins1Zins2r!r!r%test_inset_axes_completesZ     $   & rzfill_facecolor.pngZmpl20)rkrjc Cs.tdd\}}|ddtddD]}||jdq&|djtdddd}t ||dj }t ||dj }t ||dddd d d d }| d|d |t|ddd d}|dd|ddtjjgtjjgt|d|ddd ddt ||dj }t ||dj } t || dddd d dd}| d|d |t|ddd d}|dd|ddtjjgtjjgt|d|dddddt ||dj } t ||d j } t | | dddd d dd}| d|d |t|ddd d}|dd|ddtjjgtjjgt|d|dddddt ||d j } t ||dj } t | | dddd d ddd }| d|d  |t|d dd d}|dd|dd|jg|jgt|d |dddddddS)Nr:rlrmFrrrrCrurb)loc1aloc2aloc1bloc2brrr~)r皙?rry)rrrrr facecolor)rrrrg)rrrrrcolor)rrrr)rrrrrrfill)rrrrr)r(r)Zset_size_inchesrangerrNZ tick_rightrZ from_extentsrrrZ set_clip_on add_patchrrrgcaaxesrZ set_ticksr)r5r6rTbboxZbbox1Zbbox2prZbbox3Zbbox4Zbbox5Zbbox6Zbbox7Zbbox8r!r!r%test_fill_facecolors|                   rzzoomed_axes.pngzinverted_zoomed_axes.pngcCst\}}|gdgd|gdt|ddd}|gdt\}}|gdgd|gdt|ddd}|gddS) Nr:rCru)r:rur:rug@z lower rightr)皙?ffffff?rr)rur:rur:)rrrr)r(r)plotrMr)r5r6Zinset_axr!r!r%test_zooming_with_inverted_axes7s  rzanchored_direction_arrows.pngx86_64g{Gz?)r<cCs<t\}}|jtdddt|jdd}||dS)Nr^r^r_r`XY)r(r)rdrbrrrr)r5r6Z simple_arrowr!r!r%test_anchored_direction_arrowsFs rz'anchored_direction_arrows_many_args.pngcCsTt\}}|tdt|jdddddddd d d d d dddd}||dS)NrABr~redgrrCTgffffff?gQg{Gzr r^rl) rrZ aspect_ratiorrralphaZsep_xZsep_yZ back_lengthZ head_widthZ head_lengthZ tail_width)r(r)rdrbZonesrrr)r5r6Zdirection_arrowsr!r!r%(test_anchored_direction_arrows_many_argsPs   rcCsxt\}}t|}tddi|jddd}Wdn1sF0Y|jt |j ddj dstJdS) Nzfigure.subplot.wspaceg{Gz?rz5%)sizeF)originalg@/Ċ?) r(r)r rGZ rc_contextr*r+r,rbiscloserr2)r5r6r7Zcaxr!r!r%test_axes_locatable_position]s , rzimage_grid.pngZ bbox_inchesZtight)rkrjZ savefig_kwargcCsdtdd}tdd}t|dddd}td D]*}||j|d d ||d |q4dS) Nrrr:)rmrmrYrCrCr)r]Zaxes_padrmr_rz test {0}{0}) rbarangereshaper(rarrrd set_titleformat)rgr5rerTr!r!r%test_image_gridgs   rcCstjdd\}}|gdgd\}t|d}|gdgdt||dddd d t|||j}t j |j gd dS) N)r=r}rnr)rr:rrmr:rurz0.3r)g333331g+g@g@) r(r)rrrr Z get_tightbboxr+r-rbtestingrr)r5r6lZax_zoomrr!r!r%test_gettightbboxvs  rclick_onbigsmallzbig_on_axes,small_on_axes))rr)hostr)rparasite)rr)rrcsztjddddd}tjdddddd}gfd d }tjd |||f}d d d d }d|vrpt|d<d|vsd|vrtd|d<|d|d<|||||||dkr||}d} n ||}d} ||dur|d}|j | \} } t d|j j| | dd} | | |dkr,dnd} t| ksBJddD}||vs^J|dkrv||vsvJd S)z2Test pick events on normal, host or parasite axes.)?rrDrl)picker)rrrr)rrcs|dS)N)append)eventZreceived_eventsr!r%on_picksz/test_picking_callbacks_overlap..on_pickZ pick_eventN)rrrrrrrYr)333333?r)rDrDZbutton_press_eventr:)buttonrrCcSsg|] }|jqSr!)Zartist)r"rr!r!r% r'z2test_picking_callbacks_overlap..)r(Z RectangleZgcfr+Z mpl_connectrr r>rrZ transformrrapickrI)Z big_on_axesZ small_on_axesrrrrZrectangles_on_axesrZ click_axesZ axes_coordsxrmZexpected_n_eventsZ event_rectsr!rr%test_picking_callbacks_overlaps@       rc Cstdd}tdd}tdd\}\}}||||d}t|dt|t |t|gt |t dt |gd}| | d | | d|j|}|}|j|jksJ|j|jtd ksJdS) N)rmrl)rlrmr:rCrDrY) horizontalverticalrg|Gz?)rbrrr(r)rdrSizeZAxesXZFixedZAxesYZScaledZset_axes_locatorZ new_locatorr+r,rr/r2r0r1) Zarr1Zarr2r5Zax1Zax2rr7p1p2r!r!r%test_hbox_dividers$   rcCs*t}tjjjif}t|dd|d}dS)NrYrZ)r] axes_class)r(ra mpl_toolkitsZ axes_grid1Zmpl_axesZAxesr)r5rgrr!r!r%test_axes_class_tuplesrcCstt}t|dddd}t||jt|jt|jt|t |jdt|dddd}t|t |jddS)z:Test Grid axes_all, axes_row and axes_column relationship.rY)rCrurow directioncolumnN) r(rar rZaxes_allaxes_rowrbZ transposeZ axes_columnravel)r5rer!r!r%test_grid_axes_listss r r)rrcCst}t|dd|d}ddt|jD}|dj|djkrV|dj|djksZJ|dj|djkr|dj|djksJ|d j|djkr|d j|djksJd S) z%Test positioning of the axes in Grid.rYrrcSsg|] }|qSr!)Zget_axes_locator)r"r6r!r!r%rr'z+test_grid_axes_position..r:rrCruN)r(rar rbrrZ_nxZ_ny)rr5rerr!r!r%test_grid_axes_positions ,,r )B itertoolsrplatformZ matplotlibrGZmatplotlib.pyplotZpyplotr(rZmatplotlib.backend_basesrZmatplotlib.colorsrZmatplotlib.transformsrrZmatplotlib.testing.decoratorsrr Zmpl_toolkits.axes_grid1r rr r r rrZ(mpl_toolkits.axes_grid1.anchored_artistsrrZ$mpl_toolkits.axes_grid1.axes_dividerrZ%mpl_toolkits.axes_grid1.inset_locatorrrrrZ mpl_toolkits.axes_grid1.mpl_axesrr0numpyrbZ numpy.testingrrr9rXrhrrrrrmachinerrrrrmarkZ parametrizerrrr r r!r!r!r%sd         ) ,, M       2