Ë
|
oñúhçA ã ó d Z ddlmZ ddlmZ ddlZddlmZ ddl Z
|
ddl m Z ddl m Z er ddlmZ ddlmZ d d Zd d
|
Zd Zd d Zd d d Zd Zd!dZd Zd"dZd#dZ d!dZ d$dZ d%dZ d dZ i fdZ!d&dZ"d Z#d Z$d'dZ%d Z&d Z'y)(zF
|
Module consolidating common testing functions for checking plotting.
|
é )Ú annotations)Ú TYPE_CHECKINGN)Ú is_list_like)ÚSeries)ÚSequence©ÚAxesc óä |r | t d« t | « } | D ]Q }|r;|j « J t |j « j « |« @|j « QJ y)a)
|
Check each axes has expected legend labels
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
labels : list-like
|
expected legend labels
|
visible : bool
|
expected legend visibility. labels are checked only when visible is
|
True
|
Nz-labels must be specified when visible is True)Ú
|
ValueErrorÚ_flatten_visibleÚ
|
get_legendÚ_check_text_labelsÚ get_texts)ÚaxesÚlabelsÚvisibleÚaxs úOH:\Change_password\venv_build\Lib\site-packages\pandas/tests/plotting/common.pyÚ_check_legend_labelsr sp ñ FNÜÐHÓIÐIÜ DÓ !DØò +Ù Ø==?Ð.Ð .Ð.Ü r}}× 8Ñ 8Ó :¸FÕ Cà==?Ñ*Ð *Ð*ñ +ó c óÄ |r | t d« |r8| j « \ }}|D cg c] }|j « }}||k( sJ y| j « J yc c}w )a
|
Check ax has expected legend markers
|
|
Parameters
|
----------
|
ax : matplotlib Axes object
|
expected_markers : list-like
|
expected legend markers
|
visible : bool
|
expected legend visibility. labels are checked only when visible is
|
True
|
Nz.Markers must be specified when visible is True)r Úget_legend_handles_labelsÚ
|
get_markerr )r Úexpected_markersr ÚhandlesÚ_ÚhandleÚmarkerss r Ú_check_legend_markerr / sp ñ Ð$Ð,ÜÐIÓJÐJÙØ×1Ñ1Ó3
|
Ø5<Ö=¨66×$Ñ$Õ&Ð=Ð=ØÐ*Ò*Ð*Ñ*à}}Ð&Ð&Ñ&ùò >s ©A c ó8 ddl m} | j « }|j « }t |« t |« k( sJ t ||« D ]; \ }}|j « }|j « }t j ||« = |j d« y)z
|
Check each axes has identical lines
|
|
Parameters
|
----------
|
xp : matplotlib Axes object
|
rs : matplotlib Axes object
|
r NÚall) Úmatplotlib.pyplotÚpyplotÚ get_linesÚlenÚzipÚ
|
get_xydataÚtmÚassert_almost_equalÚclose) ÚxpÚrsÚpltÚxp_linesÚrs_linesÚxplÚrslÚxpdataÚrsdatas r Ú _check_datar4 F s õ $à||~HØ||~Hä x=C MÒ )Ð)Ð )Ü (Ó+ò /SØ Ó!Ø Ó!Ü
|
× Ñ v vÕ.ð/ð
|
IIeÕr c ó| ddl m} t | |« st | « s| g} | D ] }|j « |k( rJ y)zà
|
Check each artist is visible or not
|
|
Parameters
|
----------
|
collections : matplotlib Artist or its list-like
|
target Artist or its list or collection
|
visible : bool
|
expected visibility
|
r )Ú
|
CollectionN)Úmatplotlib.collectionsr6 Ú
|
isinstancer Ú get_visible)Ú collectionsr r6 Úpatchs r Ú_check_visibler< ] sE õ 2ä k :Ô .´|ÀKÔ7PØ"m à ò .Ø× Ñ Ó" gÓ-Ð-Ð-ñ.r c ól t | « } | D ]$ }|j D ] }|j |k( rJ & y)z¹
|
Check for each artist whether it is filled or not
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
filled : bool
|
expected filling
|
N)r ÚpatchesÚfill)r Úfilledr r; s r Ú_check_patches_all_filledrA q sB ô DÓ !DØò (ØZZò (EØ:: Ó'Ð 'Ð'ñ (ñ(r c ó | j « }t t ||« « }| j D cg c] }|| c}S c c}w ©N)ÚuniqueÚdictr& Úvalues)ÚseriesÚcolorsrD ÚmappedÚvs r Ú_get_colors_mappedrK s< Ø ]]_Fô #ffÓ%Ó &FØ %}}Ö -!F1IÒ -Ð-ùÒ -s ´ Ac óR ddl m} ddlm}m}m} ddlm} |j } |Ê|t ||« }|dt | « }t | « t |« k( sJ t | |« D ] \ }
|
} t |
|
|« r"|
|
j « } | j | « } n;t |
|
||f« r t |
|
j! « d « } n|
|
j! « } | j | « } | | k( rJ |·|t ||« }|dt | « }t | « t |« k( sJ t | |« D ]r \ }
|
} t |
|
|« r|
|
j# « d } n|
|
j# « } t | t$ j& « r t | « } | j | « } | | k( rrJ yy)aÿ
|
Check each artist has expected line colors and face colors
|
|
Parameters
|
----------
|
collections : list-like
|
list or collection of target artist
|
linecolors : list-like which has the same length as collections
|
list of expected line colors
|
facecolors : list-like which has the same length as collections
|
list of expected face colors
|
mapping : Series
|
Series used for color grouping key
|
used for andrew_curves, parallel_coordinates, radviz test
|
r )rH )r6 ÚLineCollectionÚPolyCollection)ÚLine2DN)Ú
|
matplotlibrH r7 r6 rM rN Úmatplotlib.linesrO ÚColorConverterrK r% r& r8 Ú get_colorÚto_rgbaÚtupleÚ get_edgecolorÚ get_facecolorÚnpÚndarray)r: Ú
|
linecolorsÚ
|
facecolorsÚmappingrH r6 rM rN rO Úconvr; ÚcolorÚresultÚexpecteds r Ú _check_colorsra s± õ "÷ñ õ
|
(à × Ñ DØÐ Ø Ð Ü+¨G°ZÓ@JØ#Ð$6¤c¨+Ó&6Ð7Jä;Ó ¤3 z£?Ò2Ð2Ð2Ü ¨ZÓ 8ò &LE5Ü% Ô(Ø Ó*à fÓ-ÜE N°NÐ#CÔDÜ u× 2Ñ 2Ó 4°QÑ 7Ó8à ×,Ñ,Ó.à|| EÓ*HØXÓ%Ð %Ð%ð &ð Ð Ø Ð Ü+¨G°ZÓ@JØ#Ð$6¤c¨+Ó&6Ð7Jä;Ó ¤3 z£?Ò2Ð2Ð2Ü ¨ZÓ 8ò &LE5Ü% Ô,à ×,Ñ,Ó.¨qÑ1à ×,Ñ,Ó.ä&¤"§*¡*Ô-Ü v à|| EÓ*HØXÓ%Ð %Ð%ñ &ð r c óò t | « s| j « |k( sJ y| D cg c] }|j « }}t |« t |« k( sJ t ||« D ] \ }}||k( r J yc c}w )a
|
Check each text has expected labels
|
|
Parameters
|
----------
|
texts : matplotlib Text object, or its list-like
|
target text, or its list
|
expected : str or list-like which has the same length as texts
|
expected text label, or its list
|
N)r Úget_textr% r& )Útextsr` Útr ÚlabelÚes r r r Ê s{ ô Ô Ø~~Ó 8Ò+Ð+Ñ+à(-Ö. 1!**,Ð.Ð.Ü6{c (mÒ+Ð+Ð+ÜF HÓ-ò HE1ØA:Ð :ñ ùò /s ¦A4c óú ddl m} t | « } | D ]d }||¬t |j j « |« r|j « }n#|j « |j d¬« z }|D ]O }|$t j |j « |« |,t j |j « |« Q ||¹t |j j « |« r|j « }n#|j « |j d¬« z }|D ]O }|$t j |j « |« |,t j |j « |« Q g y)ac
|
Check each axes has expected tick properties
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
xlabelsize : number
|
expected xticks font size
|
xrot : number
|
expected xticks rotation
|
ylabelsize : number
|
expected yticks font size
|
yrot : number
|
expected yticks rotation
|
r )Ú NullFormatterNT)Úminor) Úmatplotlib.tickerri r r8 ÚxaxisÚget_minor_formatterÚget_xticklabelsr( r) Ú get_fontsizeÚ get_rotationÚyaxisÚget_yticklabels) r Ú
|
xlabelsizeÚxrotÚ
|
ylabelsizeÚyrotri r r rf s r Ú_check_ticks_propsrw Þ sf õ 0ä DÓ !DØó GØ Ð ! TÐ%5Ü"((×6Ñ6Ó8¸-ÔHð ×+Ñ+Ó-à×+Ñ+Ó-°×0BÑ0BÈÐ0BÓ0NÑNà ò GØ Ð)Ü×*Ñ*¨5×+=Ñ+=Ó+?ÀÔLØÑ#Ü×*Ñ*¨5×+=Ñ+=Ó+?ÀÕFð Gð Ð ! TÑ%5Ü"((×6Ñ6Ó8¸-ÔHØ×+Ñ+Ó-à×+Ñ+Ó-°×0BÑ0BÈÐ0BÓ0NÑNà ò GØ Ð)Ü×*Ñ*¨5×+=Ñ+=Ó+?ÀÔLØÑ#Ü×*Ñ*¨5×+=Ñ+=Ó+?ÀÕFò Gñ+Gr c ó¤ t | « } | D ]@ }|j j « |k( sJ |j j « |k( r@J y)zö
|
Check each axes has expected scales
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
xaxis : {'linear', 'log'}
|
expected xaxis scale
|
yaxis : {'linear', 'log'}
|
expected yaxis scale
|
N)r rl Ú get_scalerq )r rl rq r s r Ú_check_ax_scalesrz sS ô DÓ !DØò -Øxx×!Ñ!Ó# uÒ,Ð,Ð,Øxx×!Ñ!Ó# uÓ,Ð,Ð,ñ-r c ó\ ddl m} |d}t | « }|5t |« |k( sJ |D ] }t |j « « dkD r J |t « }t « } || « D ]N }|j « j « } |j | d d « |j | d d « P t |« t |« f}
|
|
|
|k( sJ t j |d j j « t j |t j ¬« « y)a
|
Check expected number of axes is drawn in expected layout
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
axes_num : number
|
expected number of axes. Unnecessary axes should be set to
|
invisible.
|
layout : tuple
|
expected layout, (expected number of rows , columns)
|
figsize : tuple
|
expected figsize. default is matplotlib default
|
r ©Ú flatten_axesN)g@g333333@é )Údtype)Ú!pandas.plotting._matplotlib.toolsr} r r% Ú get_childrenÚsetÚ get_positionÚ
|
get_pointsÚaddr( Úassert_numpy_array_equalÚfigureÚget_size_inchesrX ÚarrayÚfloat64) r Úaxes_numÚlayoutÚfigsizer} Ú visible_axesr Úx_setÚy_setÚpointsr_ s r Ú_check_axes_shaper s õ ?àØ Ü# DÓ)LàÐÜ<Ó HÒ,Ð,Ð,Ø ò .BärÓ(Ó)¨AÓ-Ð -Ð-ð .ð ÐÜÜÙ tÓ$ò $Bà__Ó&×1Ñ1Ó3FØ IIfQi lÔ #Ø IIfQi lÕ #ð $ô
|
e*c %jÐ)ØÒ Ð Ð ä× Ñ ØQ× Ñ ×.Ñ.Ó0Ü
|
¤§
|
¡
|
Ô+õr c ól ddl m} || « }|D cg c] }|j « s| } }| S c c}w )z
|
Flatten axes, and filter only visible
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
|
r r| )r r} r9 )r r} Ú axes_ndarrayr s r r r K s7 õ ?á Ó%LØ%Ö :2¨¯ © Õ)9BÐ :DÐ :Ø Kùò ;s 1©1c óÄ t | « } | D ]P }|j }d}d}|D ]+ }t |dd« }t |dd« } |r|dz }| s'|dz }- ||k( sJ ||k( rPJ y)zø
|
Check axes has expected number of errorbars
|
|
Parameters
|
----------
|
axes : matplotlib Axes object, or its list-like
|
xerr : number
|
expected number of x errorbar
|
yerr : number
|
expected number of y errorbar
|
r Úhas_xerrFÚhas_yerrr~ N)r Ú
|
containersÚgetattr)
|
r ÚxerrÚyerrr r Ú
|
xerr_countÚ
|
yerr_countÚcr r s
|
r Ú_check_has_errorbarsr [ s ô DÓ !DØò "Ø]]
|
Ø
|
Ø
|
Øò AÜ q *¨eÓ4HÜ q *¨eÓ4HÙØa
|
ÚØa
|
ð ð zÒ!Ð!Ð!ØzÓ!Ð!Ð!ñ "r c ó6 ddl m} t |t d}|P|d}t | || « sJ |dk( r5t | j
|
|« sJ t | j t « sJ yy| t | « D ] }t ||« rJ yt | t « sJ t | j « « t |« k( sJ | j « D ]Æ \ }}t ||| « sJ |dk( r|s |j « |k( r3J |dk( rV|r |j
|
j « |k( sJ t |j
|
|« sJ t |j t « rJ |dk( r-|d d } | j }
|
|s¬|
|
j « |k( rÀJ t y) a<
|
Check box returned type is correct
|
|
Parameters
|
----------
|
returned : object to be tested, returned from boxplot
|
return_type : str
|
return_type passed to boxplot
|
expected_keys : list-like, optional
|
group labels in subplot case. If not passed,
|
the function checks assuming boxplot uses single ax
|
check_ax_title : bool
|
Whether to check the ax.title is the same as expected_key
|
Intended to be checked by calling from ``boxplot``.
|
Normal ``plot`` doesn't attach ``ax.title``, it must be disabled.
|
r r )rE r ÚbothNrE r¡ r Úmedians)Úmatplotlib.axesr rE rU r8 r Úlinesr r ÚsortedÚkeysÚitemsÚ get_titler ÚAssertionError) ÚreturnedÚ return_typeÚ expected_keysÚcheck_ax_titler ÚtypesÚrÚkeyÚvalueÚliner s r Ú_check_box_return_typer³ w s¦ õ& %ä 4´Ñ 7EØÐ à Ð Ø Kä( E¨+Ñ$6Ô7Ð7Ð7Ø &Ò Ü hkk¨4Ô0Ð 0Ð0Ü hnn¬dÔ3Ð 3Ñ3ð !ð
|
Ð Ü% hÓ/ò +Ü! ! TÕ*Ð*Ð*ð +à ä(¤FÔ+Ð+Ð+ähmmoÓ&¬&° Ó*?Ò?Ð?Ð?Ø"..Ó*ò %JCÜ e U¨;Ñ%7Ô8Ð 8Ð8àfÒ$Ú!Ø ??Ó,°Ó3Ð3Ð3Ø Ò&Ù!Ø 88×-Ñ-Ó/°3Ò6Ð6Ð6Ü! %§(¡(¨DÔ1Ð1Ð1Ü! %§+¡+¬tÕ4Ð4Ð4Ø Ò&Ø YÑ'¨Ñ*ØyyÚ!Ø >>Ó+¨sÓ2Ð2Ð2ä$Ð$ñ# %r c óü dd l fd}d}|D ]ê }j j ddt |« z |« |dz }j dd¬« | j
|
d d|i|¤ |« rJ j j « j j ddt |« z |« |dz }j dd ¬« | j
|
d |dd
|
|¤ |« rJ j j « |d vsúj j ddt |« z |« |dz }j dd ¬« | j
|
d d|i|¤ |« sJ j j « j j ddt |« z |« |dz }j dd¬« | j
|
d |d d
|
|¤ |« sJ j j « í y ) Nr c ó" j j « j j « } j j « j j « }t d | D « « }t d |D « « }|xr | S )Nc 3 óR K | ] }|j j « ! y wrC ©Úgridliner9 ©Ú.0Úgs r ú <genexpr>z;_check_grid_settings.<locals>.is_grid_on.<locals>.<genexpr>¼ ó è ø Ò@°Aqzz×-Ñ-Ó/Ô/Ñ@ùó %'c 3 óR K | ] }|j j « ! y wrC r· r¹ s r r¼ z;_check_grid_settings.<locals>.is_grid_on.<locals>.<genexpr>½ r½ r¾ )r# Úgcarl Úget_major_ticksrq r! )ÚxticksÚyticksÚxoffÚyoffÚmpls r Ú
|
is_grid_onz(_check_grid_settings.<locals>.is_grid_on¹ sn ø Ø Ó!×'Ñ'×7Ñ7Ó9Ø Ó!×'Ñ'×7Ñ7Ó9ÜÑ@¸Ô@Ó@ÜÑ@¸Ô@Ó@àMTÐ"Ð"r r~ é r F)ÚgridÚkindT)rÊ rÉ )ÚpieÚhexbinÚscatter© )rP r# Úsubplotr% ÚrcÚplotÚclf)ÚobjÚkindsÚkwsrÇ ÚspndxrÊ rÆ s @r Ú_check_grid_settingsr× ´ sÆ ø ó ô#ð EØó Ø
|
|
×Ñ1a¤# e£*n¨eÔ4Ø
|
Ø vEÔ"ØÑ"dÐ"cÒ"Ù <Ð Ð Ø
|
|
Ôà
|
|
×Ñ1a¤# e£*n¨eÔ4Ø
|
Ø vDÔ!ØÐ.d Ñ.¨#Ò.Ù <Ð Ð Ø
|
|
Ôà Ð3Ò 3Ø JJ× Ñ q !¤c¨%£j¡.°%Ô 8Ø QJEØ FF6 FÔ %Ø CHHÑ &$Ð & #Ò &Ù <Ð <Ø JJNNÔ à JJ× Ñ q !¤c¨%£j¡.°%Ô 8Ø QJEØ FF6 FÔ &Ø CHHÐ 1$ TÑ 1¨SÒ 1Ù <Ð <Ø JJNNÖ ñ9 r c ó8 | d D cg c] }|| c}S c c}w )zL
|
Auxiliary function for correctly unpacking cycler after MPL >= 1.5
|
zaxes.prop_cyclerÎ )ÚrcParamsÚfieldrJ s r Ú_unpack_cyclerrÛ á s" ð 'Ð'8Ñ 9Ö :AeHÒ :Ð:ùÒ :s c ó | j d S )NÚx©Ú _shared_axes©r s r Ú
|
get_x_axisrá è ó Ø ??3Ñ Ð r c ó | j d S )NÚyrÞ rà s r Ú
|
get_y_axisrå ì râ r c ó. ddl m} |rt }nt }d} |j d|j « « }|j « || |fi |¤D ] }t j |« |j |« |S # |j « w xY w)a×
|
Create plot and ensure that plot return object is valid.
|
|
Parameters
|
----------
|
f : func
|
Plotting function.
|
default_axes : bool, optional
|
If False (default):
|
- If `ax` not in `kwargs`, then create subplot(211) and plot there
|
- Create new subplot(212) and plot there as well
|
- Mind special corner case for bootstrap_plot (see `_gen_two_subplots`)
|
If True:
|
- Simply run plotting function with kwargs provided
|
- All required axes instances will be created automatically
|
- It is recommended to use it when the plotting function
|
creates multiple axes itself. It helps avoid warnings like
|
'UserWarning: To output multiple subplots,
|
the figure containing the passed axes is being cleared'
|
**kwargs
|
Keyword arguments passed to the plotting function.
|
|
Returns
|
-------
|
Plot object returned by the last plotting.
|
r Nr )
|
r" r# Ú_gen_default_plotÚ_gen_two_subplotsÚgetÚgcfrÒ r( Ú"assert_is_valid_plot_return_objectr* )ÚfÚ default_axesÚkwargsr- Ú gen_plotsÚretÚfigs r Ú_check_plot_worksrò ð s õ6 $áÜ% ä% à
|
CðØjj 3§7¡7£9Ó-Ø á Q Ñ. vÑ.ò 7CÜ × 1Ñ 1°#Õ 6ñ 7ð #à Jøð #ús AB ÂBc + ó K | di |¤ yw)z'
|
Create plot in a default way.
|
NrÎ rÎ ©rì rñ rî s r rç rç s è ø ñ +f+Óùs c + óÊ K d|vr|j d« | di |¤ | t j j u rd|vsJ |j d« |d< | di |¤ yw)z9
|
Create plot on two subplots forcefully created.
|
r éÓ éÔ NrÎ )Ú add_subplotÚpdÚplottingÚbootstrap_plotrô s r rè rè ' sf è ø ð 6ÑØ Ô Ù
|
+f+ÒàBKK× &Ñ &Ñ&Ø6Ñ!Ð!Ð!à sÓ+t Ù
|
+f+Óùs A!A#)NT)T)r úAxes | Sequence[Axes]r@ ÚboolÚreturnÚNone)NNN)NNNN)Úlinearr )r rü rþ zSequence[Axes])r r )r^ )F)(Ú__doc__Ú
|
__future__r Útypingr ÚnumpyrX Úpandas.core.dtypes.apir Úpandasrù r Úpandas._testingÚ_testingr( Úcollections.abcr r£ r r r r4 r< rA rK ra r rw rz r r r r³ r× rÛ rá rå rò rç rè rÎ r r ú<module>r
|
s¥ ðñõ #å ã å /ã Ý Ý áÝ(å$ó+ó0'ò.ó..ô((ò".ó=&ò@ ó(,Gó^-ó$)óX ó "ð: ?Có:%ðz *,ó * óZ;ò ò ó-ò`ó r
|
