U md;@sjddlZddlZddlZddlmZdZzddlm Z Wne k rLYnXddZ ej jddZdS)N)dot_plotFcCstr||t|dS)N) pdf_outputZsavefigpltclose)pdffigr`/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/graphics/tests/test_dotplot.py close_or_saves r c strddlm}|d}nd}ttd}t}t||d}|dt ||ttd}t}t||dd}|d t ||tj d d t}t d }t||d}|d | dt ||tj dd t}t d }t||dd}|d|dt ||tj d d t}t d }t||dd}|d|ddt ||tj dd t}t d }t||ddd}|d|ddt ||t t}t d}t||d}|dt ||t t}t d}t||dd}|dt ||t t}t d}t||d}|dd|dt ||tt}ddtdD}t jjdd }t|||d!}|d"t ||tt}t jjdd }d#dtdD}t|||d$d%}|d&t ||ttd'd'd(d)g}t jjdd }d*dtdD}t|||d$dd+}|d,} | d-t ||ttd'd.d/d0g}t jjdd }t td1t dt j}t t d1tdt j} d2d3tdD} t||| || d4}|d5t ||tt}td}t|t d|d6}|d7t ||tt}ttd}tt d} t|| |d6}|d8t ||tt}t d}d9dtdD} t|| |d6}|d:t ||tt}t d}d;dtdD} t|| |dd<}|d=t ||tt}t d}d>dtdD} dd?d@dAi} t|| | |dB}|dCt ||ttd'd'dDd)g}d1t jjd d }t tddE}dFdtd D} t t ddGt j} dHd| D} t|| || |ddI}|\}}tj||dJdKdLdM}|d|dNt ||ttd'd'dDd)g}t|| || |ddOdPgdQ}|\}}tj||dJdKdLdM}|d|dRt ||ttd'd'dDd)g}d1t jjd d }t tddE}dSdtd D} t t ddGt j} dTd| D} t|| || |dddU}|\}}tj||dJdKdLdM}|d|dVt ||ttd'd'dDd)g}dOdPg}t|| || |dd|dW}|\}}tt||fdXd|D}tj||dJdKdLdM}|d|dYt ||ttd'd'dDd)g}d1t jjd d }t tddE}dZdtd D} t t ddGt j} d[d| D} t|| || |dddd\}|\}}tj||dJdKdLdM}|dtd]d|dVt ||ttd'd'dDd)g}d1t jjd d }t tddE}d^dtd D} t t ddGt j} d_d| D} d`daid`dbidc} d`daid`dbidc} t|| || |d| | dd}|\}}tj||dJdKdLdM}|d|det ||ttd'd'dDd)g}d1t jjd d }t tddE}dfdtd D} t t ddGt j} dgd| D} d`daid`dbidc} d`daid`dbidc} t|| || |d| | ddh }|\}}tj||dJdKdLdM}|d|det ||tt}td}t tdidEt j}t t didGt j} t djt dkt j}dld|D}t||| ||dm}|dnt ||ttd'd'dodDg}td}t tdidEt j}t t didGt j} t djt dkt j}dpd|D}t||| ||ddq}|dr} | dst ||tt}td}t tdidEt j}t t didGt j} t djt dkt j}dtd|D}t||| ||dudvdwgdx}|dyt ||t t}t d}dzd{d|d}g}d{d|dzd}g}t||||d~}|dt ||ttd}ddtdD}dd}dd}t}t|||d$||d}|dt ||ttd}ddtdD}t}t|||d$dd}|dt ||ttd'd'dDd)g}d1t jjd d }g}d}t |d krftt j!dKdD]}|"|qH|dK7}q(t t ddGt j} dd| D} t||| |dd}|\}}tj||dJdKdLdM}|d|dt ||tr|#dS)Nr)PdfPagesztest_dotplot.pdf)axzBasic horizontal dotplotF)r horizontalzBasic vertical dotplot) )Zfigsize(zTall and skinny dotplotz x axis label)rrzShort and wide dotplotz y axis labelT)r stripedzTall and skinny striped dotploti2)r rrzShort and wide striped dotplotrz'Basic horizontal dotplot with few linesz%Basic vertical dotplot with few linesz&Dotplot with adjusted horizontal rangecSsg|]}dtjddqS)ABCDEFGHr)nprandomrandint.0krrr osztest_all..)size)linesr z4Dotplot with user-supplied labels in the left margincSs.g|]&}dtjdddt|dqSrrr::rrrstrrrrr rysr!)rr split_namesz1Dotplot with user-supplied labels in both marginsg?g)\(?g?cSs.g|]&}dtjdddt|dqSr r#rrrr rs)rr r%rz:Vertical dotplot with user-supplied labels in both margins)?g(\?gQ?g(\?g333333?cSs$i|]}|d|d|dddqS)ZrgbcZosvpg333333?)colormarkermsalpharrrrr s  ztest_all..)rstylesr marker_propsz&Dotplot with custom colors and symbols) intervalsr z Dotplot with symmetric intervalsz0Dotplot with symmetric intervals (Pandas inputs)cSsg|]}dqS)r"rrirrr rsz#Dotplot with nonsymmetric intervalscSsg|]}dqSr1rr3rrr rs)r0r rz,Vertical dotplot with nonsymmetric intervalscSsg|]}dqSr1r)rxrrr rsZ lightgreyround)r)Zsolid_capstyle)r0 line_propsr z#Dotplot with custom line propertiesg?)r"r"cSsg|]}dqSr1rrrrr rs)rr"cSsg|]}ddg|qSCatDogrr3rrr rs)r0rr.r stackedz center rightr"g-C6?)locZ numpointsZ handletextpadz Dotplot with two points per liner:r9)r0rr.r r; styles_orderz0Dotplot with two points per line (reverse order)cSsg|]}dqSr1rrrrr rscSsg|]}ddg|qSr8rr3rrr rs)r0rr.r r;rz)Vertical dotplot with two points per line)r0rr.r r;rr=csg|] }|qSrr)rlZlhrr rsz9Vertical dotplot with two points per line (reverse order)cSsg|]}dqSr1rrrrr r scSsg|]}ddg|qSr8rr3rrr r s)r0rr.r r;rricSsg|]}dqSr1rrrrr rscSsg|]}ddg|qSr8rr3rrr rsr)Zorangepurpler8)r0rr.r r;r/r7z/Dotplot with color-matched points and intervalscSsg|]}dqSr1rrrrr r2scSsg|]}ddg|qSr8rr3rrr r4s)r0rr.r r;r/r7r)rr" cSsg|]}dddg|qSAxxByyCzzrrrrr rJs)rr.sectionsr zDotplot with sectionsg?cSsg|]}dddg|qSrDrrrrr rVs)rr.rHr rzVertical dotplot with sections)r&gHzG?cSsg|]}dddg|qSrDrrrrr rdsrFrErG)rr.rHr Z section_orderz(Dotplot with sections in specified orderABCD)r line_orderr zDotplot with reordered linescSsg|]}d|d|fqSz%d::%ddrr3rrr rwscSsd|S)NZlft_rr5rrr xztest_all..cSsd|S)NZrgt_rrPrrr rQyrR)rr r%Z fmt_left_nameZfmt_right_namez'Horizontal dotplot with name formattingcSsg|]}d|d|fqSrNrr3rrr rsright)rr r%Z show_nameszShow right names onlycSsg|]}ddg|qSr8rr3rrr rs)rr.r r;z1Dotplot with different numbers of points per line)$rZmatplotlib.backends.backend_pdfr rZclfrangeZaxesr set_titler ZfigurerZarangeZ set_xlabelZ set_ylabelZset_xlimZset_ylimrnormalZ set_positionZkronZonesZastypeZint32pdZSeriesZget_legend_handles_labelsZ figlegendZ draw_framedictzipZylimlenrappendr)Z close_figuresZreset_randomstater rZpointsr rvalsrtxtr.r/r0r7Zhandleslabelslegr=rHrMZfmt_leftZ fmt_rightiirrr?r test_alls                                                                                                                                  ra)numpyrZpandasrWZpytestZstatsmodels.graphics.dotplotsrrZmatplotlib.pyplotZpyplotr ImportErrorr markZ matplotlibrarrrr s