M/Ph  dZddlmZddlmZddlZddlZddlZddlZ ddl m Z m Z ddl ZddlZddlZddlmZmZddlmZmZmZmZmZddlmZmZdd lmZm Z m!Z!ej"#ej"$e%\Z&Z'ej(ej")e&d d d d Z*e*+dZ*dZ,dZ-gdZ.gdZ/ej0e/Z1ej2e.ej3e1ej4e1Z5ej6ddZ7djdZ8GddZ9ej:;dddZd"Z?d#Z@ej:;d$e-ej:;de,d%ZAd&ZBd'ZCd(ZDd)ZEej:jFej:;d$e-ej:;de,d*ZGej:;d$e-ej:;de,d+ZHd,ZId-ZJd.ZKd/ZLej6d0ZMej6d1ZNej6d2ZOej:;d$e-ej:;de,ej:;d3d4ej:;d5d d6ZPd7ZQd8ZRej:;d9gd:d;ZSd<ZTd=ZUd>ZVej:;d?gd@dAZWdBZXej:;dCgdDgdEgdFgdGZYej:;dCgdHgdIgdJgdKZZdLZ[dMZ\dNZ]dOZ^ej:;d$dPdgej:;ddPdgdQZ_dRZ`dSZadTZbej:;d?gdUej:;d$ddPgej:;dddPgdVZcdWZddXZedYZfej:;dZgd[d\Zgd]Zhej:;d^d_d`gej:;daddbgdcZiej:;d$ddPgej:;dddPgej:;ddded`gdfZjdgZkdhZldiZmdS)kz4 Author: Terence L van Zyl Modified: Kevin Sheppard ) MONTH_END) pytest_warnsN)assert_allcloseassert_almost_equal)ConvergenceWarning ValueWarning) PY_SMOOTHERS SMOOTHERSExponentialSmoothingHoltSimpleExpSmoothing)HoltWintersArgs_test_to_restricted)r to_restrictedto_unrestrictedresultszhousing-data.csvDATET) index_col parse_datesMSaddmulN)gVWD@gmV} 8@g\sG)@@gB@gHG@g'ҨX=@gt=B@gl%}E@ga+esH@gk ".?@gLB@g5[y5D@gP|zI@gr_!?@gӸ7}D@g(.E@gϚK@gV`@@g5| E@g=ϟ6F@g %M@gclA@g)[$(F@gĖMG@)z2005-03-01 00:00:00z2005-06-01 00:00:00z2005-09-01 00:00:00z2005-12-01 00:00:00z2006-03-01 00:00:00z2006-06-01 00:00:00z2006-09-01 00:00:00z2006-12-01 00:00:00z2007-03-01 00:00:00z2007-06-01 00:00:00z2007-09-01 00:00:00z2007-12-01 00:00:00z2008-03-01 00:00:00z2008-06-01 00:00:00z2008-09-01 00:00:00z2008-12-01 00:00:00z2009-03-01 00:00:00z2009-06-01 00:00:00z2009-09-01 00:00:00z2009-12-01 00:00:00z2010-03-01 00:00:00z2010-06-01 00:00:00z2010-09-01 00:00:00z2010-12-01 00:00:00freqindexmodule)scopectjd}|d}|}t ddD](}||dz ||zd||dz zz ||<)|dd}t jd|jdt}t j ||d S) Nri皙?2000-1-1periodsry)rname) nprandom RandomStatestandard_normalcopyrangepd date_rangeshaperSeries)rser'irs r/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/tsa/holtwinters/tests/test_holtwinters.pysesr7hs   q ! !B 4  A A 1d^^00Qx!A$qQx/! #$$A M*agajy I I IE 9Qe# . . ..c|jd}tj|}tj|}tj|} tj|} ||d<||d<||z| d<||dzd|z ||zzz|d<||d|z zd|z |zz|d<td|D]Z} ||| zd|z || dz || dz zzz|| <||| || dz z zd|z || dz zz|| <[|dd|ddz| dd<|dtjd|dz|dzz| dd<|| z } ||| | | fS)zQ Simple, slow, direct implementation of double exp smoothing for testing rr!N)r1r)zerosr.arange) xalphabetal0b0 nforecastnlvalsbxhatfterrs r6_simple_dbl_exp_smootherrJts  A HQKKE  A 8A;;D AE!H AaD2gDGqt|q5yR"W55E!H 58b= !QXO 3AaD 1a[[HH1Q4<1u9q1u!a%1H"IIauQx%A,./1t8qQx2GG!QrT{QqtW$DH 9ryIM22QrU: :AaaaD d(C !QT !!r8ceZdZedZdZdZej dddZ dZ d Z ejj d Zd Zd Zd ZdZdS)TestHoltWinterscRgd}gd}tj||}tj|jtj|j|_||_gd}gd}tj||}tj|jtj|j|_||_gd}gd}tj||}tj|jtj|j|_||_t|_gd|_ dS) N) 4 ({@g'֩g|@gz|@g2nzz@g[W|@g8h{@g4Rz@gK/{#dR~@gFEŠ@g#WUv@gF&@g2U`~@) 1996-12-31 00:00:001997-12-31 00:00:001998-12-31 00:00:001999-12-31 00:00:002000-12-31 00:00:002001-12-31 00:00:002002-12-31 00:00:002003-12-31 00:00:002004-12-31 00:00:00z2005-12-31 00:00:00z2006-12-31 00:00:00z2007-12-31 00:00:00r)g=U1@#J{/5@g477@g=,Ԛ:@g~jt:@ga<@g9>@gjt>@g />@g}8g?@gzI@@g`@@g*ÂC@g^vD@g.[D@)1990-12-31 00:00:001991-12-31 00:00:001992-12-31 00:00:001993-12-31 00:00:001994-12-31 00:00:001995-12-31 00:00:00rOrPrQrRrSrTrUrVrW)g pz~p@gʩap@gLQԙJp@gj;p@gn@Xq@gG?Xq@gX$r@gB]PGr@gʅJr@g%Ts@g;5Ls@gu7Ot@g;$.u@g!3u@gL0t@g9s@g($Zs@gX֞t@gt@giƢ)u@gv@g"u@g/z@gDL$z@gkz@guy@g$)&y@g}8'x@gy@g@Sy@g@y@)z1970-12-31 00:00:00z1971-12-31 00:00:00z1972-12-31 00:00:00z1973-12-31 00:00:00z1974-12-31 00:00:00z1975-12-31 00:00:00z1976-12-31 00:00:00z1977-12-31 00:00:00z1978-12-31 00:00:00z1979-12-31 00:00:00z1980-12-31 00:00:00z1981-12-31 00:00:00z1982-12-31 00:00:00z1983-12-31 00:00:00z1984-12-31 00:00:00z1985-12-31 00:00:00z1986-12-31 00:00:00z1987-12-31 00:00:00z1988-12-31 00:00:00z1989-12-31 00:00:00rYrZr[r\r]r^rOrPrQrRrS) g^>g{%jCg&HI@gV{6?gҏB?g*| W?gfv?g?) r/r2 DatetimeIndexr infer_freq oildata_oil air_ausairlivestock2_livestockaust start_params)clsdatarrarbrcs r6 setup_classzTestHoltWinters.setup_classsq        ie,, ,  BM+2C$D$D   &   "   "YtU++ +  2=1A#B#B   $    B    B "yu55%'%5 &39::& & & "$8     r8cjt|jdddd|j}t|jdddd|j}t |ddgd d t |d gd d dS) Nrr estimatedseasonal_periodstrendseasonalinitialization_methodrez2011-03-01 00:00:00z2011-12-01 00:00:00g_vN@g9vB@g q{G@gڊeI@)end)r rdfitrerpredict)selffit1fit2s r6 test_predictzTestHoltWinters.test_predict9s# I"-    #4,# - - $ I"-    #4,# - -   LL.0E F F 0 0 0     LL2L 3 3 0 0 0      r8ct|jjdddd|j}t |dgdddS) Nrjrrrkrlrqrrrs)r rdvaluesrurerforecastrwrxs r6 test_ndarrayzTestHoltWinters.test_ndarrayZsy# I "-    #4,# - -   MM!  BBBA     r8zOptimizer does not convergeF)reasonstrictct|jddddd}t|dgdd dS) Nrjr)rmrnrobhT)method use_brutesteps)g9#zN@g/lB@gaG@gHI@rsr rdrurr}r~s r6 test_forecastzTestHoltWinters.test_forecastgst# I    #TT# * *  MMM " "$H$H$H!     r8ct|jddd}t|jddd}t|jd}t|dd gd t|jgd d t|dd gd t|dd gd t|jddd t|jddddS)Nlegacy-heuristicrpr"F optimized333333?rkr!gbL~@rj) rNgO~|@g>V|@gAV{@g=0{@g{@g9D{@g2\=$|@gV|@gsY}@g@{`4'~@gsL~@gj=f]@gBX@smoothing_levelgc?? initial_levelg{@rs)r rarurr}levelparamsrwrxryfit3s r6test_simple_exp_smoothingz)TestHoltWinters.test_simple_exp_smoothingssX!  4F   #cU# # # "  4F   #cU# # # "  K   #%%  DMM!,,zlA>>> J    !   $ DMM!,,zlA>>>DMM!,,zlA>>>DK(9:HaHHHDK8*aHHHHHr8cTt|jdddd}t|jddddd}t|jdd dd }t|d gd dt|jgddt|jgddt|d gddt|d gdddS)Nrr皙?r"F)rsmoothing_trendrT) exponentialrprk damped_trendrprr)gzGE@gQF@gףp= G@g(\H@ǧI@)gH @gTHt @g#nЬ @g)r) @g涷:@g(j @g{{@gi}Z?gi@[?gx?g3`|?g3&A?gu++l?g@g`Nb?rj)rXgoDI6@gm|9@ghTc;@g~I>@gRB>@gKH{ʔ?@g]S L@@gA@@gh4?"@@gk @@g6@@gG3ywcA@gD@gYOE@gSWE@)gLF@gQG@gQI@gzGJ@g33333L@)glE@gHzE@gGzTF@gGzF@gHzG@)r rbrurr}rn fittedfcastrs r6 test_holtzTestHoltWinters.test_holts O3E   #c3%# H H  O"4    #c3%# H H  O"-    #c3# 7 7  MM!  AAA1     J   " '   *      $ )   ,  MM!  AAA1     MM!  AAA1     r8ctt|jd}t|jdd}|dd}t|jddd}|}t |jd d d t |jd t jd t |jd t jd t |jdddt |jdt jd t |jdd t| tsJt |jd dd t |jd dd t |jd dd t |jddd t |jddd t |jdd t| tsJt |jd dd t |jd dd t |jd dd t |jdddt |jddd t |jddt| tsJdS)NrkrTr\(\? least_squares) damping_trendr)rrrpr?rrrrg(\p@r! initial_trendgYi@g(\p@g(\@g(\@g ףp= ?g33333/p@gp= ף?g·@r) r rcrur rrr)nansse isinstancesummaryas_textstr)rwrxmod4fit4mod5fit5s r6test_holt_damp_fitz"TestHoltWinters.test_holt_damp_fits"  %[   #%%   %"-    xxd?xCC  %"-     xxzzDK(9:D!DDDDK(9:BFAFFFDK8"&!DDDDK8&!DDDDK8"&!DDDDHgq111$,,..0022C88888DK(9:D!DDDDK(9:D!DDDDK8$BBBDK8&!DDDDK8$BBBDHgq111$,,..0022C88888DK(9:D!DDDDK(9:D!DDDDK8$BBBDK8&!DDDDK8$BBBDHgq111$,,..0022C8888888r8ct|jdd}dddddd }||5|d }dddn #1swxYwY|D]#}t |j|||$||5|d }dddn #1swxYwYt |j|jt |jd |d d |D}||5|d }dddn #1swxYwYtj |j j d|j j drJt |j|j z ddgd}t tj|jd|jf|gd}t tj|jd |j f|dgd}t |j|dgd}t |d|ddS)NTrkrgl9+?gG:?rg<΍o@g"&R@)rrrrrFrrc"i|] \}}d|v || S)r).0kvs r6 z4TestHoltWinters.test_holt_damp_r..#s(JJJtq!9I9Ia9I9I9Ir8rgXnh@MbP?atol) gOύo@g!0|p@gkp@g-2Pp@gpnp@g MGUVq@g05q@g%v$r@gjOIr@gd?r@gM&fcSs@gs@g۰t@g{U,u@gu%5u@g hB4t@g yKs@gg#Us@gR}t@g7t@g|a&#u@g v@gnau@gW$y@ge :z@gEz@gZy@gWJ] y@gxw?x@gAy@gV'd y@g-?y@r) gկR@g%l@gHGr|@gI0n@gRv@g.[Pn@g"t@gN1@gaq|@g is@g&צ@gLh@g4x`(@gȌW<@gg\Lq@g$"aDc@g@gmȈ@gvi~2.@gDkA@g1P?m@gM@g)H>ı@gm>m>^   F* * * * * r8rLtrend_seasonal))rN)Nr)rrc|\}}tjd }tjt5t |||dddddS#1swxYwYdS)Ndr)rnrorm)r)onesrraises ValueErrorr )rrnror's r6test_negative_multipliativers%OE8  A z " "   UX                      sAAAroctjt5ttd|ddddddS#1swxYwYdS)NFTr)rnrorrm)rr TypeErrorr housing_data)ros r6test_dampen_no_trendrs y ! !                          sAAA)rrc tjtjd tjddd}t jt5t||dddddS#1swxYwYdS)Nrr$rr%rr!)rorm) r/r2r)rr0rrrr )ror's r6test_invalid_seasonalr+s  R]:sNNN   A z " "GGQAFFFFGGGGGGGGGGGGGGGGGGsA88A<?A<ctjt5tt jt t gdddddS#1swxYwYdS)Nr!axis)rrrr r/concatrrurr8r6 test_2d_datar4s z " " I|\2 ; ; ;  #%%%sAA))A-0A-cbt}t|j|_t jd5}t |ddd}t|dksJdt|dvsJ dddn #1swxYwY|j d ksJdS) NT)recordrrkrnrorpr!rr ) rr-listrwarningscatch_warningsr lenrrm)hd2wrs r6test_infer_freqr;s     CSYCI   - - -+" uuK   1vv{{{{QqT***** +++++++++++++++  2 % % % % % %sABBBrncltt||d}|}|ddddii}t|t sJ|jd|jzksJt|jtsJdS) Nrkr basinhoppingminimizer_kwargsrSLSQPrminimize_kwargsg)\(?) r rrurrrrrrdict)rnrorresres2s r6test_start_paramsrGs )    C ''))C 77+h-@A   D ckkmm++--s 3 3333 8dSWn $ $ $ $ dk4 ( ((((((r8ctttjdd}|ddS)Nrknownrrp?r)r rrrurs r6test_no_params_to_optimizerZsC "'*%   C GGCG     r8cttd}tjt5|t jdgddddS#1swxYwYdS)Nrkrrrq)r rrrrrur)arrayrs r6test_invalid_start_param_lengthrcs |; O O OC z " ".. RXse__---..................s*A''A+.A+cttdd}|}|d}t|t sJt|t sJd}|j|j|zksJ|d}t|j|jddS) Nrrkrnrpr rgh㈵>r)decimal) r rrurrrrrr)reset_randomstaterrrtolres3s r6test_basin_hoppingr#is E   C ''))C 77.7 ) )D ckkmm++--s 3 3333 dllnn,,.. 4 4444 C 8sw} $ $ $ $ 77.7 ) )D$(A666666r8c<ttdd}|}|d}tj|j|jksJtjdddf|jz }t|dt| tsJt| tsJdS)NrrkrT) remove_biasrr) r rrur)anyrrrmeanrrrr)rrrerr2s r6 test_debiasedr)xs E   C ''))C 77t7 $ $D 6#"d&77 8 8888  QQQT "T%6 6D S))) ckkmm++--s 3 3333 dllnn,,.. 4 444444r8ctt||dd}t|jddtt||d}t|jdddS)Nrkr)rnrorprr)rnror)r rrurr)rnrors r6test_float_boxcoxr+s )      cee CJ|,c222       cee  CJ|,c22222r8c 4tt||d}|}t|t sJ|j}tjd}tj tj t}|rdnd}tj d|z}|d} |d} |d} |d } tj | rd n| } |d } |d }| | | | || f|dd<|r |d |dd<tj|tj}|}tjdd ggdz}t&||f}t(||f}t+||||||d}t-||||||d}|||}|||}t/|||||}t/||dS)Nrkrrrrrsmoothing_seasonalrrrrinitial_seasonsrrsF)r rrurrrrrr1r)squeezeasarrayr;isnan ones_likeastypeint64r-rr r PyHoltWintersArgsrr)rnrorrrrr'mpr>r?gammaphir@rAxip_copyboundspy_funccy_func py_hw_args cy_hw_argssse_cysse_pys r6test_equivalence_cython_pythonrDs  )    C ''))C ckkmm++--s 3 3333 ZF  a D 2:l++,,AAA QA $ %E # $D ' (E  !C# '##CC  B  B 4B +AbqbE*()!"" a   ) )B VVXXF XSzlQ& ' 'FHe,-G5)*G"2vvq!T5IIJ VVQ4GGJ WQ # #F WQ # #FFF### WQ # #FFF#####r8c ttd}|}t|t sJtjtj |j }|j d}t||d|j ddd\}}}}}t||jt||jjdtjdzt||dt%td d }|}tjtj |j }|j d}|j d } t||| |j d|j d d\}}}}}t||jt||z|j|jzt||jt||jt||jjd|jjdtjgd zzt||dt|t sJdS)Nrkrrrrr)r?r@rArBr:rrrr)r!rrsrjr)r rrurrrrr)r0r1endogrrJrrrrr}r rrnr) rrr=r>lrErG_rFr?s r6test_direct_holt_addrIs{ \ M M MC ''))C ckkmm++--s 3 3333 2:ci(())A J( )E/ 5ssz/:saAq!QAsy!!!Asy~b)BGAJJ6777As||A''' E   C ''))C 2:ci(())A J( )E :' (D/   :o & :o & Aq!QD#*+++AE39sy0111Asy!!!Asy!!! 39>"  r 2RXooo5N5N N NAs||A''' ckkmm++--s 3 333333r8cptjd}d|dz}tj|dddf}||dddfz}|t }t|dd}|j d d ksJdS) Ni90r)rrr!rrkrrr) r)r*r+r,cumsumr4intr rur)r r3r4y_starr'rs r6test_integer_arrayrOs   u % %B R   * **A YqAw  F111a4A  A  k    cee :' (C / / / / / /r8ctjd}t|ddd}|ddd }|d }t |tjdtjd fd |ddd }|d }t |tjdtjd fd t ||d dS)NrrTrkrnrrpr!rg#B ;)rrrr)start g|=r)r)r<r rurvrr)rFrres1pred1rpred2s r6test_damping_trend_zerorWs IbMME  )    C 771c7 O OD LLqL ! !EE25bill!23%@@@@ 771c7 K KD LLqL ! !EE25bill!23%@@@@E5u------r8cngd}tjdt|}tj||}gd}tjdt|}tj||}t |d}t |d}t |d}t |d} t |dgd t |d d gd d t |dgd t |dd gd d t |dgd t |d d gdd t | dgd t | dd gdd dS)N)r(2r$)rRr&)rrjr r)rnr<r!r)rRrt)r_FPz2000-1-6z2000-1-8@)rc)r/r0rr2r rurrv) array_input_addseries_index_addseries_input_addarray_input_mulseries_index_mulseries_input_mulrxryrrs r6test_different_inputsrls&***O}#o"6"6y2BCC'''O}#o"6"6y2BCC u = = = A A C CD  0 > > > B B D DD u = = = A A C CD  0 > > > B B D DD a000 1! 44lllAFFF a000 :: 66 a a000 1! 44nnnaHHH a000 :: 66r8cbgd}tjddd}tj||S)N)DgKq >@gt&3@g(D!TQ9@g֋h;@gp @@gQ|7@g.4y<@gp= ׏A@g, PSkB@gj9@g0>@gE_A<@g RB@g^7@g9ѮBO=@g!A 3?@gfNĖA@g v3@g5^I =@gi㈵DA@g&SD@g+p:@gduG<@gPkwA@g5 SE@gYO08@gekU@@gDB@gIKF@gY=@gcz,B@gm2D@gl gH@gQIF?@gqB@gиp $kC@g6% outer(seasonal, FUN = "concat") %>% as.vector # innov from np.random.seed(0); np.random.randn(4) innov <- c(1.76405235, 0.40015721, 0.97873798, 2.2408932) params <- expand.grid(models, c(TRUE, FALSE)) results <- apply(params, 1, FUN = function(p) { tryCatch( simulate(ets(austourists, model = p[1], damped = as.logical(p[2])), innov = innov), error = function(e) c(NA, NA, NA, NA)) }) %>% t rownames(results) <- apply(params, 1, FUN = function(x) paste(x[1], x[2])) )guS@gS\U]YJ@g('UHuP@g҇Q@)g~ƅ!i@g%Ca@grto@glg@)gMۿj@gI)_@gJvlp@g @)gvT5Ak@guYLl>`@gq@Ȱq@g{@)gOIAO@gyuN@gfc]ܒO@g48EGP@)g{e@g3dV@g-`@g 'm@)gGZ*oGd@gx&1V@g"~`@gtm@AAAMAAMAMMMMAANMANMMN)gjMFS@g"I@gx P@gkCQ@)g ui'j@g#bJ$b@g~p@geN@)g.!ES@g+j0 I@g@H0P@gVCvQ@)g'i@gy-e@g qs@g8ψ@)gq= ףj@g_LZ@gzsj@gqq~@)gsk@gVyc@gQ3y@gc('U@)gQj@g\mŝa@gc]^s@geX@)g vöEO@g89@O@ghUP@gEBP@)g됛e@g( d?d@dAdBd dCdDdEtjdFdGdHdIdJdKd dLtjdMtjdNtjdOdPdQdRd dStjdTtjdUtjdVdWdXdYd dZd[d\tjd]d^d_d`dadbd dcdddetjdfdgdhdidjdkd dltjdmtjdntjdodpdqdrd dsdttjtjdudvtjtjtjtjd dwdxtjtjdydztjtjtjtjd d{d|tjtjd}d~tjtjtjtjd dtjtjtjdtjtjtjtjtjd dtjtjtjdtjtjtjtjtjd d }||dS)a The final state from the R model fits to get an exact comparison Obtained with this R script: library(magrittr) library(fpp2) library(forecast) concat <- function(...) { return(paste(..., sep="")) } as_dict_string <- function(named) { string <- '{' for (name in names(named)) { string <- concat(string, """, name, "": ", named[name], ", ") } string <- concat(string, '}') return(string) } get_var <- function(named, name) { if (name %in% names(named)) val <- c(named[name]) else val <- c(NaN) names(val) <- c(name) return(val) } error <- c("A", "M") trend <- c("A", "M", "N") seasonal <- c("A", "M", "N") models <- outer(error, trend, FUN = "concat") %>% outer(seasonal, FUN = "concat") %>% as.vector # innov from np.random.seed(0); np.random.randn(4) innov <- c(1.76405235, 0.40015721, 0.97873798, 2.2408932) n <- length(austourists) + 1 # print fit parameters and final states for (damped in c(TRUE, FALSE)) { print(paste("damped =", damped)) for (model in models) { state <- tryCatch((function(){ fit <- ets(austourists, model = model, damped = damped) pars <- c() # alpha, beta, gamma, phi for (name in c("alpha", "beta", "gamma", "phi")) { pars <- c(pars, get_var(fit$par, name)) } # l, b, s1, s2, s3, s4 states <- c() for (name in c("l", "b", "s1", "s2", "s3", "s4")) states <- c(states, get_var(fit$states[n,], name)) c(pars, states) })(), error = function(e) rep(NA, 10)) cat(concat(""", model, "": ", as_dict_string(state), ", ")) } } gq`?g7*Ic?g{Z:?gT_(\?g әoO@go[V:ڛ?gP @gxx|}"gV'g ptׅ*@) r>r?r9r:rGrEs1s2s3s4g`M?gjj,?g?g;ȍS?gse&M@g{(e#?g^Ǚ@gZEI@gQ΁f"g%>.@gaCu?g&{}Fl?g/(ޥ?g(\?gdO@g4P -?g??g<ub?g'p?g{?g6۹?g1 N8?g$7?g͖n?gO@g*?g7d?gy4?g$ޮ?gtI@?gt ?g} ?g0([?g3ZFUN@g^!_?gll$?gđ;?g^_(\?gz7ظN@g,kxY?g7\ʸ?gOMȸ?g-V4?gݳN@gp[s(?rrgCaj?gDtA&?g۵º?g.5GO@gZ<?giG@g=Bg C)gB;i(@g.i֥?g9Nl7?g<$?g"4_O@gl?gn#- @g+U6ޫgtEd'ghOe*@gynY?g3Y?gL@gxa @gEr?r9r:F)rrr.rrrGr:rErrrr)g@9?gY,?gOQ?g[iY@r!) repetitionsr random_errorsr)r)r1r ru_level_trend_seasonr&r2rTsimulaterr|)rnrorrrprr short_name model_nameexpectedstateruinnovsims r6test_simulate_expected_rrs(S44JE"Z%66H9MMJ0888:9&A*MNN$V,Z8       cgf  >El   3ZCJrN3ZCJrNDkCKODkCKODkCKODkCKO vbhs'(()) JGGGH I I KE ,,qauE, J JC#*a00000r8c,t|ddddd}t|dddj|dddjt|dd dj|dd djt|dd dj|dd dj|dd t jj|dd t j|dd tj dd |dd d|dd d j}|dd tj d j}tj ||ksJdS)zS check whether all keywords are accepted and work without throwing errors. rjrTrkrmrnrorrpr)anchor random_staterRr:rnrtz 2016-03-01r)rr bootstraprrN) r rurrr|scipystatsnormr)r*randnr+all)rprurrs r6test_simulate_keywordsrs )     cee Qqq 118 QwQ 77> Qr 229 Q|! <<C Qu1 55< Q|! <<C LL%+2BLCCCLL%+2B2B2D2DLEEELL")//!R2H2HLIIILL+L>>> ,,qbr, : : AC << r (=(=b(A(A      6#+  r8c Nt|dddddd}|dj}|ddd j}t j|d }t jt j||z d ksJdS)zE check if simulation results with boxcox fits are reasonable rjrrFrkT)rmrnrorrprrrrr!rrN) r rur}r|rr)r'rabs)rprurrr's r6test_simulate_boxcoxrFs )    cee||A%H ,,qbq, 9 9 @C 73Q   D 6"&))A- . ......r8ix)rrrKc|tjgdt||dz}tjt 5t |dd}dddn #1swxYwY|dj }|d|dzksJ|d |d zksJdS) N) iaNi>^iLiIi1iIiiiirrrF)rnrrrr:) r/r2r.rwarnsrr rur}r)rts_1modelrs r6test_forecast_indexr[s 9NNNBR     D ( ) )$ E   #%%  NNN $ $ *E 8rBw     9R      s%A44A8;A8ctjtd5tt jddddddS#1swxYwYdS)NzCan only dampen thematchrT)rrrrr r)rrr8r6test_error_dampenrks z)> ? ? ?>>RWS\\====>>>>>>>>>>>>>>>>>>s$A  AActjd}tjt d5t |dddddn #1swxYwYt |dzd }t|t sJt |dzdd }tjtd 5| d ddddS#1swxYwYdS) Nrzuse_boxcox must be TruerarrrprTrrzuse_boxcox was setF) r)r*r,rrrr rrru)r'rs r6test_error_boxcoxrps !!#&&A y(A B B BOOQ3gNNNNOOOOOOOOOOOOOOO q!t 5 5 5C c/ 0 0000  15G   C z)= > > >"" 5!!!""""""""""""""""""s#AA!AC$$C(+C(c tjtd5t|ddddn #1swxYwYtjtd5t|ddddddn #1swxYwYtjtd5t|ddgd  dddn #1swxYwYtjt5t|d dddn #1swxYwYtjt5t|d dddn #1swxYwYtjt5t|gd dddn #1swxYwYtjt5t|ddddddn #1swxYwYtjt5t|ddddddddn #1swxYwYt|dd}tjt5|d dddn #1swxYwYtjt5|ddddddS#1swxYwYdS)Nz+initialization is 'known' but initial_levelrrrzinitial_trend set but modelr)rprrzinitial_seasonal set but model)r"333333?皙?r)rprinitial_seasonal)r)r)rr"g?rj)rr)rnrpr@)rnrorprrrprTr)use_basinhoppingrrrrr rur7rs r6test_error_initializationrs| G   AA S@@@@AAAAAAAAAAAAAAA z)F G G G   ")                     z)I J J J   ")111                     z " "55S4444555555555555555 z " "55S4444555555555555555 z " "HHS3F3F3FGGGGHHHHHHHHHHHHHHH z " "   ")                     z " "   ")                      7#   C z " "## c"""############### z " "?? o>>>??????????????????s:>> BBB&CC C ,D  DD.E  EE0FFF4GGG8HH!HI33I7:I7J<<KKr)rr zL-BFGS-BTNCr Powellz trust-constrcrtjddg}i}t|d}||||}t |jddd t |jd d d t |tsJdS) Ngp= ף?g&@rkr)rrer rgʳ?rrtolrgnt&@) r)rr rurrrrrr)rr7svr rrs r6test_alternative_minimizersrs 4- BO s+ F F FC ''B   CCJ01:DIIIICJ/4HHHH ckkmm++--s 3 333333r8ct|d}ddi}tjt5||dddn #1swxYwYtjt5|d|dddn #1swxYwYdddii}tjt5|d |dddn #1swxYwYdd d ii}|d |}t |jtsJt | tsJdS) Nrkrargsanything)r rr r r rr ) r rrrrurrrrrr)r7rkwargsrs r6test_minimizer_kwargs_errorrs s+ F F FCj !F z " "(( '''((((((((((((((( z " "@@ ???@@@@@@@@@@@@@@@ 6:"6 7F z " "?? ~v>>>??????????????? 8W"5 6F ''' @ @C cj$ ' '''' ckkmm++--s 3 333333s5AAA6BB!BC((C,/C,r)rr?)rrr")rrrc (tj|}tjdgdz}tjddggdz}tt|||t ||tj|dSNTrsrr)r)rrrrr4r5rselr=s r6test_to_restricted_equivrsXf  F (D6A:  C XSzlQ& ' 'Ffc6**FCJJrx$8$8&AAr8)rr皙?)rr"r)rrrc tj|}tjdgdz}tjddggdz}t|tt |||||dSr)r)rrrrrs r6test_restricted_round_tiprs|Xf  F (D6A:  C XSzlQ& ' 'F fc6::CHHr8cddi}tjt5t||dddn #1swxYwYdddgi}tjt5t||dddn #1swxYwYddi}tjt 5t||dddn #1swxYwYddi}tjt 5t||dddn #1swxYwYdd i}tjt 5t||ddddS#1swxYwYdS) Nbad_key)rr)r=rrr)rrr)rr)gr)rrKeyErrorr rrr7r=s r6test_bad_boundsrs $F x 11S0000111111111111111#s ,F y ! !11S0000111111111111111 1F z " "11S0000111111111111111 ,F z " "11S0000111111111111111 -F z " "11S0000111111111111111111sW<AA&BB B,C  CC2DDD8EEEcddi}t||dd}t|dd}t|jd|jdd dS) Nr)rrrk)r=rprrrrr)r rurr)r7r=rrs r6test_valid_boundsrs ,F  F+    cc!! ; G G G K K L  D $% %& r8ct|d}|ddi5|}dddn #1swxYwY|jddksJt |tsJt|ddd}|dd i5|}dddn #1swxYwY|jdd ksJt |tsJt|ddd }|d d d 5|}dddn #1swxYwY|jdd ksJ|jdd ksJt |tsJdS)NrkrrrrTrQrrrrr")r.rr.)r rrurrrrrr7rrs r6test_fixed_basicrs s+ F F FC *C0 1 1ggii :' (C / / / / ckkmm++--s 3 3333  5t;   C $/ 0 0ggii :o &$ . . . . ckkmm++--s 3 3333  55    C ssKK L Lggii :* +s 2 2 2 2 :' (C / / / / ckkmm++--s 3 333333s5A  AA C..C25C22FFFct|d}tjt5|ddi5 dddn #1swxYwYdddn #1swxYwYtjt 5|ddi5 dddn #1swxYwYdddn #1swxYwYt|dd}tjt 5|dd d 5 dddn #1swxYwYdddn #1swxYwYt|ddd }tjt 5|dd d 5 dddn #1swxYwYdddn #1swxYwYddd }t|dd|d}tjt d5|ddd5|dddn #1swxYwYddddS#1swxYwYdS)Nrkrrrrg333333ӿrrrrrrrr.)rr)ffffff?r)rnror=rpz!After adjusting for user-providedrr)rr.)r rrrrrru)r7rr=s r6test_fixed_errorsr-s s+ F F FC x  ^^.4 5 5                   z " " ^^.5 6 6                    5    C z " " ^^LL M M                    55    C z " " ^^ #3 ? ?                     ",: N NF  )    C z)L M M M ^^ #3 ? ?     GGIII               s A(A A(A A(A A((A,/A, C $B2& C 2B6 6C 9B6 :C  C C ?D=D& D=&D* *D=-D* .D==EE4F2 F F2F F2"F #F22F69F61I H+ I+H/ /I2H/ 3II Irct|||d}|d}|jjsJ|ddi5|d}dddn #1swxYwY|jjsJt |tsJdS)NrkrT)rrr) r ru mle_retvalssuccessrrrrr)r7rnrorrs r6 test_bruterSs  58;   C ''D' ! !C ? """" *C0 1 1&&ggg%%&&&&&&&&&&&&&&& ? """" ckkmm++--s 3 333333sA22A69A6ctjt5t|dddddn #1swxYwYtjt5t|ddddddn #1swxYwYtjt5t|dt jdd ddddS#1swxYwYdS) Nr heuristicrrr)rnrrprrk)rorrpr)r7s r6test_fix_set_parametersrbs z " "   s+                    z " "   "4                     z " "   WR[["-                       s19==A==BB!&CCCct|d}tjtd5|ddi5|ddddn #1swxYwYddddS#1swxYwYdS) NrkrzCannot fix a parameterrrg?r"r)r rrrrrurs r6test_fix_unfixablerws s+ F F FC z)A B B B)) ^^.5 6 6 ) ) GGCG ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )))))))))))))))))))s5BA( B(A, ,B/A, 0BBBc|ddd}tjtd5t|d|ddddn #1swxYwYd d d }tjtd 5t|d|d ddddS#1swxYwYdS)N)rr")rrrzThe bounds for smoothing_trendrrrk)rnr=rp)rr)rrrz!The bounds for smoothing_seasonal)ror=rprrs r6test_infeasible_boundsr~sT!+ K KF z)I J J J uV;   #%%%",: N NF z)L M M M "-    #%%% s#&AAA>&B11B58B5)rkrrct||||}|}|jjsJt |tsJdS)Nr)r rurrrrrr)r7rrnrorrs r6test_initialization_methodsrsq  586   C ''))C ? """" ckkmm++--s 3 333333r8ct|d}|jdksJ|jjd|jdksJt |j|jdddS)Nrkrrr:)r rurresidr1r fcastvaluesr)r7rs r6test_attributesrsr s+ F F F J J L LC 519999 9?1 1 - - - -COS_RSS%9:::::r8ct|dzdd}tjtd5|ddddn #1swxYwY|}t |}tjd|sJt| t sJdS) NrTrrzuse_boxcox was set at modelrrzBox-Cox:[\s]*True) r rrrrurrrefindallrr)r7rrsumms r6test_summary_boxcoxrs   Q4{   C z)F G G G!! 4   !!!!!!!!!!!!!!! ''))C s{{}}  D :*D 1 1111 ckkmm++--s 3 333333sAAAc.ttj|d}|}t |tsJtj td5| dddddn #1swxYwYtj td5| dd tj d  dddn #1swxYwY| dd d tj td5| dd d dddn #1swxYwYtj td5| dd d ddddn #1swxYwYtj td5| dd d ddddS#1swxYwYdS)Nrrz error must berrunknown)rz If randomr)rYrY)rrr)rrz Cannot anchorrzArgument random_state bad_value)rrrzArgument random_errors bad_values) r r)r1rurrrrrrrremptyrs r6 test_simulater sJ  3{   C ''))C ckkmm++--s 3 3333 z 9 9 9** Ry )))*************** z 5 5 5MM Rz(9K9K LLLMMMMMMMMMMMMMMMLL:cL222 z 9 9 988 Rz$ 777888888888888888 z)@ A A A   j;                    z)A B B BGG Rz FFFGGGGGGGGGGGGGGGGGGsZB22B69B6+DDD E22E69E6F>>GG$H  HH index_typ)r0periodr. irregularc |jd}i}d}d}|dkrtjd|dzd}n|dkr tjd|dzt}n|dkrtj|dz}d |d <n|d krt jd}tj t j | dd |dz }t}d |d <tjddd}| |dd}| }|dd|_t|5t!|fdddd|}dddn #1swxYwYt|5|d} dddn #1swxYwYt'| tjsJtj| j|dS)Nrr r$$rr%r0r.rrmr rj)sizerKi r!)rRstopstepirrr)r1r/ period_ranger0r RangeIndexr)r*r+IndexrLrandintrr-rrr rur}rr2testingassert_index_equal) r7r rrwarning fcast_indexrr3rfcasts r6test_forecast_index_typesrs 9QB>rr!rrrct|ddddd}|dd||}|dkrd nd|f}|j|ksJ|d d||}|dkrd nd |f}|j|ksJdS) NrjrTrkrr!r)rrr)r!r)r)r rurr1)rprrrurexpected_shapes r6test_forecast_1_simulationr's )     cee ,, !=k   C)A--TTA{3CN 9 & & & & ,, 1M{   C*Q..UUR4EN 9 & & & & & &r8rrSc~t|d|||dd}|jjsJdS)Nrjrk)rnrormrp)r rurr)r7rnrorrs r6(test_estimated_initialization_short_datar)sV  ETE )      cee  ? """"""r8ctjd}tj|}d|j_t|dddd}|dd }| d }|j d d ksJtj d d|j d z}tj ||j d d}t|}||_t|jtjsJ|jjJ|jjJt#jt&d5t|dddd}dddn #1swxYwY|dd }t#jt&d5| d ddddS#1swxYwYdS)Ni` )rgdrrrrlT)rrifrrz 2020-01-01r)r&F)rreplacezA date index has been providedrz No supported)r)r*r,r/ DataFramerrr rur}r1r0choicesortedrr_ inferred_freqrrr)r r'df_yrfittedrrs r6test_invalid_indexr3"sV !!(++A V # # # # M,AGAJ ? ? ?E I  UA  F FE 5MMEDJ dj""2 3 3333 :? " " " : # + + + l*J K K K  $ "-                   YYY 6 6F l. 9 9 9&&f%%%&&&&&&&&&&&&&&&&&&s$E55E9<E92GGGcgd}t|dd}|}|jt|jdddS)N)rrrgY@rrrrg4@rr)r rurr)serieseses_fits r6test_initial_levelr8Hs[ 2 2 2F gT   BVVXXF MMFM/2D99999r8c ttdddd}tj|jd}tj|jdd}tj|jd d}ttdddd ||| }t |jd|t |jd|t |jd |dS) Nrjrrrkrlrrr!r/r)rmrnrorprrr)r rdrur)roundrr)rxlvlrnseasrys r6test_all_initial_valuesr=Ss  )      cee   (4;/ 0 0C HT[11 5 5E 8DK 12A 6 6D  %    cee DK0#666DK0%888DK 12D99999r8)r)n__doc__statsmodels.compat.pandasrstatsmodels.compat.pytestrosrrnumpyr) numpy.testingrrpandasr/r scipy.statsrstatsmodels.tools.sm_exceptionsrrstatsmodels.tsa.holtwintersr r r r r 2statsmodels.tsa.holtwinters._exponential_smoothersrr&statsmodels.tsa.holtwinters._smoothersr6rrpathsplitabspath__file__baserHread_csvjoinrasfreq SEASONALSTRENDSrgr to_datetimeidxr2r_r`rdfixturer7rJrLr parametrizerrrrrrrrr#r)rr+rDrIrOrWrlrprrrrrrrrrrrrrrrrrrrrrrrrr rr$r'r)r3r8r=rr8r6rXs 0/////222222 >>>>>>>> LLLLLLLL '--11 2 2ar{GLLy"455 ""4(( 4   4bnUry-R-c  c8J8JKKKLLLh// /""""4B  B  B  B  B  B  B  B  JD   Y//  0/ ^44GG54G & & &&))Y//))0/*))"!!!... 7 7 7 5 5 5&))Y//330/*)3$&))Y//)$)$0/*))$X&4&4&4R 0 0 0...&@J"J"J"Z2+2+2+jg+g+g+T &))Y//=11.11616121210/*)61r'''T///*44455   65  >>> " " "/?/?/?d    4 4   4 4 4 4  A A111$   4442###L5$-00eT]33 4 44310 4   *)))     <<<4-00dE]33444310 4;;; 4 4 4GGG*???"<"<">''?>10'.4-00dE]33!R)) # #*)4310 ##&#&#&L::::::::r8