ó áp7]c@`sdZddlmZmZmZddlmZddlZddlZ ddl m Z m Z ddl ZddlZddlmZddlmZddlmZdd lmZdd lmZejƒjZejd d d dddƒe_d„Zd„Z d„Z!d„Z"d„Z#d„Z$d„Z%d„Z&d„Z'd„Z(d„Z)d„Z*d„Z+d„Z,d„Z-d „Z.d!„Z/d"„Z0d#„Z1ej2j3d$„ƒZ4d%„Z5d&„Z6d'„Z7d(„Z8d)„Z9d*„Z:d+„Z;d,„Z<dS(-sV Tests for structural time series models Author: Chad Fulton License: Simplified-BSD i(tdivisiontabsolute_importtprint_function(tPY3N(t assert_equaltassert_allclose(t macrodata(tSpecificationWarning(t structural(tUnobservedComponents(tresults_structuraltstarts 1959-01-01tends 2009-07-01tfreqtQSc C`s'tt|ƒ}x|dD]}|jƒ}|j|dƒtjƒ}|jddƒ}|dk r‘tjdddt tƒd|ƒ|_ nd|krßt j |dƒ}|dd krÒ|j jƒ}n||dtatoltllf(tregister_matplotlib_converterstfigtrecordtdispiÿÿÿÿg-Cëâ6?(gø?i g@i0g2@i((tgetattrR tcopytupdatetdtatpoptNonetpdt date_rangetlentindextnptlogtvaluestsqueezeR t start_paramsttransform_paramstuntransform_paramsRtfiltertfreqstrtinfRtcycle_frequency_boundtpitgetRtmatplotlib.pyplottpyplottpandas.plottingRt ImportErrortfiguretplot_componentstwarningstcatch_warningstTruetfittsummary(tnamettruetmodelRR.R Rtmodt roundtriptres_trueR4RRRtpltRRtres((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pytrun_ucmsh     &            cC`stdƒdS(Nt irregular(RL(t close_figures((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_irregularuscC`sWit_trtnd}tr'dnd}tj|d|ƒtdƒWdQXdS(Ns Specified model does not containtmatchtfixed_intercept(Rt__warningregistry__RRR'tpytesttwarnsRL(RNtwarningRP((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_fixed_interceptys  cC`stdƒdS(Ntdeterministic_constant(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_deterministic_constant‚scC`stdƒdS(Nt random_walk(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_random_walk†scC`stdƒdS(Nt local_level(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_local_levelŠscC`sNtr tnd}trdnd}tj|d|ƒtdƒWdQXdS(Nsirregular component addedRPt fixed_slope(RRR'RSRTRL(RNRURP((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_fixed_slopeŽscC`sWit_trtnd}tr'dnd}tj|d|ƒtdƒWdQXdS(Nsirregular component addedRPR](RRRRRR'RSRTRL(RNRURP((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_fixed_slope_warn•s  cC`stdƒdS(Ntdeterministic_trend(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_deterministic_trendŸscC`stdƒdS(Ntrandom_walk_with_drift(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_random_walk_with_drift£scC`stdƒdS(Nt local_linear_deterministic_trend(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyt%test_local_linear_deterministic_trend§scC`stdƒdS(Ntlocal_linear_trend(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_local_linear_trend«scC`stdƒdS(Nt smooth_trend(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_smooth_trend¯scC`stdƒdS(Nt random_trend(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_random_trend³scC`stdƒdS(Ntcycle(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyt test_cycle·scC`stdƒdS(Ntseasonal(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyt test_seasonal»scC`stdƒdS(Nt freq_seasonal(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_freq_seasonal¿scC`stdƒdS(Ntreg(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_regÃscC`stdƒdS(Nt rtrend_ar1(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_rtrend_ar1ÇscC`stdƒdS(Ntlltrend_cycle_seasonal_reg_ar1(RL(RN((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyt#test_lltrend_cycle_seasonal_reg_ar1Ësc C`stjdƒd}|d}|ddd…cd7<|ddd…cd8(iiÿÿÿÿ( R,tarangeR?R@RAR tFalseRBRtregression_coefficientstfilteredR(tendogRtmod1tres1tmod2tres2((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyt test_mle_regÐs c C`s¦it_ddg}tr!tnd}tr3dnd}tj|d|ƒ!t|ƒ}t|j dƒWdQXtj t ƒt|dƒWdQXtr tnd}tr²dnd}tj|d|ƒ-t|dt dt ƒ}t|j d ƒWdQXddd d g}xk|D]c}it |6}tr4tnd}trFd nd}tj|d|ƒt|d |WdQXqWtj t ƒt|d dƒWdQXdS(Niisirregular component addedRPRMs invalid specs!Trend component specified withoutttrendsdeterministic trendtstochastic_leveltstochastic_trends may be overridden when the trendRn( RRRRRR'RSRTR Rttrend_specificationtraisest ValueErrorRA(R}RURPRGttrend_attributest attributeR((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyttest_specificationsæs0       cC`s8tdƒ}tjd}tjd}tjjdƒtjtj|ƒtj|ƒdf}tjjd|ƒ}tj |d ƒ}xMt d |ƒD]<}|d |||d ||d ||||d @gÍÌÌÌÌÌì?iiiÿÿÿÿg"@g@gÍÌÌÌÌÌì¿RgH¯¼šò×z>iþÿÿÿii iiiiii iiiii(R%R#R RARzR,R7R$Rtk_statest k_state_covtloglikelihood_burnRtssmtk_posdeftk_paramsR*Rtreshapetdesignt __direct_sumtarraytcostsint transitionR•teyet selectiontdiagt state_cov(R.RFRtexpected_designtexpected_transitiontexpected_selectiontexpected_state_cov((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyt(test_matrices_somewhat_complicated_modelwsh       ,KFEF=D?H/%%%,cC`s¤tjg|D]}|j^q ddƒ}tj|ƒ}d}x]|D]U}|j\}}||ksnt‚|||||…|||…f<||7}qGW|S(sLCompute the matrix direct sum of an iterable of square numpy 2-d arrays taxisi(R,tsumtshapeR•tAssertionError(tsquare_matricestmt new_shapet new_arraytoffsettrowstcols((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pyRË¿s+ $(=t__doc__t __future__RRRtstatsmodels.compatRR?RR,t numpy.testingRRtpandasR(RStstatsmodels.datasetsRtstatsmodels.tools.sm_exceptionsRtstatsmodels.tsa.statespaceRt%statsmodels.tsa.statespace.structuralR t(statsmodels.tsa.statespace.tests.resultsR t load_pandastdataR%R)R+RLRORVRXRZR\R^R_RaRcReRgRiRkRmRoRqRsRutmarktslowRwR‚R‹RR¥R·R¼RØRË(((sOlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_structural.pytsT    ! Y                  +  7 H