ó áp7]c@sƒddlZddlZddlmZddlZeZyddl j Z Wne k r`nXd„Z ejjd„ƒZdS(iÿÿÿÿN(tdot_plotcCstr|j|ƒndS(N(t pdf_outputtsavefig(tpdftfig((sFlib/python2.7/site-packages/statsmodels/graphics/tests/test_dotplot.pyt close_or_savescCsÔtr%ddlm}|dƒ}nd}tjƒtdƒ}tjƒ}t|d|ƒ}|j dƒt ||ƒtjƒtdƒ}tjƒ}t|d|dt ƒ}|j dƒt ||ƒtj d dmƒtjƒ}t jd ƒ}t|d|ƒ}|j d ƒ|jdƒt ||ƒtj d dnƒtjƒ}t jd ƒ}t|d|dt ƒ}|j dƒ|jdƒt ||ƒtj d doƒtjƒ}t jd ƒ}t|d|dtƒ}|j dƒ|jddƒt ||ƒtj d dpƒtjƒ}t jd ƒ}t|d|dtdt ƒ}|j dƒ|jddƒt ||ƒtj ƒtjƒ}t jd ƒ}t|d|ƒ}|j dƒt ||ƒtj ƒtjƒ}t jd ƒ}t|d|dt ƒ}|j dƒt ||ƒtj ƒtjƒ}t jdƒ}t|d|ƒ}|jddƒ|j dƒt ||ƒtjƒtjƒ}gtdƒD]}dt jjddƒ^q§} t jjddƒ}t|d| d|ƒ}|j dƒt ||ƒtjƒtjƒ}t jjddƒ}gtdƒD]1}dt jjddƒd t|d!ƒ^qK} t|d| d|d"d ƒ}|j d#ƒt ||ƒtjƒtjd$d$d%d&gƒ}t jjddƒ}gtdƒD]1}dt jjddƒd t|d!ƒ^q} t|d| d|d"d dt ƒ}|j d'ƒ} | jdqƒt ||ƒtjƒtjd$d*d+d,gƒ}t jjddƒ}t jtd-ƒt jd ƒƒjt jƒ} t jt jd-ƒtd ƒƒjt jƒ} d.„td ƒDƒ} t|d| d/| d|d0| ƒ}|j d1ƒt ||ƒtjƒtjƒ}tdƒ}t|d2t jdƒd|ƒ}|j d3ƒt ||ƒtjƒtjƒ}tjtdƒƒ}tjt jdƒƒ} t|d2| d|ƒ}|j d4ƒt ||ƒtjƒtjƒ}t jdƒ}gtdƒD] }dr^qq} t|d2| d|ƒ}|j d6ƒt ||ƒtjƒtjƒ}t jdƒ}gtdƒD] }ds^qç} t|d2| d|dt ƒ}|j d7ƒt ||ƒtjƒtjƒ}t jdƒ}gtdƒD] }dt^qc} iid8d96d:d;6d6}t|d2| d<|d|ƒ}|j d=ƒt ||ƒtjƒtjd$d$d>d&gƒ}d-t jjdd ƒ}t jtdƒduƒ} gtd ƒD] }dv^q+ } t jt jdƒdwƒjt jƒ} g| D]}d?d@g|^qk } t|d2| d| d/| d|dAtƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒ|j dFƒt ||ƒtjƒtjd$d$d>d&gƒ}t|d2| d| d/| d|dAtdGd@d?gƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒ|j dHƒt ||ƒtjƒtjd$d$d>d&gƒ}d-t jjdd ƒ}t jtdƒdxƒ} gtd ƒD] }dy^q# } t jt jdƒdzƒjt jƒ} g| D]}d?d@g|^qc } t|d2| d| d/| d|dAtdt ƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒ|j dIƒt ||ƒtjƒtjd$d$d>d&gƒ}d@d?g}t|d2| d| d/| d|dAtdt dG|ƒ}|jƒ\}}t"t#||ƒƒ}g|D]}||^qž }tj ||dBdCd!dDdEƒ}|j!t ƒ|j dJƒt ||ƒtjƒtjd$d$d>d&gƒ}d-t jjdd ƒ}t jtdƒd{ƒ} gtd ƒD] }d|^q_ } t jt jdƒd}ƒjt jƒ} g| D]}d?d@g|^qŸ } t|d2| d| d/| d|dAtdtdt ƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒtj$dKdƒ|j dIƒt ||ƒtjƒtjd$d$d>d&gƒ}d-t jjdd ƒ}t jtdƒd~ƒ} gtd ƒD] }d^q¾} t jt jdƒd€ƒjt jƒ} g| D]}d?d@g|^qþ} iidLd96d?6idMd96d@6} iidLd96d?6idMd96d@6}t|d2| d| d/| d|dAtd0| d<|ƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒ|j dNƒt ||ƒtjƒtjd$d$d>d&gƒ}d-t jjdd ƒ}t jtdƒdƒ} gtd ƒD] }d‚^qQ} t jt jdƒdƒƒjt jƒ} g| D]}d?d@g|^q‘} iidLd96d?6idMd96d@6} iidLd96d?6idMd96d@6}t|d2| d| d/| d|dAtd0| d<|dt ƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒ|j dNƒt ||ƒtjƒtjƒ}tdƒ}t jtdOƒd„ƒjt jƒ} t jt jdOƒd…ƒjt jƒ} t jd†t jdQƒƒjt jƒ}g|D]}dRdSdTg|^q"}t|d| d/| dU|d|ƒ}|j dVƒt ||ƒtjƒtjd$d$dWd>gƒ}tdƒ}t jtdOƒd‡ƒjt jƒ} t jt jdOƒdˆƒjt jƒ} t jd‰t jdQƒƒjt jƒ}g|D]}dRdSdTg|^q)}t|d| d/| dU|d|dt ƒ}|j dXƒ} | jdŠƒt ||ƒtjƒtjƒ}tdƒ}t jtdOƒd‹ƒjt jƒ} t jt jdOƒdŒƒjt jƒ} t jdt jdQƒƒjt jƒ}g|D]}dRdSdTg|^q6}t|d| d/| dU|d|dZdSdRdTgƒ}|j d[ƒt ||ƒtj ƒtjƒ}t jd ƒ}d\d]d^d_g} d]d^d\d_g}t|d| d`|d|ƒ}|j daƒt ||ƒtjƒtdƒ}gtdƒD]}db|dc|f^qF} dd„}de„}tjƒ}t|d| d|d"d df|dg|ƒ}|j dhƒt ||ƒtjƒtdƒ}gtdƒD]}db|dc|f^që} tjƒ}t|d| d|d"d didjƒ}|j dkƒt ||ƒtjƒtjd$d$d>d&gƒ}d-t jjdd ƒ}g} d}xSt%| ƒd krôx0tt jjd!d ƒƒD]}| j&|ƒqÐW|d!7}q¢Wt jt jdƒdŽƒjt jƒ} g| D]}d?d@g|^q#} t|d| d/| d|dAtƒ}|jƒ\}}tj ||dBdCd!dDdEƒ}|j!t ƒ|j dlƒt ||ƒtrÐ|j'ƒndS(Niÿÿÿÿ(tPdfPagesstest_dotplot.pdfitaxsBasic horizontal dotplott horizontalsBasic vertical dotplottfigsizeii i(sTall and skinny dotplots x axis labelsShort and wide dotplots y axis labeltstripedsTall and skinny striped dotplotiöÿÿÿi2sShort and wide striped dotplots'Basic horizontal dotplot with few liness%Basic vertical dotplot with few linesis&Dotplot with adjusted horizontal rangetABCDEFGHiitsizetliness4Dotplot with user-supplied labels in the left margins::it split_namess1Dotplot with user-supplied labels in both marginsgš™™™™™¹?g)\Âõ(ì?gš™™™™™é?s:Vertical dotplot with user-supplied labels in both marginsgà?gö(\Âõð?gìQ¸…ë±?gö(\Âõè?g333333ë?icSs=i|]3}id|d6d|d6dd6dd6|“qS(trgbctcolortosvptmarkeritmsg333333ã?talpha((t.0tk((sFlib/python2.7/site-packages/statsmodels/graphics/tests/test_dotplot.pys s tstylest marker_propss&Dotplot with custom colors and symbolst intervalss Dotplot with symmetric intervalss0Dotplot with symmetric intervals (Pandas inputs)is#Dotplot with nonsymmetric intervalss,Vertical dotplot with nonsymmetric intervalst lightgreyRtroundtsolid_capstylet line_propss#Dotplot with custom line propertiesgè?tCattDogtstackeds center rightt numpointst handletextpadg-Cëâ6?s Dotplot with two points per linet styles_orders0Dotplot with two points per line (reverse order)s)Vertical dotplot with two points per lines9Vertical dotplot with two points per line (reverse order)iìÿÿÿtorangetpurples/Dotplot with color-matched points and intervalsiii tAxxtByytCzztsectionssDotplot with sectionsgÍÌÌÌÌÌì?sVertical dotplot with sectionsgHáz®Gñ?t section_orders(Dotplot with sections in specified ordertAtBtCtDt line_ordersDotplot with reordered liness%d::%didcSsd|S(Ntlft_((tx((sFlib/python2.7/site-packages/statsmodels/graphics/tests/test_dotplot.pytwtcSsd|S(Ntrgt_((R1((sFlib/python2.7/site-packages/statsmodels/graphics/tests/test_dotplot.pyR2xR3t fmt_left_nametfmt_right_names'Horizontal dotplot with name formattingt show_namestrightsShow right names onlys1Dotplot with different numbers of points per line(ii (i i(ii (i i(gà?gö(\Âõð?(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(ii(iii(ii(ii(iii(gà?gHáz®Gñ?(ii(ii(iii(ii((Rtmatplotlib.backends.backend_pdfRtNonetplttclftrangetaxesRt set_titleRtFalsetfiguretnptaranget set_xlabelt set_ylabeltTruetset_xlimtset_ylimtrandomtrandinttnormaltstrt set_positiontkrontonestastypetint32tpdtSeriestget_legend_handles_labelst figlegendt draw_frametdicttziptylimtlentappendtclose(t close_figurestreset_randomstateRRtpointsRRtvalsRR ttxtRRRtiR1RthandlestlabelstlegR#tlhtlR)R/tfmt_leftt fmt_righttii((sFlib/python2.7/site-packages/statsmodels/graphics/tests/test_dotplot.pyttest_allsŒ                                      2    A   A    --      !                    '#          '#           '#     '#      '#       $''&$    $''&     $''&!        -       -     "'#    (tnumpyRBtpytesttstatsmodels.graphics.dotplotsRtpandasRRR@Rtmatplotlib.pyplottpyplotR;t ImportErrorRtmarkt matplotlibRk(((sFlib/python2.7/site-packages/statsmodels/graphics/tests/test_dotplot.pyts