a ▓<bву@stdZddlZddlZddlmmZddlm Z m Z mZddlm ZddlmZejjZejGddДdeГГZdS)z Test cases for GroupBy.plot щN)┌ DataFrame┌Index┌Series)┌TestPlotBasec@sxeZdZddДZddДZddДZddДZd d ДZej а ddd gбddДГZej а dddgбddДГZddДZ ddДZdS)┌TestDataFrameGroupByPlotscCs╕d}ttjjdd|dНГ}ttjjdd|dНГ}tаdбП$tjjddg|dН}WdГn1sd0Y|а|бабtа б|а|ба бtа б|а|бjd d Нtа бdS)Nщ щжщ)┌sizeщ<щ*ZmaleZfemalegр?)┌alpha)r┌np┌random┌normal┌tmZ RNGContext┌choice┌groupby┌plot┌close┌hist)┌self┌nZweight┌heightZgenderйr·r/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/tests/plotting/test_groupby.py┌,test_series_groupby_plotting_nominally_workss2zFTestDataFrameGroupByPlots.test_series_groupby_plotting_nominally_workscCs`tgdвtjаdбdЬgdвdН}|аdбdабtаб|аdбdаdd ДбtабdS) N) щrrщrrщrrщ )┌def┌val) чЁ?ч@ч@r#r$r%r#r$r%)┌indexr!r"cSs|абS)N)r)┌xrrr┌.єzPTestDataFrameGroupByPlots.test_plotting_with_float_index_works..) rrr┌randnrrrr┌applyйr┌dfrrr┌$test_plotting_with_float_index_works%s■z>TestDataFrameGroupByPlots.test_plotting_with_float_index_workscCsptаdddб}tddgddgddgd ЬГ}|d j|d|dНtdgdgdgd ЬГ}|d j|d|dНdS) NщPщfrZAAAZBBBrщUщY)┌Name┌ByCol┌Markr5r4)Zby┌bins)rZarangerr)rr6r-rrr┌test_hist_single_row1s z.TestDataFrameGroupByPlots.test_hist_single_rowcCsXtgdвgdвtdГdЬГ}|аdбjаddбtаб|аdбdjабtабdS)Nйrrrщщйrrrrr┌ababaйr'┌y┌zr?r'r>)r┌listrr┌scatterrr┌liner,rrr┌test_plot_submethod_works9s z3TestDataFrameGroupByPlots.test_plot_submethod_workscCsxtgdвgdвtdГdЬГ}|аdбjdddd Н}t|d jГdksHJВ|аdбjjdddН}t|d jГdkstJВdS) Nr8r;r<r=r?rAr'r>)┌kindr'r>┌ar)r'r>)rr@rr┌len┌collectionsrA)rr-┌resrrr┌test_plot_kwargs@s z*TestDataFrameGroupByPlots.test_plot_kwargszcolumn, expected_axes_num)Nr)┌brcCsжd|f}|pdgdgg}tddgddgddН}ttjаd d б|ddgdН}|аdб}|jd|d НD]8}|j|||dНt|d|ГD]\} } |а | | бqКqhdS)NrrErJщ┌1┌2┌cй┌nameщrйr&┌columnsT)┌legend┌columnйZaxes_numZlayoutr) rrrrr*rr┌_check_axes_shape┌zip┌_check_legend_labels)rrUZexpected_axes_numZexpected_layoutZexpected_labelsr&r-┌gZaxes┌axZexpected_labelrrr┌#test_groupby_hist_frame_with_legendLs z=TestDataFrameGroupByPlots.test_groupby_hist_frame_with_legendrUNrJcCsДtddgddgddН}ttjаddб|dd gd Н}|аdб}tjtddНП |j d |ddНWdГn1sv0YdS)NrKrLrMrNrOrQrrErJrR· Cannot use both legend and labelй┌matchT┌d)rTrU┌labelй rrrrr*r┌pytestZraises┌ ValueErrorr)rrUr&r-rZrrr┌*test_groupby_hist_frame_with_legend_raises]s zDTestDataFrameGroupByPlots.test_groupby_hist_frame_with_legend_raisescCs|tddgddgddН}ttjаddб|dd gd Н}|аdб}|djddНD]$}|j|d ddН|а|ddgбqRdS)NrKrLrMrNrOrQrrErJrRT)rTr)rrrV) rrrrr*rrrWrY)rr&r-rZr[rrr┌$test_groupby_hist_series_with_legendgs z>TestDataFrameGroupByPlots.test_groupby_hist_series_with_legendcCsВtddgddgddН}ttjаddб|dd gd Н}|аdб}tjtddНП|j d ddНWdГn1st0YdS)NrKrLrMrNrOrQrrErJrRr]r^Tr`)rTrarb)rr&r-rZrrr┌+test_groupby_hist_series_with_legend_raisesqs zETestDataFrameGroupByPlots.test_groupby_hist_series_with_legend_raises)┌__name__┌ __module__┌__qualname__rr.r7rCrIrc┌markZparametrizer\rerfrgrrrrrs r)┌__doc__ZnumpyrrcZpandas.util._test_decorators┌utilZ_test_decorators┌tdZpandasrrrZpandas._testingZ_testingrZpandas.tests.plotting.commonrrkZslowZ pytestmarkZskip_if_no_mplrrrrr┌s