U
|
ý°dä1 ã @ s d dl Z d dlZd dlZd dlm mZ d dlm Z d dl
|
m Z d dl mZmZ ejdd ZejG dd deZejG dd d eZdS )
|
é N)Ú DataFrame)Ú TestPlotBaseÚ_check_plot_worksc C sZ t j d¡ tt j dd¡ddgd} t j ddd gd¡| d
|
< t j ddd gd¡| d < | S ) Nr é é ÚAÚB©ÚcolumnsÚaÚbÚcÚCÚD)ÚnpÚrandomÚseedr ZrandnÚchoice)Údf© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/plotting/frame/test_hist_box_by.pyÚhist_df s
|
r c @ s² e Zd Zejjej ddddddgdggd fdddgdddgddggd fdd dddgddggd fdd
|
gdd d d ddddddg dggd fdd
|
gddgd d d ddddddg ddggd fdd
|
gd d d d ddddddg ddggd fg¡dd Zej ddddddgdggd fdd dddgddggd fdd
|
gdd d d ddddddg dggd fg¡dd Zej dg dgfg ddgfdd ddgfg¡d d Z ejjej d dgdd dfd!dgdgd"dfd#dddgd$dfdd
|
gdd%dfdd
|
gdd&dfdd
|
gdgd'dfdd
|
gddgd%dfdd
|
gd d%dfdd
|
gddgd'dfg ¡d(d) Z
|
ej d*d+dd
|
gd,fe d-¡dd.fd/g¡d0d1 Z ejjd2d3 Zejjd4d5 Zej d6d7d8g¡d9d: Zd S );ÚTestHistWithByzby, column, titles, legendsr r r r r é r Nr ú(a, a)ú(a, b)ú(a, c)ú(b, a)ú(b, b)ú(b, c)ú(c, a)ú(c, b)ú(c, c)é c C sL t |jj||dd}dd |D }dd |D }||ks<t||ksHtd S )NT)ÚcolumnÚbyÚ default_axesc S s g | ] }| ¡ qS r ©Z get_title©Ú.0Zaxr r r Ú
|
<listcomp>Y s z=TestHistWithBy.test_hist_plot_by_argument.<locals>.<listcomp>c S s g | ]}d d | ¡ jD qS )c S s g | ] }| ¡ qS r ©Zget_text©r) Zlegendr r r r* [ s zHTestHistWithBy.test_hist_plot_by_argument.<locals>.<listcomp>.<listcomp>©Z
|
get_legendZtextsr( r r r r* Z s )r ÚplotÚhistÚAssertionError) Úselfr% r$ ÚtitlesÚlegendsr ÚaxesÚ result_titlesÚresult_legendsr r r Útest_hist_plot_by_argument s ; ÿÿ z)TestHistWithBy.test_hist_plot_by_argumentr c
|
C sd | ¡ }|jddid}t|jjd||d}dd |D }dd |D } | |ksTt||ks`td S ) Nr r r T©r& r$ r% c S s g | ] }| ¡ qS r r' r( r r r r* ~ s z6TestHistWithBy.test_hist_plot_by_0.<locals>.<listcomp>c S s g | ]}d d | ¡ jD qS )c S s g | ] }| ¡ qS r r+ r, r r r r* s zATestHistWithBy.test_hist_plot_by_0.<locals>.<listcomp>.<listcomp>r- r( r r r r* s )ÚcopyÚrenamer r. r/ r0 )
|
r1 r% r$ r2 r3 r r r4 r5 r6 r r r Útest_hist_plot_by_0a s ÿ z"TestHistWithBy.test_hist_plot_by_0ú
|
by, column)r Nr c C s6 d}t jt|d t|jjd||d W 5 Q R X d S ©NzNo group keys passed©ÚmatchTr8 )ÚpytestÚraisesÚ
|
ValueErrorr r. r/ ©r1 r% r$ r Úmsgr r r Ú)test_hist_plot_empty_list_string_tuple_by s ÿz8TestHistWithBy.test_hist_plot_empty_list_string_tuple_byú by, column, layout, axes_num©r r )r r rG r )é r )r N©r rH r rI )r# rH )r r )é r c C sB t jtdd t|jj|||d}W 5 Q R X | j|||d d S )NF)Zcheck_stacklevel©r$ r% Úlayout©Úaxes_numrL )ÚtmZassert_produces_warningÚ UserWarningr r. r/ Ú_check_axes_shape©r1 r% r$ rL rN r r4 r r r Ú test_hist_plot_layout_with_by s ÿz,TestHistWithBy.test_hist_plot_layout_with_byúmsg, by, layoutúlarger than required size©rH rH ú)Layout must be a tuple of (rows, columns)©rH ©z1At least one dimension of layout must be positiver )éÿÿÿÿrZ c C s4 t jt|d |jjddg||d W 5 Q R X d S ©Nr> r r rK )r@ rA rB r. r/ ©r1 rD r% rL r r r r Ú,test_hist_plot_invalid_layout_with_by_raises± s z;TestHistWithBy.test_hist_plot_invalid_layout_with_by_raisesc C sÌ |j jdddd\}}}| |¡ ||¡s.t| |¡ ||¡sDt| |¡ ||¡sZt| |¡ ||¡spt| |¡ ||¡rt| |¡ ||¡rt| |¡ ||¡r²t| |¡ ||¡rÈtd S )Nr r T)r$ r% Zsharex)r. r/ Ú
|
get_x_axisÚjoinedr0 Ú
|
get_y_axis©r1 r Zax1Zax2Zax3r r r Útest_axis_share_x_with_by¿ s z(TestHistWithBy.test_axis_share_x_with_byc C sÌ |j jdddd\}}}| |¡ ||¡s.t| |¡ ||¡sDt| |¡ ||¡sZt| |¡ ||¡spt| |¡ ||¡rt| |¡ ||¡rt| |¡ ||¡r²t| |¡ ||¡rÈtd S )Nr r T)r$ r% Zsharey)r. r/ r` r_ r0 r^ ra r r r Útest_axis_share_y_with_byÐ s z(TestHistWithBy.test_axis_share_y_with_byÚfigsize©é é ©é é
|
c C s& |j jdd|d}| j|d|d d S )Nr r ©r$ r% rd r ©rN rd )r. r/ rQ ©r1 rd r r4 r r r Ú test_figure_shape_hist_with_byá s z-TestHistWithBy.test_figure_shape_hist_with_by)Ú__name__Ú
|
__module__Ú __qualname__r@ ÚmarkÚslowÚ parametrizer7 r; rE rS ÚreÚescaper] rb rc rn r r r r r sà ÷
|
ò÷ ò÷ òÜþ8 ÷
|
òýþ
|
|
|
üþ
|
õþ ýþ
|
|
|
r c @ sþ e Zd Zej ddddgdddggfddgddgdd d
|
d d d dddg gfdddgddgdddggd fddgddgddgdd d
|
d d d dddg gd fdgdddgdddggd fg¡dd Zej ddddgdddggfddgddgdd d
|
d d d dddg gfddddgdddggd fg¡dd Zej dg dgfdg dfdddgfg¡d d Zejj ej d dgdd d fd!d"dddgd#dfddgdd d fddgdd#dfg¡d$d% Z
|
ej d&d'ddgd fe d(¡dd)fd*g¡d+d, Z ej d-d.d/g¡d0d1 ZdS )2Ú TestBoxWithByz by, column, titles, xticklabelsr r r r r r r r r r r r r r! r" r r Nc C sL t |jjd||d}dd |D }dd |D }||ks<t||ksHtd S )NTr8 c S s g | ] }| ¡ qS r r' r( r r r r* s z;TestBoxWithBy.test_box_plot_by_argument.<locals>.<listcomp>c S s g | ]}d d | ¡ D qS )c S s g | ] }| ¡ qS r r+ ©r) Úlabelr r r r* s zFTestBoxWithBy.test_box_plot_by_argument.<locals>.<listcomp>.<listcomp>©Zget_xticklabelsr( r r r r* s )r r. Úboxr0 ) r1 r% r$ r2 Ú xticklabelsr r4 r5 Úresult_xticklabelsr r r Útest_box_plot_by_argumentê s / ÿÿ z'TestBoxWithBy.test_box_plot_by_argumentr c
|
C sd | ¡ }|jddid}t|jjd||d}dd |D }dd |D } | |ksTt||ks`td S ) Nr r r Tr8 c S s g | ] }| ¡ qS r r' r( r r r r* C s z4TestBoxWithBy.test_box_plot_by_0.<locals>.<listcomp>c S s g | ]}d d | ¡ D qS )c S s g | ] }| ¡ qS r r+ rx r r r r* E s z?TestBoxWithBy.test_box_plot_by_0.<locals>.<listcomp>.<listcomp>rz r( r r r r* D s )r9 r: r r. r{ r0 )
|
r1 r% r$ r2 r| r r r4 r5 r} r r r Útest_box_plot_by_0$ s ÿ z TestBoxWithBy.test_box_plot_by_0r< )r r r c C s6 d}t jt|d t|jjd||d W 5 Q R X d S r= )r@ rA rB r r. r{ rC r r r Ú%test_box_plot_with_none_empty_list_byK s z3TestBoxWithBy.test_box_plot_with_none_empty_list_byrF rV rH )r r rV rH )r N)r rH r )rH r c C s* t |jjd|||d}| j|||d d S )NT)r& r$ r% rL rM )r r. r{ rQ rR r r r Ú test_box_plot_layout_with_byZ s ÿz*TestBoxWithBy.test_box_plot_layout_with_byrT rU rW rX rY c C s4 t jt|d |jjddg||d W 5 Q R X d S r[ )r@ rA rB r. r{ r\ r r r Ú+test_box_plot_invalid_layout_with_by_raisesm s z9TestBoxWithBy.test_box_plot_invalid_layout_with_by_raisesrd re rh c C s& |j jdd|d}| j|d|d d S )Nr r rk rH rl )r. r{ rQ rm r r r rn { s z,TestBoxWithBy.test_figure_shape_hist_with_by)ro rp rq r@ rr rt r~ r r rs r ru rv r rn r r r r rw è sº ÷ÿü ÷ÿ óü Øþ-
|
÷ÿüìþ
|
|
üþ
|
úþ ýþ
|
rw )ru Únumpyr r@ Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr Zpandas._testingZ_testingrO Z pandas.tests.plotting.commonr r Zfixturer Zskip_if_no_mplr rw r r r r Ú<module> s
|
O
|
