ó áp7]c@s•ddlZddlmZmZddlmZddlZddlm Z m Z yddl j Z Wnek rznXdefd„ƒYZdS(iÿÿÿÿN(t assert_raisest assert_equal(tSeries(tinteraction_plott_recodetTestInteractionPlotcBs¹eZed„ƒZejjd„ƒZejjd„ƒZejjejj dddgƒd„ƒƒZ ejjd„ƒZ ejjd„ƒZ ejjd „ƒZ d „ZRS( cCswtjjdƒtjjddddƒ|_tjjddddƒ|_tjtjjddddƒƒ|_dS(Ni90iitsizei<ii(tnptrandomtseedtrandinttweighttdurationtlogtdays(tcls((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyt setup_classsc Cs=t|j|j|jdddgdddgddƒ}dS( NtcolorstredtbluetmarkerstDt^tmsi (RR R R(tselft close_figurestfig((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyttest_plot_bothscCs1t|j|j|jdddgddƒ}dS(NRRRRi (RR R R(RRR((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyttest_plot_rainbowstastypetstrtintc CsÈt|jddƒj|ƒ}t|jddƒ}t|jddƒ}t|||dddgdd ƒ}|jd }|jƒjƒj ƒ}t |dƒt |j ƒd ƒt |j ƒdƒdS( NtnametWeighttDurationtDaysRRRRi is mean of Days( RR RR RRtaxest get_legendt get_titletget_textRt get_ylabelt get_xlabel( RRRR R RRtaxttrace((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyttest_plot_pandas!s  c CsPt|j|j|jdddgdddgƒ}tt|tjƒtƒdS(NRtrtgt linestyless--s-.( RR R RRt isinstancetplttFiguretTrue(RRR((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyttest_formatting/s3c Csˆttt|j|j|jddgƒttt|j|j|jddddgƒttt|j|j|jddd d gƒdS( NRRRtbR-R.R/s--s-.t:(Rt ValueErrorRR R R(RR((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyttest_formatting_errors4s(.cCst|j|j|jddƒ}tt|tjƒtƒt|j|j|jddƒ}tt|tjƒtƒt t t|j|j|jddƒdS(Ntplottypetlinetscattertunknown( RR R RRR0R1R2R3RR7(RRR((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyt test_plottype:s !!cCsqtddgddtjdddƒdd ƒ}t|idd6d d6ƒ}t|jj|jjd d ƒdS( NtaR5i tindexii(iR t index_testiterr_msgs_recode changed the index(RRtarangeRRR?tvalues(Rtseriestseries_((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyttest_recode_seriesBs ( (t__name__t __module__t classmethodRtpytesttmarkt matplotlibRRt parametrizeR,R4R8R=RF(((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyRs ' (tnumpyRt numpy.testingRRtpandasRRJt statsmodels.graphics.factorplotsRRtmatplotlib.pyplottpyplotR1t ImportErrortobjectR(((sJlib/python2.7/site-packages/statsmodels/graphics/tests/test_factorplots.pyts