a fø1bæHã@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 d„Z-edgdgdddd„ƒZ.dd„Z/edgddddd„ƒZ0edgddddd„ƒZ1d d!„Z2ed"gdd#d$d%d&„ƒZ3ed'd(gƒd)d*„ƒZ4ed+ge 5¡d,krŽdnd-d.d/d0„ƒZ6ed1gƒd2d3„ƒZ7d4d5„Z8ed6gdd#d7d8id9d:d;„ƒZ9dd?d@g¡e'j; dGdH„Z?dIdJ„Z@e'j; *óz,test_divider_append_axes..r )ÚpltÚsubplotsr Ú append_axesÚcanvasÚdrawÚ get_rendererÚdpiÚheightÚpytestÚapproxÚwidthZy0Úy1Zx0Úx1)ÚfigÚaxÚdividerZrendererZbboxesr.r!r$r%Útest_divider_append_axess0 û  ((((000r8Ztwin_axes_empty_and_removedZpngé)Ú extensionsÚtolcCstj ddddœ¡gd¢}gd¢}tt|ƒdt|ƒdƒ}|jdddd d d tt||ƒt|ƒdƒD]œ\}\}}tt|ƒdt|ƒ|ƒ}t||ƒƒ}d |vr´|j dd…  d ¡d |vrÄ|  ¡d|vrà|j dd…  d ¡|jdd||röd|ndd d d qht j ddddS)Né)z font.sizezxtick.labelsizezytick.labelsize)ZtwinxZtwinyÚtwin)Úúhost invisibleú twin removedútwin invisibleztwin removed host invisibler9éçà?r Úcenter)ZhorizontalalignmentZverticalalignmentrAFr@r?Ú r>)ZwspaceZhspace)ÚmplZrcParamsÚupdater ÚlenÚtextÚ enumeraterÚgetattrÚaxisÚ set_visibleÚremover(Úsubplots_adjust)Ú generatorsZ modifiersÚhÚiÚmodÚgenÚtr!r!r%Ú test_twin_axes_empty_and_removed:s0 ÿ ÿ  ÿ ÿrVc CsZt ¡}t|ddddddd}dtj d d ¡}|d j|d tƒd }|jd   |¡dS) Néo©r9r9r9ÚLrZsingle)Ú nrows_ncolsZngridsZ label_modeZ cbar_locationZ cbar_modei'é rÚnearest)Ú interpolationZnorm) r(ÚfigurerÚnpZrandomZrandÚimshowrZ cbar_axesZcolorbar)r5ÚgridÚZÚimr!r!r%Ú$test_axesgrid_colorbar_log_smoketestVsûrdzinset_locator.pngÚdefaultT)ÚstyleÚ remove_textc Cs$tjddgd\}}tjddd}d}t d¡}|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)Néé©Úfigsizeúaxes_grid/bivariate_normal.npyT©Znp_load©éýÿÿÿriéüÿÿÿé©é–rsér\Úlower©Úextentr]Úoriginéú upper right©ZzoomÚlocé©Znbins©gø¿gÍÌÌÌÌÌì¿gÀgffffffþ¿F©ZvisiblerBÚnoneú0.5©Úloc1Úloc2ÚfcÚecrCú lower centerr©r|rÚ borderpadÚsepÚframeon)r(r)rÚget_sample_datar_ÚzerosÚshaper`rÚyaxisÚget_major_locatorÚ set_paramsÚxaxisÚset_xlimÚset_ylimÚxticksÚyticksrrÚ transDataÚ add_artist© r5r6rbrwZZ2ZnyZnxÚaxinsr4Zx2r3Zy2Zasbr!r!r%Útest_inset_locatorfs:   ÿ ÿ     ûrœzinset_axes.pngc Cs*tjddgd\}}tjddd}d}t d¡}|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)NrhrirjrlTrmrnrrrtr\rurvçð?rX)r2r/Úbbox_to_anchorÚbbox_transformr}r~rFr€rBrr‚rƒrCrˆrr‰)r(r)rrr_rŽrr`rÚ transAxesrr‘r’r“r”r•r–r—rrr˜r™ršr!r!r%Útest_inset_axess>   ÿ ÿ ÿ     ûr¡c Csöd}d}tj||d\}}| dddd¡t|dddd}|j ¡t| ¡jt   d|dd|dd|d d|d ddf¡ƒt|d d dd}|j ¡t| ¡jt   d ¡ƒt|d ddddd}|j ¡t| ¡jt   d||dd||d d|d |dd|d|d f¡ƒt|dddd d}t|ddd|j dd d}|j ¡t | ¡j| ¡jƒt  t¡ t|d d dd}Wdƒn1sª0Yt  t¡"t|d d |j d}Wdƒn1sè0YdS)Néd)ryrh)rkr.rçÍÌÌÌÌÌì?g@r)r2r/rŠr9z40%z30%)gÂõ(\â?g…ëQ¸å?r£r£rr)éÈr¢rq)r2r/ržr|rŠgi@gY@z35%z60%)r2r/r|rŠz100%)rrgffffffÖ?ç333333ã?)r2r/ržrŸr|rŠ)çš™™™™™Ù?rC)r2r/rž)r2r/rŸ)r(r)rOrr+r,rÚ get_positionÚextentsr_Zarrayr rr0ZraisesÚ ValueErrorZwarnsÚ UserWarning)r.rkr5r6ZinsZins1Zins2r!r!r%Útest_inset_axes_complete½sZ ÿþ þ ÿ  $ÿþþ  ÿ ÿ& ÿr«zfill_facecolor.pngZmpl20)rgrfc Cs.t dd¡\}}| dd¡tddƒD]}||j d¡q&|dj ¡t dddd¡}t ||dj ƒ}t ||dj ƒ}t ||dddd d d d }|  d¡|d  |¡t|ddd d}| dd¡| dd¡t ¡jj g¡t ¡jj g¡t|d|ddd ddt ||dj ƒ}t ||dj ƒ} t || dddd d dd}|  d¡|d  |¡t|ddd d}| dd¡| dd¡t ¡jj g¡t ¡jj g¡t|d|dddddt ||dj ƒ} t ||d j ƒ} t | | dddd d dd}|  d¡|d  |¡t|ddd d}| dd¡| dd¡t ¡jj g¡t ¡jj g¡t|d|dddddt ||d j ƒ} t ||dj ƒ} t | | dddd d ddd }|  d¡|d   |¡t|d dd d}| dd¡| dd¡|j g¡|j g¡t|d |dddddddS)Nr9rhriFrr¦r¥rBrqÚrÚb)Úloc1aÚloc2aÚloc1bÚloc2br‡r†rz)r|çš™™™™™É?r‚rƒÚy)r®r¯r°r±r‡Ú facecolor)r„r…r´r‡Úg)r®r¯r°r±r‡Úcolor)r„r…r¶r‡)r®r¯r°r±r‡r†Úfill)r„r…r†r‡r·)r(r)Zset_size_inchesÚrangerrMZ tick_rightrZ from_extentsrr˜rZ set_clip_onÚ add_patchrr”r•ÚgcaÚaxesr“Z set_ticksr)r5r6rRÚbboxZbbox1Zbbox2Úpr›Zbbox3Zbbox4Zbbox5Zbbox6Zbbox7Zbbox8r!r!r%Útest_fill_facecolorés|  þ    þ    þ    þ     r¾zzoomed_axes.pngzinverted_zoomed_axes.pngcCs˜t ¡\}}| gd¢gd¢¡| gd¢¡t|ddd}| gd¢¡t ¡\}}| gd¢gd¢¡| gd¢¡t|ddd}| gd¢¡dS) N©r9rBrq)r9rqr9rqg@z lower rightr{)çš™™™™™ñ?çffffffö?rÀrÁ)rqr9rqr9)rÁrÀrÁrÀ)r(r)ÚplotrLr)r5r6Zinset_axr!r!r%Útest_zooming_with_inverted_axes7s  rÃzanchored_direction_arrows.pngZx86_64g{®Gáz„?)r;cCs<t ¡\}}|jt d¡ddt|jddƒ}| |¡dS)N©r[r[r\©r]ÚXÚY)r(r)r`r_rŽrr r™)r5r6Z simple_arrowr!r!r%Útest_anchored_direction_arrowsFs rÈz'anchored_direction_arrows_many_args.pngcCsTt ¡\}}| t d¡¡t|jdddddddd d d d d dddd}| |¡dS)NrÄÚAÚBrzZredgà¿r¥rBTgffffffæ?g¸…ëQ¸®¿g{®Gáz´¿ré r[rh) r|r¶Z aspect_ratiorrŠrŒZalphaZsep_xZsep_yZ back_lengthZ head_widthZ head_lengthZ tail_width)r(r)r`r_Zonesrr r™)r5r6Zdirection_arrowsr!r!r%Ú(test_anchored_direction_arrows_many_argsPs   ürÌcCsxt ¡\}}t|ƒ}t ddi¡|jddd}Wdƒn1sF0Y|j ¡t  |j ddj d¡stJ‚dS) Nzfigure.subplot.wspaceg{®Gáz”?rz5%)ÚsizeF)Zoriginalg@¸Ù/ÄŠ¢?) r(r)r rFZ rc_contextr*r+r,r_Ziscloser§r2)r5r6r7Zcaxr!r!r%Útest_axes_locatable_position]s , ÿrÎzimage_grid.pngZ bbox_inchesZtight)rgrfZ savefig_kwargcCsdt d¡ d¡}t dd¡}t|dddd}td ƒD]*}||j|d d || d   |¡¡q4dS) Nr¢rÄr9)ririrW©rBrBr)rZZaxes_padrir\rÅz test {0}{0}) r_ÚarangeÚreshaper(r^rr¸r`Z set_titleÚformat)rcr5rarRr!r!r%Útest_image_gridgs   rÓcCsˆtjdd\}}| gd¢gd¢¡\}t|dƒ}| gd¢gd¢¡t||dddd d t|ƒ| |j ¡¡}t j   |j gd ¢¡dS) N)r<ryrjr¿)rr9rrir9rqrz0.3rƒ)g33333³1ÀgÍÌÌÌÌÌ+ÀgÍÌÌÌÌÌ@gš™™™™™@) r(r)rÂrrr Z get_tightbboxr+r-r_Ztestingrr¨)r5r6ÚlZax_zoomr¼r!r!r%Útest_gettightbboxvs  ÿrÕÚclick_onÚbigÚsmallzbig_on_axes,small_on_axes))rºrº)ÚhostrÙ)rÙÚparasite)rÚrÙ)rÚrÚcsztjddddd}tjdddddd}g‰‡fd d „}t ¡j d |¡||f}d d d d œ}d|vrpt ¡|d<d|vs€d|vrœtdƒ|d<|d ¡|d<|| |¡|| |¡|dkrÎ||}d} n ||}d} ||durî|d}|j   | ¡\} } t d|j j| | dd} |  | ¡|dkr,dnd} tˆƒ| ksBJ‚dd„ˆDƒ}||vs^J‚|dkrv||vsvJ‚d S)z2Test pick events on normal, host or parasite axes.)çÐ?rÛrCrh)Úpicker)r¦r¦r²r¬)r´rÜcsˆ |¡dS)N)Úappend)Úevent©Zreceived_eventsr!r%Úon_pickšsz/test_picking_callbacks_overlap..on_pickZ pick_eventN)rºrÙrÚrºrÙrÚrWr×)ç333333Ó?rá)rCrCZbutton_press_eventr9)ZbuttonrØrBcSsg|] }|j‘qSr!)Zartist)r"rÞr!r!r%Ú ºr'z2test_picking_callbacks_overlap..)r(Z RectangleZgcfr+Z mpl_connectrºr r=r¹r Z transformrr^ZpickrH)Z big_on_axesZ small_on_axesrÖr×rØràZrectangles_on_axesr»Z click_axesZ axes_coordsÚxr³ÚmZexpected_n_eventsZ event_rectsr!rßr%Útest_picking_callbacks_overlap†s@     ÿ  råc Csôt d¡ d¡}t d¡ d¡}t dd¡\}\}}| |¡| |¡d}t|dt |¡t  |¡t |¡gt  |¡t  d¡t  |¡gd}|  |  d ¡¡|  |  d¡¡|j ¡| ¡}| ¡}|j|jksÖJ‚|j|jt d ¡ksðJ‚dS) Né)rirh)rhrir9rBrCrW)Z horizontalZverticalrg|®Gázä?)r_rÐrÑr(r)r`rÚSizeZAxesXZFixedZAxesYZScaledZset_axes_locatorZ new_locatorr+r,r§r/r2r0r1) Zarr1Zarr2r5Zax1Zax2rr7Zp1Zp2r!r!r%Útest_hbox_dividerÀs$  ý rècCs*t ¡}tjjjif}t|dd|d}dS)NrWrX)rZÚ axes_class)r(r^Ú mpl_toolkitsZ axes_grid1Zmpl_axesZAxesr)r5réZgrr!r!r%Útest_axes_class_tuple×srëcCstt ¡}t|dddd}t||jƒt|jt |j¡ƒt|t  |j¡dƒt|dddd}t|t  |j¡dƒdS)z:Test Grid axes_all, axes_row and axes_column relationship.rW)rBrqÚrow©Ú directionÚcolumnN) r(r^r rZaxes_allÚaxes_rowr_Z transposeZ axes_columnÚravel)r5rar!r!r%Útest_grid_axes_listsÝs ròrî)rìrïcCs¶t ¡}t|dd|d}dd„t |j¡Dƒ}|dj|djkrV|dj|djksZJ‚|dj|djkr‚|dj|djks†J‚|d j|djkr®|d j|djks²J‚d S) z%Test positioning of the axes in Grid.rWrÏrícSsg|] }| ¡‘qSr!)Zget_axes_locator)r"r6r!r!r%râír'z+test_grid_axes_position..r9rrBrqN)r(r^r r_rñrðZ_nxZ_ny)rîr5rar|r!r!r%Útest_grid_axes_positionès ,,ró)BÚ itertoolsrÚplatformZ matplotlibrFZmatplotlib.pyplotZpyplotr(rZmatplotlib.backend_basesrZmatplotlib.colorsrZmatplotlib.transformsrrZmatplotlib.testing.decoratorsrr Zmpl_toolkits.axes_grid1r rçr 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_axesrêr0Znumpyr_Z numpy.testingrrr8rVrdrœr¡r«r¾rÃÚmachinerÈrÌrÎrÓrÕZmarkZ parametrizerårèrëròrór!r!r!r%Úsd         ) ,, M  ÿ   þ  2