o Gf<@sTddlZddlmZddlZddlZddlmZddl m Z m Z e dgdddd d Z e d gdd dd ddZ e dgddZddZe ddZe ddZe ddZe dgddddZe dgddd d!Ze d"gddd#d$Ze d%gddd&d'Ze d(gddd)d*Ze d+gddd,d-Ze d.gddd/d0Ze d1gddd2d3Ze d4gd5d6d7Ze d4gd5d8d9Zd:d;Zdd?Zd@dAZ dBdCZ!dDdEZ"e d4gd5dFdGZ#e d4gd5dHdIZ$dJdKZ%dLdMZ&dNdOZ'dPdQZ(e d4gd5dRdSZ)dTdUZ*dVdWZ+dXdYZ,e dZgddd[d\Z-d]d^Z.dS)_N)assert_allclose)pyplot)image_comparisoncheck_figures_equalZ polar_axesdefaultg~jt?)styletolc Cstddd}dtj|}t}|jdd}|j||ddd \}|jd d d d d \}d}||||}}||g|gd|jd||fddtdddddd|j ddddddS)N?MbP?@Tpolarz#ee8d18)ZcolorZlwrrrz#0000ffr oza polar annotation)皙?rzfigure fractionblackr facecolorZshrinkleftbaseline)xyxytext textcoords arrowpropshorizontalalignmentverticalalignmentxout)Zaxistick1Ontick2On direction) nparangepipltfigure add_subplotplotannotatedict tick_params)rthetafigaxlineZindZthisrZ thisthetar5;lib/python3.10/site-packages/matplotlib/tests/test_polar.pytest_polar_annotations s$    r7Z polar_coordsT)rZ remove_textrc Cstjjdddddd}t}|jdd}||||j|j d t j d d ft j d d fddt ddddddd | dd|dddS)Nr r0?)ralphaZequal)Zaspectzthe top@$@rg4@rrrrrrT)rrZxycoordsrrrr Zclip_oni)mplZpatchesZEllipser)r*r+Z add_artistZ set_clip_boxZbboxr-r&r(r.Zset_xlimset_ylim)elr2r3r5r5r6test_polar_coord_annotations,s"       rAzpolar_alignment.pngcCstddd}gd}t}gd}|j|ddd}|||j|dd d}|jd td D]}|j |j |||d d dq2dS)NrihZ)r皙?皙?g333333?皙?r皙?rGrErET horizontal)rZlabelverticalFrtop)Zanglerr ) r&r'r)r*add_axesZset_thetagridsZpatchZ set_visiblerangeaxesZ set_rgrids)ZanglesZ grid_valuesr2ZrectrHrIir5r5r6test_polar_alignmentIs    rPcCsJt}tddgddgtddgddgt|jdks#Jd dS) NrrJrGrCrg333333?rDz!More than one polar axes created.)r)r*rlenrN)r2r5r5r6test_polar_twice^srScCs|jdd}|tddgddg|tddgddg|jdd}|tdd gddg|tdd gddgdS) Nr projectioniMrCrGrJif)r+r,r&deg2radfig_testfig_refr3r5r5r6test_polar_wrapes  r]cs~ddlmmgd}gd}t|jtfdd|D||jdd}| t |||j dd dS) Nrg>@gF@gN@V@r r<@r cg|]}|jqSr5deg.0r!Zunitsr5r6 wz&test_polar_units_1..rrTrd)xlabel) matplotlib.testing.jpl_unitstesting jpl_unitsregisterr)r*numberrr+r,r&rYset)r[r\xsysr3r5rgr6test_polar_units_1os  rscsddlmmgd}fdd|D}gd}fdd|D}t|jtj||ddd tt j j sCJ|j d d }|t|||j tjd j|jddd dS)Nrr^crbr5rcrergr5r6rhriz&test_polar_units_2..r`crbr5)km)rfyrgr5r6rhriZradrt)Z thetaunitsZrunitsrrTz{:.12})rjZylabel)rkrlrmrnr)r*ror isinstancegcaxaxisZget_major_formatterZUnitDblFormatterr+r,r&rYZset_major_formatterr>ZtickerZ FuncFormatterformatrp)r[r\rqZxs_degrrZys_kmr3r5rgr6test_polar_units_2~s  rzZ polar_rmin)rcCsZtddd}dtj|}t}|jgddd}||||d|d dS) Nrra{Gz?rJrFTr r<r: r&r'r(r)r*rLr,set_rmaxset_rminr0r1r2r3r5r5r6test_polar_rmin  rZpolar_negative_rmincCsZtddd}dtj|}t}|jgddd}||||d|ddS)Ngr r{rJrFTr r|rr5r5r6test_polar_negative_rminrrZ polar_rorigincCsdtddd}dtj|}t}|jgddd}||||d|d | d dS) Nrrar{rJrFTr r<r:r ) r&r'r(r)r*rLr,r}r~ set_roriginrr5r5r6test_polar_rorigins   rzpolar_invertedylim.pngcCs*t}|jgddd}|dddS)NrFTr rJr)r)r*rLr?r2r3r5r5r6test_polar_invertedylimsrzpolar_invertedylim_rorigin.pngcCsVt}|jgddd}|jd|jddgddgdd|d|ddS) NrFTr rrJnone)cr)r)r*rLyaxisZ set_invertedr,Zmarginsrrr5r5r6test_polar_invertedylim_rorigins   rZpolar_theta_positioncCs\tddd}dtj|}t}|jgddd}||||dd |d dS) Nrrar{rJrFTr ZNW clockwise) r&r'r(r)r*rLr,Zset_theta_zero_locationset_theta_directionrr5r5r6test_polar_theta_positions  rZpolar_rlabel_positioncCs.t}|jdd}|d|jdddS)NrrTi;auto)rotation)r)r*r+Zset_rlabel_positionr/rr5r5r6test_polar_rlabel_positions  rZpolar_theta_wedgec Cs tddd}dtj|}tddd}tddd}d }tjt|t|d d id d \}}t|D]Z\}}t|D]Q\} } ||| f} | |||| krZ| || | n| | | || d| j d d ||t|dd| j j d dd| jjjgddq;q3dS)Nrrar{rJg.@gv@r_gI@)r"inZinoutrT)) subplot_kwZfigsizerr)r#r$r%r)Zlabel2Onr)rrJg@r8)Zsteps)r&r'r(r)subplotsrR enumerater,Z set_thetaminZ set_thetamaxrr/rZset_tick_paramsrxZget_major_locatorbaseZ set_params) r0r1Z theta_minsZ theta_maxsZ DIRECTIONSr2axsrOstartjendr3r5r5r6test_polar_theta_limitss:         rZpng) extensionscCsP|jddid}|jdd|jdd|jddid}|d|ddS) NrTrr8)rKr:Zbottomr=rset_rlimr}r~rZr5r5r6test_polar_rlims    rcCsH|jddid}|jddgd|jddid}|d|ddS)NrTrr:r8rr=rrZr5r5r6test_polar_rlim_bottom s  rcCsDtjdd}|tdtdd|ddks JdS)NrrTr8r{r)r)r*r+r,r&r'get_ylimr3r5r5r6test_polar_rlim_zerosrcCsjtjddt}|dkr|dksJtdtt}|dkr1|dks3JdS)NrrTrrall)r)subplotrwget_rminZget_rmaxcloserrr5r5r6test_polar_no_datas   rcCs2tjddt}|d|dksJdS)NrrTlogr)r)rrw set_rscalerrr5r5r6test_polar_default_log_lims&s  rcCsJtjdd}tt|dWddS1swYdS)NrrTZdatalim)r)r*r+pytestraises ValueErrorZset_adjustablerr5r5r6!test_polar_not_datalim_adjustable-s  "rcCsvt}|jdd}|jddtj|jdd|j|j j dj dks,J|jj dj dks9JdS)NTr rC)r;F)Zvisibler) r)r*r+ZgridZsetprget_ticklabelscanvasdrawrx majorTicksZgridlineZ get_alpharr5r5r6test_polar_gridlines3s   rcCsFtjddid\}}|j||j}t|jgddddS)NrUrr)g yZ@g,ԚG=@g_)Gހ@gR!+|@r )Zrtol)r)rrrZ get_tightbboxZ get_rendererrZextents)r2r3Zbbr5r5r6test_get_tightbbox_polar?s    rcCs|jdddjdgdgdtjddd }|jd ddjdgdgd tjddd }|jdddjdgdgd tjddd }|jd ddjdgdgd tjddd }dS)NyrrTrrrrF)Z edgecolorZ antialiasedzrJrW)r+Zbarr&r()r[r\Zp1Zp2Zp3Zp4r5r5r6)test_polar_interpolation_steps_constant_rGs     rcCs`|jdddtjdgddg\}d|_|jddtdtjddtddddS)NrrTrrJrde)r+r,r&r(get_path_interpolation_stepslinspace)r[r\lr5r5r6)test_polar_interpolation_steps_variable_rUs $  rcCstjdd}|ddtj|jdddtjtdd |dd tjWdn1s1wYtjtdd |jdd dWddS1sPwYdS) NrrTrrJri)ZthetaminZthetamaxz+angle range must be less than a full circle)matchri)r)r set_thetalimr&r(rrrrr5r5r6test_thetalim_valid_invalid]s "rcCsjtjdd}|ddtt||fdksJ|dtt||fdks3JdS)NrrTrrr)rJr)r)rrtupler&ZradiansZ get_thetaminZ get_thetamaxrr5r5r6test_thetalim_argsis  " &rcCs|tjdddid\}}|D] }|dtjq|D]}t|j }t d|vs2Jt d|vs;JqdS)NzAAAABB.rUrrrrBr) r)subplot_mosaicvaluesrr&r(ZdegreesrxZget_majorticklocstolistrZapprox)r2rr3Zticklocsr5r5r6test_default_thetalocatorqs  rcCs4tjdd}|dtjd}|jdksJdS)NrrTrrQr)r)raxvspanr&r(rr)r3spanr5r5r6 test_axvspans rcCs`|jdddddid}dD] }|j|q |jdddddid}dD] }|j|q$dS)NrJTrUr)sharexr)rrr)shareyr)rZflatremove)r\r[rrOr5r5r6test_remove_shared_polars  rcCstjddddiddd\}}|j|djjdsJ|djjds+Jtjdddiddd\}}|j|d jjdsIJ|d jjdsUJdS) NrJrUrT)rrrrrzab cdb) r)rrrrxrZ get_visiblerr)r2rr5r5r6"test_shared_polar_keeps_ticklabelss   rcCsDtjdd}|dd|d|ddg|dks JdS) NrrTrrr:rGrC)rrC)r)rrZaxvliner,rrr5r5r6(test_axvline_axvspan_do_not_modify_rlimss   rcCstjdd}|dddksJ|dddksJ|ddd ks$J|ddd ks.J|ddd ks8J|ddd ksBJ|d ddksLJ|d ddksVJ|d ddks`JdS)NrrTrg{Gzt?uθ=0.0π (0°), r=0.005rGuθ=0.00π (0°), r=0.100ruθ=0.000π (0.0°), r=1.000uθ=0.3π (57°), r=0.005uθ=0.32π (57°), r=0.100uθ=0.318π (57.3°), r=1.000rJuθ=0.6π (115°), r=0.005uθ=0.64π (115°), r=0.100uθ=0.637π (114.6°), r=1.000)r)rZ format_coordrr5r5r6test_cursor_precisions rz polar_log.pngcCsXt}|jdd}|d|ddd}|tddtj|t dd|dS) NTr rrirrrJ) r)r*r+rrr,r&rr(Zlogspace)r2r3nr5r5r6test_polar_logs    *rcCsNt}|jdd}|tj tjdd|jD}|gdks%JdS)NrrTcSsg|]}|qSr5)Zget_text)rfrr5r5r6rhsz-test_polar_neg_theta_lims..)u-180°u-135°u-90°u-45°u0°u45°u90°u135°)r)r*r+rr&r(rxr)r2r3labelsr5r5r6test_polar_neg_theta_limss  r)/Znumpyr&Z numpy.testingrrZ matplotlibr>rr)Zmatplotlib.testing.decoratorsrrr7rArPrSr]rsrzrrrrrrrrrrrrrrrrrrrrrrrrrrrrr5r5r5r6s~    !