U mdG@s< ddlZddlZddlZddlmZmZmZmZddlZ ddl m Z m Z m Z ddlmZddlmZmZddlmZmZddlmZddlmZmZdd lmZmZdd lmZdd l m!Z!ddl"m#Z$ddl%Z&ej'ed d dZ(edgdddZ)e(dgddddZ*e(dgddddZ+ddZ,e(dgddZ-e(d gd!d"Z.d#d$Z/e(d%gd&d'Z0e(d(gd)d*Z1d+d,Z2e(d-gd.d/Z3e(d0gd1d2Z4e(d3gd4d5Z5e(d6gd7d8Z6ej78d9d:d;dd;de(dQgdRdSZ?edgddTdUZ@e(dVgdWdXZAe(dYgdZd[ZBe(d\gd]d^ZCedgdd_d`ZDedgddadbZEej78dcd dgedgddddeZFej78dfdgdhgedgddidjZGedgddkdlZHe(dmgdndEdodpZIe(dqgdrdsZJe(dtgdudvZKe(dwgdxdyZLedgddzd{ZMe(d|gd}d~ZNe(dgddddZOedgdddZPe(dgddEddZQe(dgddEddZRe(dgddZSe(dgddZTe(dgddZUddZVddZWe(dddgddZXedgdddZYe(dgddZZddZ[edgdddZ\ddZ]e(dgddZ^ddZ_e(dgddZ`e(dgddZae(dgddZbe(dgddddZce(dgddddZde(dgddddZeddZfddÄZgddńZhdtddȄZie(dgdd˄Zje(dgdd΄ZkddЄZldd҄Zme(dgddՄZnddׄZoddلZpej78ddۡej78ddݡdd߄ZqddZre(dgddddZse(dgddddZte(dgddZuej78de&jve&jwgej78dddddddgddZxGdddZyddZzedgdddZ{ddZ|ddZ}edgdddZ~edgdddZedgddddZe(dgd d Ze(d gd d ZedgdddddZedgddddZddZddZddZddZddZej78d d!e jdd"fd!e jd#d$fd!e jd%d&fd!e jdd'fd(e jdd)fd(e jd#d*fd(e jd%d+fgd,d-Ze jd edgdd.d/Zd0d1Zed2gd dgd3d4d5d6Zd7d8Zd9d:Zd;d<Zej78d=ed>id?fed@id?fedAid?feed#dBid?feed%dBid?feedCdBid?fedDd#didEfedDddddۜdEfedDddƐdFdEfedGidHff dIdJZedgddKdLZdMdNZedgddOdPZedQgd d dRdSdTZdUdVZdWdXZej78dYdCdZd[dZd\gdZdZd]d^gdZdZdZd_gd`dZdZdaggdbdcgdddegfdfdcgdedegfdfdfgdedggfgdddgfd%d[dZdZd\gdZd]dZd^gdZdZdZd_gdZdZd`daggdfdcgdgdggfdcdcgdedggfdhdfgdidggfgd;d;dgfd#dZdZd[d\gd]dZdZd^gdZdZdZd_gdZd`dZdaggdfdfgdgdegfdcdbgdgdigfdcdfgdgdggfgdd;dgfgdjdkZedlgd d dRdmdnZedogd ddRdpdqZedgddrdsZdS(uN)Axes3Daxes3dproj3dart3d) MouseButton MouseEventNavigationToolbar2)cm)colorspatches)image_comparisoncheck_figures_equal) mock_event)LineCollectionPolyCollection)Circle PathPatch)Path)TextTdefault) remove_textstyleZpng) extensionscCs |jtddd}|ddS)N3d projection subplot_kwF)subplotsdictZ set_visiblefig_testfig_refaxr$_/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/mpl_toolkits/mplot3d/tests/test_axes3d.pytest_invisible_axessr&z aspects.pngF)rc Csd}tjdt|ddid\}}ddg}tdddg}tttt|||d}|D]N\}}t t |||d|dkr\|D]}|j t ||||qq\t |D]$\} }|d |j|| d d qdS) NautoequalZequalxyZequalyzZequalxzrrrrr+ZdatalimZ adjustable)pltrlennparray itertools combinationslistproductsumabsplot3Dzip enumerateset_box_aspect set_aspect ZaspectsfigaxsrscaleZptsstartendr#ir$r$r% test_aspects!s  $ rHzaspects_adjust_box.pngc Csd}tjdt|ddidd\}}ddg}tdd d g}tttt|||d }|D]N\}}t t |||d|dkr^|D]}|j t ||||qq^t |D]\} }|j|| d d qdS)Nr'r*rr) r.)rfigsizerr/r.r+r,boxr0)r1rr2r3r4r5r6r7r8r9r:r;r<r=r?r@r$r$r%test_aspects_adjust_box3s   $rLcCsZt}|jdd}|d|d|d|d|dt|dksVt dS) NrrlabeltitlexyzzV) r1figure add_subplotZ set_label set_title set_xlabel set_ylabel set_zlabelreprAssertionErrorrAr#r$r$r%test_axes3d_reprEs      r[zaxes3d_primary_views.pngcCsddddddg}tjddd d id \}}t|jD]X\}}|d |d |d|d|j||d||d||ddq2t dS)N)Zr)rr]rrrr)r]r\r)rr\r)rrr,r.rrrrOrPrQorthorr*elevazimroll) r1rr=flatrUrVrW set_proj_type view_init tight_layout)viewsrArBrGr#r$r$r%test_axes3d_primary_viewsRs    *rjz bar3d.pngc Cs~t}|jdd}tddddgddd d gD]J\}}td}td}|gt|}d |d <|j|||dd |d dq.dS)NrrrCgbrP rcedge皙?)zszdiraligncoloralpha)r1rRrSr<r3aranger2bar)rAr#rprQxsyscsr$r$r% test_bar3dgs "  r}c Cs\t}|jdd}dD]>}tt|}t|}t|}|j|||ddd|dqdS)Nrr)redgreenblueyellowr*rv)r1rRrSr3rxr2 zeros_likebar3d)rAr#rprzr{rsr$r$r%test_bar3d_colorsss   rzbar3d_shaded.pngc Cstd}td}t||\}}||}}||d}ddddg}tjtdt|d}|jdt|t d d d }t ||D]:\}\} } } |j |||d dd|d d|j | | | dq|j dS)Nr/r+r*)rmir)rmrmrm)rmr]),rrrJrrrrTshadera)r3rxmeshgridravelr1rR figaspectr2rrr<rrgcanvasdraw) rOrPx2dy2drQrirArBr#rbrcrdr$r$r%test_bar3d_shadeds     rzbar3d_notshaded.pngc Cs|t}|jdd}td}td}t||\}}||}}||}|j|||ddd|dd|j dS) Nrrr/r+rr*Fr) r1rRrSr3rxrrrrr)rAr#rOrPrrrQr$r$r%test_bar3d_notshadeds   rc st}|jddddd}tjddd}d\}}||tt|t|\}}|}|}||}fdd t D}|j ||ddd||d |d } tj || j ddd dS) Nr*rrrr\)ZazdegZaltdeg)r.r/csg|]}t|qSr$)r coolwarm).0rGZarear$r% sz*test_bar3d_lightsource..T) rOrPrQZdxZdydzrvrZ lightsource)r1rRrSmcolorsZ LightSourcer3rrxrrangertestingassert_array_equalZ _facecolor3d) rAr#ZlslengthwidthrOrPrrvZ collectionr$rr%test_bar3d_lightsources& rz contour3d.pngcCst}|jdd}td\}}}|j|||ddtjd|j|||ddtjd|j|||d d tjd|dd | dd | dd dS Nrr皙?rQ)rtoffsetcmaprOrP(d r1rRrSr get_test_datacontourr rset_xlimset_ylimset_zlimrAr#XYZr$r$r%test_contour3ds   rzcontour3d_extend3d.pngc Csft}|jdd}td\}}}|j|||ddtjdd|dd | d d | d d dS)NrrrrQrT)rtrrZextend3drrmrPrrr$r$r%test_contour3d_extend3ds   rzcontourf3d.pngcCst}|jdd}td\}}}|j|||ddtjd|j|||ddtjd|j|||d d tjd|dd | dd | dd dSr) r1rRrSrrcontourfr rrrrrr$r$r%test_contourf3ds   rzcontourf3d_fill.pngcCst}|jdd}ttdddtddd\}}|dd}d|ddddddf<|j|||dd dgtj d | dd| dd| d d dS) Nrrr,?r皙?r+皙)rlevelsrr*) r1rRrSr3rrxcliprr rrrrrr$r$r%test_contourf3d_fills $   rzextend, levelsbothr,r/rminmaxc Csttdddtddd\}}|d|d}tjd}||d||ddd|d }|jd d } | j |||fd ddd ddgi||jd d } | j ||||fd|i|| | fD](} | dd| dd| ddqdS)Nrr,rviridisrr*)vminvmaxrrrrr/rrextendro) r3rrxmpl colormapscopyZ set_underZset_overrSrrrr) r!r"rrrrrrkwargsax_refax_testr#r$r$r%test_contourf3d_extends$  $    rztricontour.png{Gz?)tolcCst}tjdtjdd}tjdd}|d|d }|jddddd}|||||jddddd}||||dS)Ni!N,?r,r*rr) r1rRr3randomseedZrandrSZ tricontourZ tricontourf)rArOrPrQr#r$r$r%test_tricontour s rcCsbt}|jdd}d\}}tdd|}tdd|}tjdd||g}||||dgdS) Nrr)rmrnrrorr,r)r1rRrSr3linspacerrandintr)rAr#nxnyrOrPrQr$r$r%test_contour3d_1d_inputs rz lines3d.pngcCsvt}|jdd}tdtjdtjd}tddd}|dd}|t|}|t|}||||dS) Nrrr/rrr,r*) r1rRrSr3rpisincosplot)rAr#thetarQrCrOrPr$r$r% test_lines3d%s  rcCs<|jdd}|dgdgd|jdd}|ddddS)Nrrr*orSrr!r"ax1ax2r$r$r%test_plot_scalar1s  rc CstjtddtdggW5QRXtjtddtgdgW5QRXtjtddtggdW5QRXtggg}tjtdd|dggW5QRXtjtdd|gdgW5QRXtjtdd|ggdW5QRXdS)Nz x must bematchrz y must bez z must be)pytestraises RuntimeErrorrZLine3D set_data_3d)liner$r$r%test_invalid_line_data9srzmixedsubplot.pngc Csdd}tddd}tddd}tjtdd}|d d d }|j|||d |||d d d|d|jd d d dd}ttdddtddd\}}t ||}t |}|j |||ddddd| dd dS)NcSs tdtj|t| S)Nr,)r3rrexp)tr$r$r%fLsztest_mixedsubplots..f@rr@rr,r*Zbozk--r)markerfacecolorTrrr+rrrF)rcountccount linewidth antialiasedr) r3rxr1rRrrSrgridrhypotr plot_surface set_zlim3d) rt1t2rAr#rrRrr$r$r%test_mixedsubplotsJs   $  rcCsP|jdd}|jddddd||jdd}||jddddddS)Nrrrz some strings)rStextrhrr$r$r%test_tight_layout_textbs   r z scatter3d.pngcCst}|jdd}|jtdtdtddddtdd}}}|j|||dd dd |d <|jgggdd ddS) NrrrorCr)rpmarkerrnrl^rrrr1rRrSscatterr3rxrAr#rOrPrQr$r$r%test_scatter3dos rzscatter3d_color.pngcCst}|jdd}|jtdtdtddddd|jtdtdtddddd|jtddtddtddd d d dS) NrrrorCnoner) facecolor edgecolorr rnrlr )rvr rrZr$r$r%test_scatter3d_color|s  "rzscatter3d_linewidth.pngcCsDt}|jdd}|jtdtdtddtdddS)Nrrror)r rrrZr$r$r%test_scatter3d_linewidths  rcCs||jdd}|jtdtdtddd}|td|jdd}|jtdtdtddtdddS)Nrrrorr )r  linewidths)rSrr3rxset_linewidthsr"r!rrprr$r$r%%test_scatter3d_linewidth_modifications  rc Cs|jdd}|jtdtdtddd}|d|d|dd gd |sbt| d |rxt| t dd | d |jdd}|jtdtdtdddddd gd d d d d dS)NrrrorrC1C2333333?ffffff?r+FKr.)r rrrw depthshader r) rSrr3rx set_facecolor set_edgecolor set_alphaget_depthshaderYset_depthshadeZ set_sizesfullrrr$r$r%test_scatter3d_modifications*         r)r"c Cs2tjddddf\}}t|j|j}t|jd}d|ddddddf<d|ddddddf<t|jd}d |dd dd f<d |d ddd f<d|d dd df<t|jd}d|dd dd f<d|d ddd f<d|d dd df<t|jd} d | ddddddf<d | ddddddf<dd||||||| fD\}}}}}}} |jdd} dd|||| fD} tj | D]p\} } }}|| k|| kB||kB| |kB}tj j ||t d}t | t|j} | j|||| | ||d|d q|jdd}|j||||||| d|d dS)z1Test that marker properties are correctly sorted.Nrorrr,r*C0rr+rrr/ZC3ZC4ZC5ZC6 ZC7cSsg|] }|qSr$)flattenrar$r$r%rsz*test_scatter3d_sorting..rrcss|]}t|VqdS)N)r3uniquer.r$r$r% sz)test_scatter3d_sorting..Zdtype)r fceclwrwr")r3mgridrxsizereshapeshaper(rSr5r8maZ masked_arrayfloatrepeatr9maskr)r"r!r"rPrOrQsizes facecolors edgecolorsrrZsetsr r3r4r5Zsubsetrr$r$r%test_scatter3d_sortingsR   rArcc Csddg}ddg}ddg}ddg}|jdd}|j|||d|d |jd|dd |jdd}|j|d d d|d d d|d d dd|d d dd |jd|dd d S) z Test that the draw order does not depend on the data point order. For the given viewing angle at azim=-50, the yellow marker should be in front. For azim=130, the blue marker should be in front. rr*rrlrPrr r rpraN)rSrrgr!r"rcrOrPrQrvr#r$r$r%$test_marker_draw_order_data_reverseds   <rGc Csd}ddg}ddg}ddg}ddg}|jdd}||j|||d |d |jd|dd |jdd}||j|||d |d d dd |jd|d dd d S)z Test that the draw order changes with the direction. If we rotate *azim* by 180 degrees and exchange the colors, the plot plot should look the same again. rCrr*rrlrPrrrDrEraNr_)rS set_axis_offrrgrFr$r$r%#test_marker_draw_order_view_rotateds  rIzplot_3d_from_2d.pnggQ?cCsTt}|jdd}tdd}tdd}|j||ddd|j||ddddS) Nrrrr+rorOrsrtrP)r1rRrSr3rxr)rAr#rzr{r$r$r%test_plot_3d_from_2ds    rKz surface3d.pngc Csdtjd<t}|jdd}tddd}tddd}t||\}}t||}t|}|j |||ddt j d dd }| d d |j |d dddS)NFzpcolormesh.snaprrrr+rrr)rrrr5r)\()\(?r)shrinkaspect)r1rcParamsrRrSr3rxrrrrr rrcolorbar)rAr#rrrrZsurfr$r$r%test_surface3d(s     rRzsurface3d_shaded.pngc Cst}|jdd}tddd}tddd}t||\}}t|d|d}t|}|j|||dddddgddd | d d dS) Nrrrr+rr,r*F)rstridecstridervr5rrLrM) r1rRrSr3rxrsqrtrrr)rAr#rrrrr$r$r%test_surface3d_shaded:s  rVzsurface3d_masked.pngcCsvt}|jdd}ddddddd d d d d g }ddddddd d g}t||\}}tdddddddddddg dddddddddddg dddddd dddddg ddd d d dd d d ddg ddd d d dd d d d dg ddddddddd ddg dddddddddddg dddddddddddg g}tj|d}tj | | d}t j d||}|j||||d|ddddS)Nrrr*r,r.r/r+rrrr,rorIrrg rnr*)rrZplasmar?rmr)r1rRrSr3rr4r: masked_lessr Normalizerrrrrrg)rAr#rOrPmatrixrQnormr r$r$r%test_surface3d_maskedHs*  rccCs`ttdtd\}}||}|jdd}|j|||dd|jdd}||||dS)Nr+rrr^)r3rrxrSr)r!r"rOrPrQrrr$r$r%test_plot_surface_None_argcs   rdzsurface3d_masked_strides.pngcCsjt}|jdd}tjddddddf\}}tj||d}|j|||ddd|d d d dS) Nrrigffffff@r*r,r/rSrT<r) r1rRrSr3r6r:r_rrgrr$r$r%test_surface3d_masked_stridesms  rhz text3d.pngc Cst}|jdd}d}d}d}d}t||||D].\}}}} d||| |f} |||| | |q2|jdddd d d |jd d d |jd|dd|dd| dd| d| d| ddS)NrrNrOrPrQ)r*r*rr*r*r*r,rr/r,rr,rr/rrr,r,r/r,r+rr*rz(%d, %d, %d), dir=%sr*r~rrgffffff?z2D Text) transformrrozX axiszY axiszZ axis) r1rRrSr<r Ztext2DZ transAxes set_xlim3d set_ylim3drrUrVrW) rAr#zdirsrzr{rsrtrOrPrQrMr$r$r% test_text3dys"      rrc Csd}d}d}d}|jdd}|dd|dd|ddt||||D]H\}}} } |dddd |d | d | d |} | j|| | f|d qN|jdd} | dd| dd| ddt||||D]8\}}} } | j|| | d |d | d | d ||d qdS) Nrirkrlrmrrrro(z, z), dir=)rt)rSrorprr<r Zset_position_3d) r"r!rqrzr{rsrrtrOrPrQrrr$r$r%test_text3d_modifications"    (    rtz trisurf3d.pnggZd;?c Csd}d}tdd|}tjddtj|dd}tj|d tjf|d d }|ddd ddftj|7<td|t|}td|t|}t| |}t }|j d d }|j |||t jdddS)N$r??rr,FZendpoint.r*axisrr皙?)rr)r3rrr<newaxisappendrr-rr1rRrS plot_trisurfr Zjet Zn_anglesZn_radiiZradiiZanglesrOrPrQrAr#r$r$r%test_trisurf3ds$ rztrisurf3d_shaded.pngQ?c Csd}d}tdd|}tjddtj|dd}tj|d tjf|d d }|ddd ddftj|7<td|t|}td|t|}t| |}t }|j d d }|j |||d ddgdddS)Nrurrvrwrr,Frx.r*ryrrrr{)rvr) r3rrr<r|r}rr-rr1rRrSr~rr$r$r%test_trisurf3d_shadeds$ rzwireframe3d.pngcCs<t}|jdd}td\}}}|j|||ddddS)Nrrr rrr1rRrSrrplot_wireframerr$r$r%test_wireframe3ds rzwireframe3dzerocstride.pngcCs<t}|jdd}td\}}}|j|||ddddS)Nrrrrrrrrr$r$r%test_wireframe3dzerocstrides rzwireframe3dzerorstride.pngcCs<t}|jdd}td\}}}|j|||ddddS)Nrrrrrorerrr$r$r%test_wireframe3dzerorstrides rc CsRt}|jdd}td\}}}tt|j|||dddW5QRXdS)Nrrrrre) r1rRrSrrrr ValueErrorrrr$r$r% test_wireframe3dzerostrideraisess   rc Cs|t}|jdd}td\}}}tt|j|||dddW5QRXtt|j |||dddW5QRXdS)Nrrrro2)rSr)rTr) r1rRrSrrrrrrrrr$r$r%test_mixedsamplesraisess   rz quiver3d.pngzquiver3d_pivot_middle.pngzquiver3d_pivot_tail.pngc Cstjdddddddddf\}}}ttj|ttj|ttj|}ttj| ttj|ttj|}dttj|ttj|ttj|}dD]0}tjdd }|j||||||d |d d qdS) Nrrr$@333333?@>, p ?)tipmiddletailrrrTrpivot normalize) r3ogridrrrr1rRrSquiver)rOrPrQuvwrr#r$r$r% test_quiver3ds(024rc CsP|jddg}}}}}}|jdd}|j||||||dddd dS)NrrrrTr)rSr) r!r"rOrPrQrrrr#r$r$r%test_quiver3d_emptys  rzquiver3d_masked.pngc Cs.t}|jdd}tjdddddddddf\}}}ttj|ttj|ttj|}ttj| ttj|ttj|}dttj|ttj|ttj|}tjj d |k|d k@|d d }tjj d |k|d k@|d d }|j ||||||d ddd dS)Nrrrrrrrrr皙ٿrF)rr rTr) r1rRrSr3r6rrrr:Z masked_wherer)rAr#rOrPrQrrrr$r$r%test_quiver3d_masked s (024rcCspt}|jdd}td}||t||dt | dsNt |j t | dslt dS)Nrrrr)rwrrr*)r*rrr*)r1rRrSr add_patchrZpatch_2d_to_3dr#rZ same_color get_facecolorrYrr)rAr#circler$r$r%test_patch_modifications     rcCstdd}tdd}tddddgddddgg}tj||gdd }|jd d }|||d ||| d | st | d| rt tdd}tdd}tddddgddddgg}tj||gdd |d dd}|jd d }||dS)Nrr)rrrrrrwr.)rrrrr F)rrrrwr") rr3r4rZPatch3DCollectionrSadd_collection3dr$r#r%r&rYr')r!r"Zpatch1Zpatch2r?rprrr$r$r%"test_patch_collection_modification+s.              rc Cs~dddgdddgdddgdddgg}tjtddt|W5QRXtj|td}tjtddt|W5QRXdS)Nrr*zlist of \(N, 3\) array-likerr2)rrrrPoly3DCollectionr3r4r;)polyr$r$r%&test_poly3dcollection_verts_validationGs $rzpoly3dcollection_closed.pngcCst}|jdd}tdddgdddgdddggt}tdddgdddgdddggt}tj|gddddd }tj|gddd d d }||||dS) Nrrrr*r.k)rrr*rTrrrclosed)r*rrrF) r1rRrSr3r4r;rrrrAr#Zpoly1Zpoly2c1c2r$r$r%test_poly3dcollection_closedQs $$   rcCsvtg}t|t|tjs"t|gks2ttjt ddd\}}| || }t |sht|jdS)Nrrr)rrZpoly_collection_2d_to_3d isinstancerrY get_pathsr1rr add_artistZdo_3d_projectionr3isnanrr)rrAr#Zminzr$r$r%#test_poly_collection_2d_to_3d_empty`s  rzpoly3dcollection_alpha.pngcCst}|jdd}tdddgdddgdddggt}tdddgdddgdddggt}tj|gddddd }|d tj|gdd d }| d | d|d | || |dS)Nrrrr*r.rrrr*TrrF)rr)r*rr) r1rRrSr3r4r;rrr%r#r$rrr$r$r%test_poly3dcollection_alphaos $$      rzadd_collection3d_zs_array.pngc CsDtdtjdtjd}tddd}|dd}|t|}|t|}t|||gddd}tj|dd|ddgdd }t }|j d d }t d dtj} t |ddddddfd | d} | t|dtj|j| |dddddfd} | dk st|dd|dd|dddS)Nrr/rrr,r*rr.ryrrrtwilightrrbrsrr+r)r3rrrr column_stackr8 concatenater1rRrSr`r set_arraymodrrYrrr rrQrCrOrPZpointssegmentsrAr#rblcrr$r$r%test_add_collection3d_zs_arrays" " $   rzadd_collection3d_zs_scalar.pngc Cstddtjd}d}|dd}|t|}|t|}t||gddd}tj|dd|ddgdd}t }|j dd}t ddtj} t |d | d } | ||j| |d } | dk st|d d |dd|dddS)Nrr,rr*rryrrrrrrr+rr)r3rrrrrr8rr1rRrSr`rrrrYrrrrr$r$r%test_add_collection3d_zs_scalars" "     rzaxes3d_labelpad.pngcCst}|t|}|jjtjdks,t|j ddd|jjdksJt| d|j ddd|j jdksrt| dkstd|j_d|j _t|jD]\}}|||d qdS) Nz axes.labelpadzX LABELro)labelpadzY LABELzZ LABELrnrr+)r1rRZadd_axesrxaxisrrrPrYrUrVrWzaxisZ get_zlabelyaxisr=get_major_ticksZset_padZget_pad)rAr#rGtickr$r$r%test_axes3d_labelpads rzaxes3d_cla.pngcCs.t}|jddddd}||dS)Nr*rr)r1rRrSrHclarZr$r$r%test_axes3d_clasrzaxes3d_rotated.pngcCs,t}|jddddd}|ddddS)Nr*rrr\-r)r1rRrSrgrZr$r$r%test_axes3d_rotatedsrc Cstjdddd}tjdddd}t||\}}tjd}tjdd}|jdddd d }t t | |||W5QRXdS) Nrror)num)rrr*r,rr) r3rrrrandnr1rRrSrrrr)rOrPrrrQrAr#r$r$r%test_plotsurface_1d_raisess   rc Csxtdddg}tdddg}tdddg}d}t||||\}}}t||||}tddd}t||} | S)Nriirrr*r)r3r4r _view_axes_view_transformation_uvwZpersp_transformationdot) ErVrdrrrviewMZperspMMr$r$r%_test_proj_make_Ms rc Cst}tddddddddddg d}tddddddddddg d}tddddddddddg d}t||||\}}}t||||\}}} tj||tj||tj| |dS)Nrr*r@)rr3r4rproj_transformZ inv_transformrZassert_almost_equal) rrzr{rstxstystzsZixsZiysZizsr$r$r%test_proj_transforms"""rr*cOsd|ddg}dd|dg}ddd|g}t||||\}}} t||\} } } } | | f| | f| | fg}tj||\}} t|}| |t||ddddgD]\}}}| |||q|| fS)NrrrOrPrQ)rrr<r1rrZadd_collectionr )rr argsrrzr{rsrrrrr#ZayazlinesrAZlinecrOrPrr$r$r%_test_proj_draw_axess    rzproj3d_axes_cube.pngc Cst}d}tddddddddddg d}tddddddddddg d}tddddddddddg d}t||||\}}}t|dd\}} | j|||d| j||ddt |||D]\} } } | | | | q| d d | d d dS) N0 1 2 3 0 4 5 6 7 4rr*rir rprC皙ɿr{) rsplitr3r4rrrrrr<r rr) rtsrzr{rsrrrrAr#rOrPrr$r$r%test_proj_axes_cubes""" rzproj3d_axes_cube_ortho.pngc Csntdddg}tdddg}tdddg}d}t||||\}}}t||||}tdd}t||} d} tddddddddddg d} tddddddddddg d} tddddddddddg d} t| | | | \}}}t | dd\}}|j ||d |d|j ||d d t ||| D]\}}}| |||q6|d d|d ddS) Nrrr*rrr rrCri8)r3r4rrrZortho_transformationrrrrrrr<r rr)rrrrdrrrrZorthoMrrrzr{rsrrrrAr#rOrPrr$r$r%test_proj_axes_cube_ortho-s(  """ rcCsvddddg}t|tjd}tj|ddddgddddg}t|tjd}tj|dtddddgdS)Nr*rrr.r,r)rZrot_xr3rrassert_allcloserU)rZ rotated_Vr$r$r%test_rotJs   rc Csfd\}}d\}}d\}}t||||||}tj|ddddgddddgddd d gdddd ggdS) N)rx)rr)rr{rrrg{Gzt?rg$@rr*)rZworld_transformationr3rr)ZxminZxmaxZyminZymaxzminzmaxrr$r$r% test_worldTs    rzproj3d_lines_dists.pngc Cstjdtddd\}}d}d}|||t||\}}d}d}|||t|d |d f||}tt ||fj ||}||d kst t|||D]&\}} } t || f| d d } | | q|d d |d d dS)N)r/rr))rO)rJr)rrm)rnr)rrrnrm)rrrmrr)fillrBrr)r1rrrr<rr_line2d_seg_distr3r4TrYrrrr) rAr#rzr{p0p1Zdist0distrOrPdrpr$r$r%test_lines_dists`s     rcCs.d}d}t|||tt|||dS)N)rormr)rnrr_)rrr3r4)s0pr$r$r%test_lines_dists_nowarningysrcCstjddid\}}|dks$t|jdddd|dd gdd gdd g|d ks^t|d |d |dd gdd gdd g|dkst|jdd|dd gdd gdd g|dkstdS)NrrrZlinearrrr{rOrPrQr*)rr*r皙?rg333333?FTr,)rr*rrr333333@rOry)rr,rrrr) r1rZ get_zscalerYmarginsrZ get_w_limsZ autoscaleZset_autoscalez_onrZr$r$r%test_autoscales   rrzrr()TFNc Cst}|jdd}td}tddd}|||t|d|d}t|d|d }t|d |d }|dkr||n|} |d |d | |kst|j tj |d dS) NrrrrrZ get_autoscaleZ_onset_Zlimget_)r)r() r1rRrSr3rxrrgetattrrYrrrr) rzr(rAr#rOrPZget_autoscale_onZset_limZget_limZ post_autor$r$r%test_unautoscales     r c Cs`t}|jdd}tt|jdddW5QRXtt|jdddW5QRXdS)NrrpersprZ focal_lengthr`r*)r1rRrSrrrrfrZr$r$r%test_axes3d_focal_length_checkss    r zaxes3d_focal_length.pngcCsBtjddddid\}}|djdtjd|djdd ddS) Nr*r,rrrrr r 333333?)r1rrfr3inf)rArBr$r$r%test_axes3d_focal_lengthsrzaxes3d_ortho.pngcCs"t}|jdd}|ddS)Nrrr`)r1rRrSrfrZr$r$r%test_axes3d_orthos rzaxes3d_isometric.pngc Csddlm}m}tjtddddd\}}d}|tt||||d D]>\}}t || |d |dkrJ|j t ||d d iqJ|j ttd td ddd|ddS)Nr)r6r8rr`)r/r/r/)rZ proj_typeZ box_aspectr)rr*r,r*rprrwrgraT)r5r6r8r1rrr3r4r7r:r9r;r<rgdegreesZarctanrUr)r6r8rAr#rCr er$r$r%test_axes3d_isometrics$ &rvalue)setterside)roleft)roright)rpbottom)rptop)rr)rrc CsF||i}t}|jdd}ttt||f|W5QRXdS)Nrr)r1rRrSrrrr)rrrlimitrAobjr$r$r%test_invalid_axes_limitss   rc@seZdZedgddZedgddZedgdd Zed gd d Zed gddZedgdddddZ ddZ dS) TestVoxelszvoxels-simple.pngcCsBtjddid\}}td\}}}||k||kB}||dS)Nrrr)r+r/r.)r1rr3indicesvoxels)selfrAr#rOrPrQr r$r$r% test_simpleszTestVoxels.test_simplezvoxels-edge-style.pngcCsvtjddid\}}td\}}}|dd|dd|dddk}|j|dd d }|t|d dS) Nrrr)r+r+r/r,g?g](\@r.r)rrr)r1rr3rr rkeysr$)r!rAr#rOrPrQr rr$r$r%test_edge_styles (zTestVoxels.test_edge_stylezvoxels-named-colors.pngcCstjddid\}}td\}}}||k||kB}||||dk@}tjddtjd}d||d k|d k@<d |||d k<|j||d d S)z5Test with colors set to a 3D object array of strings.rrrrororor*r+r2z0.25r+cyanror^N)r1rr3rr(Zobject_r r!rAr#rOrPrQr r r$r$r%test_named_colorsszTestVoxels.test_named_colorszvoxels-rgb-data.pngcCsttjddid\}}td\}}}||k||kB}td}|d|d<|d|d<|d|d <|j||d d S) z5Test with colors set to a 4d float array of rgb data.rrrr%)rororor.r,.r.r*.r,r^N)r1rr3rzerosr r'r$r$r% test_rgb_datas    zTestVoxels.test_rgb_datazvoxels-alpha.pngc Cstjddid\}}td\}}}||k}t||dk}||B}td} dddd g| |<dddd g| |<|j|| d } t| tkst | D](\} } || st d t | t j st qdS) Nrrrr%r,)rororor/r*rrr^zfaces returned for absent voxel)r1rr3rr:r,r typerrYitemsrrr) r!rAr#rOrPrQZv1Zv2r r rZcoordrr$r$r% test_alpha s zTestVoxels.test_alphazvoxels-xyz.png{Gz?F)rrc Cstjddid\}}dd}tdd\}}}||}||}||} |dd |dd | dd d k} t| jd } || d <|| d <| | d<|j|||| | td | ddddddS)NrrrcSsZd}t|jD]F}||tjdd||tjddd}|tjdd7}q|S)Nr$rr*r)rndimr3Z index_exp)rOslrGr$r$r% midpoints"sz&TestVoxels.test_xyz..midpoints)r5r5g0@rr,r)r.r)r*r+rr*)r?r@r)r1rr3rr,r9r r) r!rAr#r4rCrkrlrcgcbcZspherer r$r$r%test_xyzs  ( zTestVoxels.test_xyzc Cstd\}}}td}tjddid\}}ttddfD]T}|j|f||jfd|i||j||||f||j|||fd|i|q>tjt d d |j|||||d W5QRXtjt d d |||W5QRXtt |j||||d W5QRXdS) Nr-)r,r.r/rrrr)rfilledr r)r:)r:rOrPrQ) r3rZonesr1rrr rr TypeErrorAttributeError)r!rOrPrQr:rAr#kwr$r$r%test_calling_conventions?s  z#TestVoxels.test_calling_conventionsN) __name__ __module__ __qualname__mpl3d_image_comparisonr"r$r(r-r0r9r>r$r$r$r%rs      rc Cs ddgddgddg}}}ddgd d gd d g}}}t}|jd d}||||}|d} tj|||f| | |||tj|||f| | || || j |ddtj|||f| | j dddtj||t |f| dS)Nrr*r,r.r/r+rrrr,rorIrrrQrJ) r1rRrSrr3rrZ get_data_3drZ set_xdataZ set_ydataZset_3d_propertiesr) rOrPrQZx2y2z2rAr#rrr$r$r%test_line3d_set_get_data_3dXs   rEcCsx|jdd}|ddddgddddgddddg||jdd}||ddddgddddgddddgdS)Nrrr*ro)rSrZ invert_yaxisr r$r$r% test_invertedjs  & rFcCstjddid\}}|r t|r,t|r8t|dd|dd|dd|sht|stt|st| |rt|rt|rtdS)Nrrrr*r) r1rZxaxis_invertedrYZyaxis_invertedzaxis_invertedrrrrrZr$r$r%test_inverted_clavs           rHcCst}t|}|jddd|jddd|jddd|j|jD]}|jj dksNt qN|j D]}|jj dksnt qn|j D]}|jj dkst qdS)NrOr~)rzr rPrQ) r1rRrZ tick_paramsrrrrZ tick1line_colorrYrr)rAr#rr$r$r%test_ax3d_tickcolours rJc Cs|jddddid}|jD]}|ddqt|dd d d gD]z\}}|d j|d d|dj|dd|dj|tjd d|dj|tjd d|dj|tjd dq>dd}|jddddid}|jD]}|ddqt|dgdgdgdddggD]\}}||d |D]}|dq"||d|D]}|dqB||d|D]}| tjd qb||d|D]}| tjd q||d|D]}| tjd qq dS) Nr/r+rrrgcAg@cArOrPrQrrplain)rzrr*)rr,)rzZ scilimitsr,zaxes.formatter.useoffset)rzZ useOffsetr.zaxes.formatter.use_locale)rzZ useLocalezaxes.formatter.use_mathtext)rzZ useMathTextcsfdd|DS)Ncsg|]}t|qSr$)rZget_major_formatter)rnamer#r$r%rszAtest_ticklabel_format..get_formatters..r$)r#namesr$rMr%get_formatterssz-test_ticklabel_format..get_formattersrrrF) rrerr<Zticklabel_formatrrPZset_scientificZset_powerlimitsZ set_useOffsetZ set_useLocaleZset_useMathText) r!r"rBr#rowrLrOrNfmtr$r$r%test_ticklabel_formatsX       rRc Csd}ttdddtdddtddd\}}}t|}}}t||fdD].\} } | jdd} | j||||||| |d qVdS) Nrgr*r{rr)r*rwrr)rr)r3rrx ones_liker<rSr) r!r"rrOrPrQrrrrArr#r$r$r%test_quiver3D_smokes    rTzminor_ticks.pngZmpl20)rcCsttjdd}|jdgdd|jdgdd|jdgdd|jdgdd|jd gdd|jd gdddS) NrrrT)minorZquartergQ?thirdrZhalf) r1rRrSZ set_xticksZset_xticklabelsZ set_yticksZset_yticklabels set_zticksZset_zticklabelsrMr$r$r%test_minor_tickssrXzerrorbar3d_errorevery.pngc Cstddtjdd}t|td|td|}}}t}|jdd}d }t|j}||dk||ddk@}||dk||ddk@} |j |||d || |d d S) z0Tests errorevery functionality for 3D errorbars.rr,rr1r.r+rrr{)zuplimszlolimsZ erroreveryN) r3rxrrrr1rRrSr7errorbar) rrOrPrQrAr#ZesteprGrZr[r$r$r%test_errorbar3d_erroreverys*  r]zerrorbar3d.pngcCsrt}|jdd}dddddg}dddddg}|j||||||dd d d d d gd d d d d gd d d d |dS)z9Tests limits, color styling, and legend for 3D errorbars.rrr*r,r.r/r+rFTpurplez Error lines) rOrPrQZxerrZyerrZzerrZcapsizerZr[ZyuplimsZecolorrMN)r1rRrSr\Zlegend)rAr#rrr$r$r%test_errorbar3ds   r_z stem3d.pngg~jth?)rrc Cs2tjddddddid\}}tddtj}t|tjd}t|tjd}|}t|dd d d gD]*\}}|j||||d | d |qnttj dtjdd}t |}t|}t|dd d d gD]N\}}|j|||ddd|d\}} } | d ||j ddd| dqdS)Nr,r.)rrTrrrJZconstrained_layoutrrrOrPrQ) orientationz orientation=rnr*zC4-.ZC1Dr)linefmtZ markerfmtZbasefmtrar)rZmarkeredgewidth) r1rr3rrrrr<stemrTrSsetZ set_linewidth) rArBrrOrPrQr#rtZ markerlineZ stemlinesZbaseliner$r$r% test_stem3ds6     rezequal_box_aspect.pngc Csddlm}m}t}|jdd}tddtjd}tdtjd}t t |t |}t t |t |}t t |t |}| |||ddg} |tt|| | | dD]B\} } tt| | | d| dkr|jt| | d d iqt|||g} t| dt| dg} || || || |d |d tj t!d d|jd ddW5QRXdS)Nr)r8r6rrr,rrr*rvrloffrjzArgument zoom =r)zoom)"r5r8r6r1rRrSr3rrouterrrrSrr4r7r9r:r;r<r get_xlim3d get_ylim3d get_zlim3drrrorprrzr>rrr)r8r6rAr#rrrOrPrQrCr rZxyzlimZXYZlimr$r$r%test_equal_box_aspects0 $$     rlcCsd}tjd|ddddid\}}|D]*}|tjdtjdtjd}q$tj||d d }|j|j j dd kst dS) Nr,r*)r/r+Trrr`r+ horizontal)r#rar{) r1rr~r3rrrQrrr#Z get_positionZextentsrY)Z num_plotsrArBr#Zp_triZcbarr$r$r%test_colorbar_posDs    rncCs"t}|jdd}|r t|dks0t|dks@t|d|rVt|dksft|dksvt||st|dkst|dkst|jdd|st|dkst|dkst|jd d |st|d ks t|d kstdS) Nrrrr*rr,)r,rr*)upper)r*rr,)lower)r,r*r*r,) r1rRrSrGrYget_zlimZ get_zboundZ set_zboundZ invert_zaxisrZr$r$r%test_inverted_zaxisTs*         ruc Cst}|jdd}|dks$t|jdd|dks@t|jdd|d ks\ttjtd d |jd dd W5QRXtjtdd |jd ddW5QRXdS)Nrrror,)rrpr*)rrsz$Cannot pass both 'bottom' and 'zmin'rr)rrz!Cannot pass both 'top' and 'zmax')rr) r1rRrSrtrYrrrr;rZr$r$r% test_set_zlimts"   rvcCst}|jddd}|jdd|d}|ddgddgddg|ddgddgddg|d ddd g|d ks|t|d kstdS) Nrr)rZsharezrr*r,r@)rry)r1rRrSrrWrtrY)rArrr$r$r%test_shared_axes_reticksrzcCs8dd}tjdd}|ddd|jj||\}}||\}}||\}}| t |t j ddd| t |t j ddd||\}} ||\} } ||\} } | t|kst| t|kst| t|kst|t|ks t| t|ks t| t|ks4tdS) z1Test mouse panning using the middle mouse button.cSs||d}||}||fS)z+Convert min/max limits to center and range.r,r$)ZdminZdmaxcenterZrange_r$r$r% convert_lims ztest_pan..convert_limrrr)buttonZxdataZydatar*N)r1rRrSrrrrirjrkZ _button_pressrrZMIDDLEZ_on_moverapproxrY)r|r#Z x_center0Zx_range0Z y_center0Zy_range0Z z_center0Zz_range0Zx_centerZx_rangeZy_centerZy_rangeZz_centerZz_ranger$r$r%test_pans, rztool,button,key,expectedrg))rQ?)r1gQ?)r{Gz?rO))g{GzrgQrQrrP))Q뱿g{Gz?r)rg)\(?))g ףp= r )rg(\?)rg ףp= ?pan))gffffff(\)g{GzgQgRQgffffff))gQgzG)rgq= ףpݿr))r{g{Gz?)gRQg(\ؿrcCsjt}|jdd}|ddd|j|||}}}d} d} |j |  t } |j |  t } t d|jf| |fd|i} t d|jf| |fd|i}t|j}|d kr||| ||||n&||| |||||\}}}|tj|d d ks2t|tj|d d ksNt|tj|d d ksjt||t|kst|t|kst|t|kst||tj|d d kst|tj|d d kst|tj|d d kst||t|ks6t|t|ksNt|t|ksftdS) Nrrrrr*r*Zbutton_press_eventkeyZbutton_release_eventrgr1)r:)r1rRrSrrrrirjrkZ transDatarnZastypeintrrrgZ press_zoomZ drag_zoomZ release_zoomrZ press_panZdrag_panZ release_panrr~rYbackforwardhome)Ztoolr}rexpectedrAr#Zxlim0Zylim0Zzlim0Zd0Zd1rs1Z start_eventZ stop_eventtbZxlimZylimZzlimr$r$r%test_toolbar_zoom_pansj          rc Cstttjtdddtdddtdddgj\}}}||}|jddd}|j||||ddd }|j | |jddd}|j||||ddd } dS) Nrr+r*orrrr)rpr r) r3r4r7r5r8rxrrSrrrchanged) r!r"rOrPrQrprZsc_testrZsc_refr$r$r%test_scalarmap_updates   rcCsFt}|dd}|djddddd}|djdddddd}dS)Nr*r,rrrother)rrM)r1rRZ subfiguresrS)rAZsfr#r$r$r%test_subfigure_simples rcomputed_zorder)ZclassicZ_classic_test_patch)baseline_imagesrrrc Cst}|jddd}|jddd}d|_d}||fD]h}tj|gddd d }|||jd d d d dd|jdddd dd| d| d| dq6|jddd}|jddd}d|_d}t | |f| |f\} } t d } d} t | |fd|f\} }|| }t | |f| df\}}|| }d}d}t ddt j|}|t ||t || |t |}}}||fD]}|j| ||ddddd|j||dk||dk||dkddd dd!|j| | | d ddd d|j|t ||t |t |ddddd!|j| ||dddd"d|j||dk||dk||dkddd d#d!|jd$d%dd&|d'qdS)(NrrF)r^)rr+r)r+r+r)r+rrwhiteblackr*)r?r@zorder)r,r,)rr/r~r,)rpr)r.r.)r*r.ro)rr)rryrrrrrr)rvrwrrr+z--r)r5Z linestylervrr.r/rnrrarf)r1rRrSrrrrrrrrrr3rr,rrrrrrgrz)rArrZcornersr#ZtriZax3Zax4dimrrrangleZX2ZY2ZZ2ZX3ZY3ZZ3rCrthrOrPrQr$r$r%test_computed_zorder"s      2  " $ $rcCst}|jdd}td}||t||j| dddksNt |j ddd|j| ddd kszt | |j| dddkst dS) Nrrrorux=1.8066, y=1.0367, z=−0.0553rmrP)rd vertical_axisu"x=9.1651, y=−0.9215, z=−0.0359) r1rRrSr3rxrrrrZ format_coordrYrgrAr#rOr$r$r%test_format_coordus     rcCsNt}|jdd}td}||t||j| dksJt dS)Nrrro)FTF) r1rRrSr3rxrrrrZget_axis_positionrYrr$r$r%test_get_axis_positions    rcCst}|jdd}|d|dks.t|ddd|dksLt|jdd |d ksht|jdd |d kst|jdd |dkstdS)Nrrr{)r{r{r{rr)rr{rr)rO)rr{r)rP)rrrrQ)rrr)r1rRrSrrYrZr$r$r% test_marginss     rzerr, args, kwargs, match)rz!margin must be greater than -0\.5)r*rr*)r*r*rrrQ)r*z'Cannot pass both positional and keyword)rOrPrz*Must pass a single positional argument forc Cs>tj||d&t}|jdd}|j||W5QRXdS)Nrrr)rrr1rRrSr)errrrrrAr#r$r$r%test_margins_errorss rcCsz|jdd}tddd}tj|dd|||dks@t|jdd}tdddd}|||dksvtdS)NrrrzFoo bar $\int$r*rr)rSrrZ text_2d_to_3drZget_position_3drYZText3D)r!r"r#txtZt3dr$r$r% test_text_3ds     rcCs>t}|jdd}|dgdggdgdggdgdggdS)Nrrrr*)r1rRrSrrZr$r$r%test_draw_single_lines_from_Nx1s rcCs`|jdd}t}t|}tj|ddd|||jdd}tj|ddd}||dS)Nrr)rrr r*rrPrQrtrJ)rSrZunit_rectanglerrpathpatch_2d_to_3drZ PathPatch3D)r!r"r#pathpatchZpp3dr$r$r%test_pathpatch_3ds   rzscatter_spiral.png)rrrcCsft}|jdd}tddtjdd}|jt|t||d|d|dd }|j dS) Nrrrr,rr*r+rE) r1rRrSr3rrrrrrr)rAr#rscr$r$r%test_scatter_spirals  ,rcCsntjddddddf\}}t|dt|d}t}|jddd}|j|||d d }|dS Nr*royY@r.r,rrrZhot)r) r3rrrr1rRrSrrrPrOrDrAr#rCr$r$r%#test_Poly3DCollection_get_facecolors rcCsntjddddddf\}}t|dt|d}t}|jddd}|j|||d d }|dSr) r3rrrr1rRrSrZ get_edgecolorrr$r$r%#test_Poly3DCollection_get_edgecolors rzDvertical_axis, proj_expected, axis_lines_expected, tickdirs_expectedrg#k$I?g._x%Ig(m?g@mۿg$g#k$IgW$%@g˗ì?g2m 3?g溾IgMg2m 3gM?g˗ìg溾I?c Csd}tjddddd}|jddd|d|jj|}tjj |||dt | | | gD]J\}}||} |j} tjj | | |d||} |} tj| | qhdS) a Test the actual projection, axis lines and ticks matches expected values. Parameters ---------- vertical_axis : str Axis to align vertically. proj_expected : ndarray Expected values from ax.get_proj(). axis_lines_expected : tuple of arrays Edgepoints of the axis line. Expected values retrieved according to ``ax.get_[xyz]axis().line.get_data()``. tickdirs_expected : list of int indexes indicating which axis to create a tick line along. g>r*rrr)rbrcrdr)rtolN)r1subplotrgrRrrZget_projr3rrr=Z get_xaxisZ get_yaxisZ get_zaxisrget_dataZ _get_tickdirr) rZ proj_expectedZaxis_lines_expectedZtickdirs_expectedrr#Z proj_actualrGrzZaxis_line_expectedZaxis_line_actualZtickdir_expectedZtickdir_actualr$r$r%test_view_init_vertical_axissD "  rzarc_pathpatch.pngcCsFtjddddd}tjdddddd d }||tj|d d d dS)Nr*rr)rrrg?rnrorC)rheightrZtheta1Ztheta2rrQr)r1rmpatchZArcrrr)r#r/r$r$r%test_arc_pathpatchSs  rzpanecolor_rcparams.pngc Cs:tddddtjdd}|jddW5QRXdS) NrCrkrl)zaxes3d.xaxis.panecolorzaxes3d.yaxis.panecolorzaxes3d.zaxis.panecolorrrrr)r1Z rc_contextrRrS)rAr$r$r%test_panecolor_rcparams^s   rcCs|jddd}dddg}dddg}dddg}||||ddddg|d d <dddg|d d <|jddd}dddg}dddg}dddg}||||dd S) z~ Test to see if the `z` axis does not get mutated after a call to `Axes3D.plot` test cases came from GH#8990 rrrr*r,r.rzo-Nr)r!r"rrOrPrQrr$r$r%"test_mutating_input_arrays_y_and_zis      r)r*) functoolsr5rZmpl_toolkits.mplot3drrrrZ matplotlibrZmatplotlib.backend_basesrrrr r rr rZmatplotlib.testing.decoratorsr r Zmatplotlib.testing.widgetsrZmatplotlib.collectionsrrZmatplotlib.patchesrrZmatplotlib.pathrZmatplotlib.textrZmatplotlib.pyplotZpyplotr1numpyr3partialrBr&rHrLr[rjr}rrrrrrrrmarkZ parametrizerrrrrrrr rrrrr)rArGrIrKrRrVrcrdrhrrrtrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrrnanrrrErFrHrJrRrTrXr]r_rerlrnrurvrzrLEFTRIGHTrrcontextrrrrrrrtupler;rrrrrrrrrrrr$r$r$r%s|                                      6                                                        z    '    %   &        =   Q        2(