!gjdZddlmZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z ddlZddlZddlmZddlmZmZmZmZddlmZmZddlmZmZej<j?ej<jAe!Z"d d ggd ggd ggd ggd d Z#d/d Z$d Z%dZ&dZ'dZ(ejRjTdZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2d0dZ3dZ4dZ5dZ6d1dZ7dZ8dZ9d Z:d!Z;d"ZejRjTd%Z?d&Z@d'ZAd(ZBd)ZCd*ZDd+ZEd,ZFd-ZGd.ZHy)2zM Tests for the generic MLEModel Author: Chad Fulton License: Simplified-BSD ) MONTH_ENDN)assert_assert_allcloseassert_almost_equal assert_equal assert_raises)nile) kalman_filterkalman_smoothersarimaxvarmax)MLEModelMLEResultsWrapper)results_sarimaxresults_var_miscapproximate_diffuse)k_statesdesign transition selection state_covinitializationctjddz}d|z}|rFtjddd}tj||}tj||}t j ||dd d d }|rHtj5tjd |jd }ddd||fSd}||fS#1swY|fSxYw)Nd? 1960-01-01MSperiodsfreqindex)rrrTF)exogordertime_varying_regressionmle_regressionuse_exact_diffuseignore)disp) nparangepd date_rangeSeriesr SARIMAXwarningscatch_warnings simplefilterfit)r6pandasendogr%r$modress k/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/tsa/statespace/tests/test_mlemodel.py get_dummy_modr</s IIcN3 E U7D  lCdC %u-yyU+ // D $U C   $ $ & #  ! !( +''r'"C # 8O 8O  # 8Os (CCc d}d}d}tjd}tj|dz}tjtj||zdz||f}tjtj|dzdz||f}tj|dz}tjtj|dzdz||f}tjtj||zdz||f} tjtj|dzdz||f} t ||||||||| |  } t | d|t | d|t | d |t | d |t | d |t | d | t | d | y)Nrr rr rk_posdef obs_interceptrobs_covstate_interceptrrrrCrrDrErrr)r-zerosr.reshaperr) k_endogrrBr8rCrrDrErrrr9s r;!test_init_matrices_time_invariantrIIsGHH HHW EIIg&,M ZZ 'H$%+gx-@BFjj7A:.4w6HIGii)C/OBIIhk2S88X:NOJ  (X%&,x.BDI 299Xq[1C7(H9MNI 58h!.v"O(I&  (C C(-8CM6*C NG,C)*O<C %z2C $i0C $i0c d}d}d}d}tjd}tjtj||zdz||f}tjtj||z|zdz|||f}tjtj|dz|zdz|||f}tjtj||zdz||f}tjtj|dz|zdz|||f} tjtj||z|zdz|||f} tjtj|dz|zdz|||f} t |||||||| | |  } t | d|t | d |t | d |t | d |t | d | t | d | t | d| y)Nr@rr>rr?rrArCrrDrErrr)r-rFrGr.rr) nobsrHrrBr8rCrrDrErrrr9s r;test_init_matrices_time_varyingrMjs DGHH HHW EJJryy483> '0M ZZ 'H$t+,s2Wh4MOFjj '1*t#$s*Wgt,DFGjj (T/"S(8T*:}t|j(t*j,t|j.t*j0t|j2t*j4|jAd|jCd|jEd|jGdt|jj(dt|jj.dt|jj2dt|jj6dt|j(t*j,t|j.t*j0t|j2t*j4|jAd |jj=|jj>}t|j(d t|j.dt|j2dy)NFr6rrr@rrrr)rrr@rrrgrh|?g _BgmBTgj@rg@(known stationaryefg)$r<rr-r.ssmshapeloglikelihood_burn toleranceinitial_variance isinstancerobjectinitialize_defaultinitialize_approximate_diffuseinitialization_typeapproximate_diffuse_varianceinitialize_knownconstantstationary_covinitialize_stationary filter_methodr FILTER_CONVENTIONALstability_methodSTABILITY_FORCE_SYMMETRYconserve_memoryMEMORY_STORE_ALLsmoother_outputr SMOOTHER_ALL_initialize_filter_kalman_filterset_filter_methodset_stability_methodset_conserve_memoryset_smoother_output)r9_kfs r; test_wrappingrys(u %FC^$cBIIcN&:;C!Q1a*+Q/"((+6TFCM("-"((&1''+C++Q/ 1 12CM""E*%%t,C))40 C..7> &&s+##779NO##@@#F"v&##77A##,,rd3##22bTF;##77F && (I(IJ   ..0((-*H*HI((/*F*FGGG   B!!=#D#DE$$m&L&LM##]%C%CD #S!C C &&,))3/((#.((#.!!=#D#DE$$m&L&LM##]%C%CD!GG    B!!1%$$c*##S)rJc|tj}tj|ddd}t j |dd}t j5t jd|jddd d }|jddd d }dddtjjd td \}}t j5t jd|jdd}dddtddgdy#1swYxYw#1swY%xYw)Ndatarrrrc)r&trendr*ncgropgF)methodr, optim_hessianoptim_complex_stepoim{Gz?)rtolrOr+T)r, return_params)rwpi1_stationaryr-diffr r2r3r4r5r6rllfr<r)truer8r9res1res2rw res_paramss r; test_fit_miscrs  * *D GGDL !!" %E //%y )r from_formularJr;z#test_from_formula.. sx/D/DQ1/MrJ)rNotImplementedErrorrrJr;test_from_formulars%'MNrJc dtjddgd}d}d}tjd}tj||d|d||d|df}|j |d d }t |||j |d d }t ||d |j |d d d }t ||d |j |d dd }t |||j |d dd } t | |d |j |d dd d } t | |d d} d} |j |} tjtjtj| | d| | dftj||d|d||d|df}|j | d d }t |||j | d d }t ||d |j | d d d }t ||d |j | d dd }t |||j | d dd } t | |d |j | d dd d } t | |d tjd}d}||d|d}tj5tjdt |j|dz|d t |j|dz|d dddy#1swYyxYw)Nr?rrrr&cBd|dzd|z|zzdz z|d|dzz zz S)Nrrrphisigma2s r; partial_phiz,test_score_analytic_ar1..partial_phi)s5sAv#f ,q01Vq36z5JKKrJc*dd|z|zdz z|dzz S)Nrr?rrs r;partial_sigma2z/test_score_analytic_ar1..partial_sigma2,s#qx#~,-;;rJ)rrT) transformedapprox_complex_stepFh㈵>atol)rrapprox_centeredharvey)rrr)rrrrcdd|dzzdzz S)Ngrrg?r)rs r;partial_transform_phiz6test_score_analytic_ar1..partial_transform_phiUsa#q&jD)))rJc d|zS)NrQr)rs r;partial_transform_sigma2z9test_score_analytic_ar1..partial_transform_sigma2Xs F{rJ)rrctjd}|dz dz |dzdz dzz |d<dd|dzzz x|d<|d<||zdz |d zz |d <|S) N)rrrrrrr+rr)rrrr>rS)r-rF)rrhessians r;rz(test_score_analytic_ar1..hessian~ss((6"q&1a!a7 (*a&!)m(<<  #, 9 rJr*r皙?)r r2r-r_rruntransform_paramsdotdiagr3r4r5_hessian_complex_step_hessian_finite_difference)r9rrranalytic_score approx_cs approx_fdapprox_fd_centered harvey_cs harvey_fdharvey_fd_centeredrruparamsranalytic_hessians r;test_score_analytic_ar1r#st //1c() 4CL<UU5\FUUF1Ivay)vay&), -.N  &d MII~. &d NII~D9 &d"&  (I~D9 &d8.24II~. &d8.35II~D96tH7<37#9&TB*$$V,GVV +GAJ7.wqz:;< = k&)VAY/VAYq 23 45N  'u$ OII~. 'u.35II~D9 'u%"&  (&TB 'uX.24II~. 'uX.35II~D97h7<37#9&TBUU7^Fvay&)4  "5h'11&9A=(t 566v>B(t 5 555s AL&&L/c<t\}}tj5tjd|j |j dd}t |jdd|j |j dd}t |jdt |jdd|j |j dd }t |jd t |jdd |j |j dd }t |jd t |jdd |j |j dd }t |jd t |jdd|j |j dd}t |jdt |jdd|j |j dd}t |jdt |jddtjt5|j |j ddddddddy#1swYxYw#1swYyxYw)Nr*r+none)r,cov_type descriptionz!Covariance matrix not calculated.approxzLCovariance matrix calculated using numerical (complex-step) differentiation.rzmCovariance matrix calculated using the observed information matrix (complex-step) described in Harvey (1989).rzQCovariance matrix calculated using the outer product of gradients (complex-step).robustzQuasi-maximum likelihood covariance matrix used for robustness to some misspecifications; calculated using the observed information matrix (complex-step) described in Harvey (1989). robust_oim robust_approxzQuasi-maximum likelihood covariance matrix used for robustness to some misspecifications; calculated using numerical (complex-step) differentiation.invalid_cov_type) r<r3r4r5r6rrcov_kwdsrpytestraisesrrs r;test_cov_paramsrsEHC  "5Fh'ggcjjrFg; LL ' / 1ggcjjrHg=S\\8, LL '  ggcjjrEg:S\\5) LL ' @ A ggcjjrEg:S\\5) LL ' ( ) ggcjjrHg=S\\8, LL ' L M ggcjjrLgAS\\<0 LL ' L M ggcjjrOgDS\\?3 LL ' . / ]]. / F GGCJJR2DG E Fi5F5Fh F Fi5F5Fs$H4JJ5JJ JJctddgfit}t|jddgddgt|j ddgddg|j gd|j gd|jgd|jgdtd\}}t|jddgddgt|j ddgddg|j ddgd }t|jddg|j ddgd}t|jddgy) Nrrr>F)rrO T) rkwargsrtransform_paramsrfilterupdateloglike loglikeobsr<r)r9rwr:s r;test_transformrs1 Aq6 $V $CC((!Q01a&9C**Aq62QF;JJruJ%JJruJ%KKK&NN25N)u %FCC((!Q01a&9C**Aq62QF; **aV* .CCJJA' **aV* /CCJJA'rJctjddg}t|fit}|j gd}t t |tjd|j g}t t |tdt |jd|j gd}t t |tdt |jdy)NrrQT) return_ssmrr)r) r-arrayrrrrr_r FilterResultsrr)r8r9r:s r; test_filterrs HHb"X E 5 #F #C **RD* )CC!.s s/?/?rJa) rrrrr param_names _start_params _param_namesrrs@r; test_paramsrsj Aq6 $V $C%'?@"%CuC!!A3'3%(rJc(t|\}}t|jdd|jddj t|j ddt j|jdz t|jdy)Nr7rr) r<r fittedvaluesr8squeezeresidr-rFrLrr\)r7r9r:s r; check_resultsr suF+HC((,ciim.C.C.EF !" rxx ';<''+rJc4tdtdy)NFrT)rrs r; test_resultsrsrJc$tjddd}tjddg|}t|fit}|j g}|j }t|j|jft|jj|t|j d|j tddgfit}|j g}tt|j d y) N 1980-01-01 1981-01-01YSstartendr"rrr#)dynamicstring)r/r0r1rrrpredictrr[rLrget_predictionpredicted_meanrKeyError)datesr8r9r:rs r; test_predictrs MM ,T JE IIq!fE *E 5 #F #C **R.CkkmG ,C&&(77ACKK K5s{{}E Aq6 $V $C **R.C(CKK:rJctddgfit}|jg}|jd}t |t j ddzt |jdj|tjddd}ttjddg| fit}|jg}t |jdt j ddzt |jd t j ddzt |jdjt j ddzy) Nrrr@)steps)r@rrr r#z 1960-12-01) rrrforecastrr-ones get_forecastrr/r0r1)r9r:r r$s r; test_forecastr 7s Aq6 $V $C **R.C||"|%HHbggenq01C$$2$.==xH MM, =E 299aV51 ?CLL|L4bggenq6HIC$$2$.==GGENQ&(rJctjddd}tjgd|}t|fit}|j g}t |j}ttjd|dudttjd |dudttjd |dudtj5tjd tj|j j"dd|jd |j _|jd |j _|jdddy#1swYyxYw)Nrz 1984-01-01rr)rrr>rrr#zSample:\s+01-01-1980Tz\s+- 01-01-1984zModel:\s+MLEModelr*rr)r/r0r1rrrstrsummaryrresearchr3r4r5r-nanfilter_results_standardized_forecasts_error)rr8r9r:txts r; test_summaryrIs) MM ,T JE IIoU 3E 5 #F #C **R.C ckkm C2C8DdK-s34?F/5TA4H  "h'>@ff88; ;<8 ;>8 s &BE>>Fc t|fi|}t|jjdt|jjddt|jj ||ft|j jjdt|j jjddt|j jj ||ft|j jj|juxs!|jjj d|S)Nr C_CONTIGUOUST F_CONTIGUOUS) rrr8ndimflagsr[rZbase writeable)r8rLrHrr9s r; check_endogrds 5 #F #C#0$74/2##Q'$$^4d;$$wo6  cii'Hsyy/H/H+H  JrJc\tttddtttddtttddtdgfit}|j g}t |j jdggtdgfit}|j g}t |j jdggtdgfit}|j g}t |j jdggtdgfit}tt|jgddg}t|fit}|j gdgdgg}t|fit}|j gd}t|fit}|j gy)Nr)r8rrTrrQ)rrQ) r ValueErrorrrrrrr8r)r9r:r8s r;test_basic_endogr"{sd *ha!<*hcA>*hdQ? A3 !& !C **R.C##))QC51 B4 "6 "C **R.C##))QC51 D6 $V $C **R.C##))QC51 C5 #F #C*cjj"-HE e &v &CJJrNTB4LE e &v &CJJrN E e &v &CJJrNrJcL tjddg}t|fit}t |j j |jjudt |j j |udt |jjj |udd|d<t |j tjdjddt |jj|tjd}ttt|fittjddg}t |jdt |jddt |jd dt |j d t|fit}|j#gtjddgjdd}t |jdt |jddt |j d t|fit}|j#gtjddgjdd}t |jdt |jddt |j dtt$t|fittjddgjddj'}t |jdt |jd dt |j d t|fit}|j#gtjddgjddj'}t |jdt |jd dt |j dtt$t|fittjddgjddd}tt$t|fitddgd ggddgddggdggdggdggd d}tjddgddgg}t|fddi|}|j#gy)NrrQTrrrrrrr)rrrrrrrrDrrrrrR@rH)r-rrrrr8rr{ orig_endogrrGr TypeErrorrrrr[rr! transpose)r8r9kwargs2s r;test_numpy_endogr,s HHb"X E 5 #F #Csxx':'::DAu,d3$$))6=E!HBEE$K//156$$e, HHRLE)[%:6: HHb"X EQ^,d3^,d3d# e &v &CJJrN HHb"X  & &q! ,EQ^,d3f% e &v &CJJrN HHb"X  & &q! ,EQ^,d3f%*k5;F; HHb"X  & &q! , 6 6 8EQ^,d3f% e &v &CJJrN HHb"X  & &q! , 6 6 8EQ^,d3f%*k5;F; HHb"X  & &q!Q /E*k5;F; 1#t1a&1a&9IseA3%se/G HHr2hR) *E e 2Q 2' 2CJJrNrJctjddg}tjdtjddd}tjddg|}t |fit }|jgtjd d g|}ttt |fit tjddg|}t |fit }|jgtjd ddgi|}t |fit }|jgtjddgd d gd |}ttt |fit tjd ddgi|}t |fit }t|jj|jjudt|jj|udt|jjj j|udd|j"d<t|jt$j&dj)ddt+|jj|ddgdggddgddggdggdggdggdd}tjddgd d gd |}t |fddi|}|jgy)NrrQalwaysrrrrr#rbrRr')rr/Trrr$rrrrr&rH)r/r1r3r5r0rrrrr! DataFramerr8rr{r(valuesilocr-rrGr)r8rr9r+s r;test_pandas_endogr3sg IIr2h E (# MM ,T JE IIr2he ,E e &v &CJJrN IIsCj .E*k5;F; IIr2he ,E e &v &CJJrN LL#Bx 6E e &v &CJJrN LLBxr2h7u EE*k5;F; LL#Bx 6E e &v &Csxx':'::DAu,d3$$++00=tDEJJtBEE$K//156CHH''/ 1#t1a&1a&9IseA3%se/G LLBxr2h7u EE e 2Q 2' 2CJJrNrJct\}}|jjj}tj j ||j_|jd}|jd}t||tt|jd|jd}|jd}t||tjt5|jdddddtjt5|jdddd|jd}|jd}t||tjt5|jdddd|jddd |jd y#1swYxYw#1swYxYw#1swYHxYw) N)sizer jarqueberainvalidbreakvar)r alternativeljungboxdF)rr:use_f boxpierce)r<rrr[r-randomnormaltest_normalityrrrtest_heteroskedasticityrrr!test_serial_correlation)r9r:r[actualdesireds r;test_diagnosticsrF=sHC    < < B BE e$4  t  ,F   5GFG$%s'9'9)L  ( ( ( 5F)))FFVV,48rJc`tjjj}tjddd|_t |ddddd }d|jd <tjd |jd <d|jd <d|jd<tjd|jd<|jg}|jddddddf}t|ddgd|jddddf}t|ddgdy) NrHrIrrJrKrr4&k Crrr^r\rPgN@a3#@rDrrrrrrrrgke@rrrr;r@rrMrr+gCl;*@g-?rOrr7r6rgf W?g W?r)r r{rQr/r0r$rrZr-exprrCrrA)rRr9r:rDs r;test_diagnostics_nile_eviewsr]wsyy$$&++H]]<DIHN Q,t C CGGN!vvh/CGGO"#CGG !"CGG !#!1CGG  **R.C ( (  ( DQ2X NFFVUO$7  |  4QU ;FFXx0t)g?gQgQ@r9rgQ?) r r{rQr/r0r$rrZrrCrrArB)rRr9r:rDnorms r;#test_diagnostics_nile_durbinkoopmanr`s5yy$$&++H]]<DIHN Q,t C CGGN%CGGO"#CGG !"CGG !'CGG  **R.C ( (  ( CH MFFTF.   \  21 5D1gtAwQ (FF/d; ( ( ( ;D AFFTF.rJc^t\}}|j}|jyr)r<r summary_frame)r9r:rs r;test_prediction_resultsrcs) HC  "G rJc  tjtjgdtjddd}t j |dj|d<t j |d j|d <t j |d j|d <|jd dgdf}tj}tj|dddd}|j|d}t|j|d|jddd|j z z }|jdddt j |j z|j z z }|jdddt j t j |j z|j z z }t||dt||dt||dtj"}|j$dz }|j&t j |j z }t||dt||d|jddz }|jdt j |j z }t||dt||dtj(}t+j,|dd d!d"}|j|d}t|j|d|jddd#|j z z }|jddd$t j |j z|j z z }|jddd#t j t j |j z|j z z }t||dt||dt||dtj.}|j$d#z }|j&d$t j |j zz }t||dt||d|jdd#z }|jdd$t j |j zz }t||dt||dy)%N)invincconsumprz 1982-10-01QSrJ)columnsr$redln_invrfdln_incrg dln_consumpz 1960-04-01z 1978-10-01)rjrkrlrr}r)r&r~r\rraic lutkepohlr6rbicrhqic estat_aic estat_bicrrL unstructured)r&r~error_cov_typer\ )r/r0rlutkepohl_datar0r-logrloclutkepohl_ar1_lustatsr r2rrr info_criterianobs_effective lutkepohl_ar1rmrolutkepohl_var1_lustatsr VARMAXlutkepohl_var1)dtar8rr9r:rmrorps r;#test_lutkepohl_information_criteriarsP ,,''1JmmL,TB DCVVCJ',,.C NVVCJ',,.C NI/446C  GGL-9: ;E  1 1D //% *)3-. 0C **T(^ $CCGGT)_-   U;  7 3%% % &C   U;  7 rvvc(() )C,>,> > ?C   f[  9 BFF266#"4"456 69K9K K LDCe%Ce%D$v,'  ) )D ''A+C ''BFF3--. .CCk*+Ck*+  E "Q &C  E "RVVC,>,>%? ?CCk*+Ck*+  2 2D --V3'5! NC **T(^ $CCGGT)_-   U;  7 3%% % &C   U;  7 rvvc(() )C,>,> > ?C   f[  9 BFF266#"4"456 69K9K K LDCe%Ce%D$v,'  * *D ''E/C ''As1122 2CCk*+Ck*+  E "U *C  E "Q0B0B)C%C CCCk*+Ck*+rJc tjjjd}tjddd|_|j dd}|j dd}tj|dd }|jd g}tt|j|d d itt|j|d d itt|j|d d itt|j|diitt|j|diitt|j|dii|j dd}tj|j dt|d|_d}t!j"t|5|j|dddt!j"t|5|j|dddd}t!j"t|5|j||dddd}t!j"t|5|j||ddd|j dd}d}t!j"t|5|j|dddt!j"t|5|j|dddd}t!j"t|5|j||dddd}t!j"t|5|j||dddtj$|j ddj&}tj$|j ddj&j dd} tj|d|d } | jdd g} tj$|ddj&dd}d}t!j"t|5| j|dddt!j"t|5| j|dddd}t!j"t|5| j| |dddd}t!j"t|5| j| |dddy#1swY3xYw#1swY xYw#1swYxYw#1swYxYw#1swYzxYw#1swYRxYw#1swY&xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw)NrKrHrIrrJrTrT)r&concentrate_scalerrr) fit_kwargsrrrrr!r"zeGiven `endog` does not have an index that extends the index of the model. Expected index frequency is)matchzdGiven `exog` does not have an index that extends the index of the model. Expected index frequency is)r%z.The indices for endog and exog are not aligned)zJGiven `endog` does not have an index that extends the index of the model.$zIGiven `exog` does not have an index that extends the index of the model.$r&r%rg?)r r{rQr/r0r$r2r r2smoothrr!appendextendapplylenrrr1r1) rRendog1endog2r9r wrong_freqmessagenot_ctsendog3endog4mod2rs r; test_append_extend_apply_invalidr!sxyy$$&++H5H]]<DIHN ]]3B F ]]2b !F //& T JC ::se D*dkk6((35*dkk6((35*djj&((35*dkk6z2>NO*dkk6z2>NO*djj&j"=MNr"%J}}nnQZtEJ9G z 1  J z 1  J 9G z 1- F ,->G z 1- F ,- mmBr"GG z 1 G z 1 GG z 1* F )*>G z 1* F )*YYx}}Sb)00 1F YYx}}Sb)00 1 6 6rs ;F ??6-1 3D ;;Sz "Dii" ,,-bc2GG z 1 G z 1 GG z 1* F )*>G z 1* F )**e    ----********s*S-S:TTT! T.T;<U+U U"U.U:-S7:TTT!T+.T8;UUU"U+.U7:Vctjgddgdd}|jddg}|jddgd  }t |j t jy) N)rrrr)rrrTrrrr)r>r>r)rrr%)r r2rrrdtyper-float64)r9r:ps r;test_integer_paramsrssQ //)99,0 2C **aV C 4i 8A"**%rJcntjj||gj}tjj||gj}tjj||gj}t |j j j|t |j jj|t |jj j|t |j jj|t |jj j|t |jjj|t |jj j|t |jjj|t |jj j|t |jjj|t |jj j|t |jjj|yr)r/ MultiIndex from_product swaplevelr predictedr$equalsri predicted_covfiltered filtered_covsmoothed smoothed_cov)statesix predicted_ixcolspredicted_cov_ixfiltered_cov_ixsmoothed_cov_ixs r;check_states_indexr|s}}11 t'ikmm00"d<FFHOmm00"d<FFHO F   " " ) ), 78 F   $ $ + +D 12 F & & - -.> ?@ F   $ $ + +D 12 FOO ! ! ( ( ,- FOO # # * *4 01 F   % % , ,_ => FOO # # * *4 01 FOO ! ! ( ( ,- FOO # # * *4 01 F   % % , ,_ => FOO # # * *4 01rJczd}tjd|d}tjtj||}t j |d}|jgd}tj|d |d zd}tjd d g}t|j|||y) Nr@2000Mrr#rrrrrrrrrstate.0state.1) r/ period_ranger1r-rFr r2rIndexrrrLrr8r9r:rrs r;test_states_index_periodindexrs D vt# >B IIbhhtnB /E //%y 1C **_ %C??AqsKL 88Y * +Dszz2|T:rJcd}tjd|t}tjt j ||}t j|d}|jgd}tj|d|d zt}tjd d g}t|j|||y) Nr@rrr#rrrrrrr) r/r0rr1r-rFr r2rrrrrs r;test_states_index_dateindexrs D VT BB IIbhhtnB /E //%y 1C **_ %C==r!udQhYOL 88Y * +Dszz2|T:rJcd}tjtjd}tjtj ||}t j|d}|jgd}tjtjd}tjddg}t|j|||y) Nr@r#rrr rr) r/rr-r.r1rFr r2rrrrs r;test_states_index_int64indexrs D "))B- B IIbhhtnB /E //%y 1C **_ %C88BIIbM*L 88Y * +Dszz2|T:rJcd}tjd}tjtj||}t j |d}|jgd}tjd}tjddg}t|j|||tjd d d }tjtj||}t j |d}|jgd}tjd d d }tjddg}t|j|||y) Nr@r#rrrrrrr r>#) r/ RangeIndexr1r-rFr r2rrrrrs r;test_states_index_rangeindexrs D r B IIbhhtnB /E //%y 1C **_ %C==$L 88Y * +Dszz2|T: q"a B IIbhhtnB /E //%y 1C **_ %C==B*L 88Y * +Dszz2|T:rJcgd}tj|tjt5tj|ddddy#1swYyxYw)N)rrrT) invalid_kwarg)r r2rwarns FutureWarning)r8s r;test_invalid_kwargsrsB E OOE m $3T2333s AA)TF)Fr%)I__doc__statsmodels.compat.pandasrosrr3numpyr- numpy.testingrrrrrr7r/rstatsmodels.datasetsr statsmodels.tsa.statespacer r r r #statsmodels.tsa.statespace.mlemodelrr(statsmodels.tsa.statespace.tests.resultsrrpathdirnameabspath__file__ current_pathrr<rIrMryrmarksmokerrrrrrrrrrr rrr"r,r3rFrSr]r`rcrrrrrrrrrrrJr;rs 0  % L wwrwwx89 qcUA3%#qcU+ 41B%1Ph*V/4Oi5X9Fx(>&& ) , ;0($6.%PZz:z"4J9*=>!/HZ,zO*d&28 ; ; ;;83rJ