ó áp7]c@`srdZddlmZmZmZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZmZmZddlmZdd lmZmZe jje jjeƒƒZd efd „ƒYZ d e fd „ƒYZ!de!fd„ƒYZ"de fd„ƒYZ#de#fd„ƒYZ$de fd„ƒYZ%de%fd„ƒYZ&de fd„ƒYZ'de'fd„ƒYZ(de fd„ƒYZ)de)fd„ƒYZ*d e fd!„ƒYZ+d"e+fd#„ƒYZ,d$efd%„ƒYZ-d&e-fd'„ƒYZ.d(e-fd)„ƒYZ/d*e-fd+„ƒYZ0d,e-fd-„ƒYZ1dS(.sÛ Tests for collapsed observation vector These tests cannot be run for the Clark 1989 model since the dimension of observations (2) is smaller than the number of states (6). Author: Chad Fulton License: Simplified-BSD i(tdivisiontabsolute_importtprint_functionN(tdatasets(tMLEModel(tFILTER_UNIVARIATE(tSMOOTH_CLASSICALtSMOOTH_ALTERNATIVEtSMOOTH_UNIVARIATE(tresults_kalman_filter(t assert_equaltassert_allcloset TrivariatecB`sæeZdZeeed„ƒZd„Zd„Zd„Z d„Z d„Z d„Z d„Z d „Zd „Zd „Zd „Zd „Zejjd„ƒZejjd„ƒZd„Zd„Zd„Zd„Zd„ZRS(sP Tests collapsing three-dimensional observation data to two-dimensional cK`s•tj|_tj|jddtjddddƒddd gƒd }tj|dƒ|d<|d d |d st!TestTrivariateUnivariateAlternatecB`s eZed„ƒZd„ZRS(cO`s#tt|ƒjdt||ŽdS(NR.(RlRƒR0R&(R,RwR/((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR0YscC`s|jjjdkst‚dS(Ni(R%RxRyR3(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRz^s(R`RaRcR0Rz(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRƒXst&TestTrivariateUnivariatePartialMissingcB`seZeed„ƒZRS(cK`stt|ƒj|||jj|jj|jj}tj|jj dd…dd…fR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR?+scC`st|jj|jjƒdS(N(R R1R@R4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRA1s cC`st|jj|jjƒdS(N(R R1RBR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRC7s cC`st|jj|jjƒdS(N(R R1RDR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRE=s cC`st|jj|jjƒdS(N(R R1RFR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRGCs cC`st|jj|jjƒdS(N(R R1RHR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRIIs cC`st|jj|jjƒdS(N(R R1RJR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRKOs cC`s#t|jj|jjddƒdS(NRLg-Cëâ6?(R R1RMR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRNUs  cC`s×t|jj|jjƒt|jjdd…dd…dd…f|jjdd…dd…dd…fddƒt|jjdd…dd…dd…f|jjdd…dd…dd…fdd ƒdS( NiiiiiRLg-Cëâ6?iÿÿÿÿgH¯¼šò×z>(R R1ROR4R¨RM(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRP\s  ++++cC`st|jj|jjƒdS(N(R R1RQR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRRps cC`st|jj|jjƒdS(N(R R1RSR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRTys cC`st|jj|jjƒdS(N(R R1RUR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRV€s cC`st|jj|jjƒdS(N(R R1RWR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRX†s cC`st|jj|jjƒdS(N(R RYRZRt(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR[Œs cC`st|jj|jjƒdS(N(R RYR\Rt(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR]”s cC`st|jj|jjƒdS(N(R RYR^Rt(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR_›s (R`RaRcR‹R”R0R:R;R=R?RARCRERGRIRKRNRPRfRgRhRRRTRVRXR[R]R_(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRˆ´s. >                tTestDFMClassicalSmoothingcB`s eZed„ƒZd„ZRS(cO`s#tt|ƒjdt||ŽdS(Nt smooth_method(RlR®R0R(R,RwR/((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR0£scC`sCt|jjtƒt|jjjtƒt|jjjtƒdS(N(R R%R¯Rt_kalman_smoothert_smooth_method(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyttest_smooth_method¨s (R`RaRcR0R²(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR®¢stTestDFMUnivariateSmoothingcB`s eZed„ƒZd„ZRS(cO`s#tt|ƒjdt||ŽdS(Nt filter_method(RlR³R0R(R,RwR/((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR0±scC`sCt|jjdƒt|jjjdƒt|jjjtƒdS(Ni(R R%R¯R°R±R(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR²¶s(R`RaRcR0R²(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR³°stTestDFMAlternativeSmoothingcB`s eZed„ƒZd„ZRS(cO`s tt|ƒjdt|dS(NR¯(RlRµR0R(R,RwR/((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR0¾scC`sCt|jjtƒt|jjjtƒt|jjjtƒdS(N(R R%R¯RR°R±(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR²Ãs (R`RaRcR0R²(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRµ½stTestDFMMeasurementDisturbancecB`s;eZed„ƒZd„Zd„Zd„Zd„ZRS(cO`s&tt|ƒjdtdd|dS(NR¯R©tnone(RlR¶R0R(R,RwR/((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR0ÌscC`s#t|jj|jjddƒdS(NRLgH¯¼šò×z>(R R1RUR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRVÑs cC`s2t|j|jjjƒj|jjddƒdS(NRLgH¯¼šò×z>(R R”R1RQtTR4(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyRRÖscC`s2t|j|jjjƒ|jjjddƒdS(NRLgH¯¼šò×z>(R R”RYR\R¸Rt(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR]ÛscC`s#t|jj|jjddƒdS(NRLgH¯¼šò×z>(R RYR^Rt(R9((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR_às (R`RaRcR0RVRRR]R_(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyR¶Ës    (2Rbt __future__RRRtnumpyR tpandasRRftost statsmodelsRt#statsmodels.tsa.statespace.mlemodelRt(statsmodels.tsa.statespace.kalman_filterRt*statsmodels.tsa.statespace.kalman_smootherRRRt(statsmodels.tsa.statespace.tests.resultsR t numpy.testingR R tpathtdirnametabspatht__file__t current_pathtobjectR RiRvR{R~RR€RRƒR„R…R†R‡RˆR®R³RµR¶(((sNlib/python2.7/site-packages/statsmodels/tsa/statespace/tests/test_collapsed.pyt sD    ©      î