'YHh  p dZddlmZddlmZddlZddlZddlZddlZ ddl m Z m Z ddl ZddlZddlZddlmZmZddlmZmZmZmZmZddlmZmZdd lmZm Z m!Z!ejDjGejDjIe%\Z&Z'ejPejDjSe&d d d d Z*e*jWdZ*dZ,dZ-gdZ.gdZ/ej`e/Z1ejde.ejfe1ejhe1Z5ejlddZ7djdZ8GddZ9ejtjwdddZd"Z?d#Z@ejtjwd$e-ejtjwde,d%ZAd&ZBd'ZCd(ZDd)ZEejtjejtjwd$e-ejtjwde,d*ZGejtjwd$e-ejtjwde,d+ZHd,ZId-ZJd.ZKd/ZLejld0ZMejld1ZNejld2ZOejtjwd$e-ejtjwde,ejtjwd3d4ejtjwd5d d6ZPd7ZQd8ZRejtjwd9gd:d;ZSd<ZTd=ZUd>ZVejtjwd?gd@dAZWdBZXejtjwdCgdDgdEgdFgdGZYejtjwdCgdHgdIgdJgdKZZdLZ[dMZ\dNZ]dOZ^ejtjwd$dPdgejtjwddPdgdQZ_dRZ`dSZadTZbejtjwd?gdUejtjwd$ddPgejtjwdddPgdVZcdWZddXZedYZfejtjwdZgd[d\Zgd]Zhejtjwd^d_d`gejtjwdaddbgdcZiejtjwd$ddPgejtjwdddPgejtjwddded`gdfZjdgZkdhZldiZmy)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)scopecjtjjd}|jd}|j }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 k/var/www/html/planif/env/lib/python3.12/site-packages/statsmodels/tsa/holtwinters/tests/test_holtwinters.pysesr7hs   q !B 4 A A 1d^0Qx!A$qQx/!0 #$A MM*aggajy IE 99Qe# ..cz|jd}tj|}tj|}tj|} tj|} ||d<||d<||z| d<||dzd|z ||zzz|d<||d|z zd|z |zz|d<td|D]H} ||| zd|z || dz || dz zzz|| <||| || dz z zd|z || dz zz|| <J|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 HHQKE  A 88A;D AE!H AaD2gDGqt|q5yR"W55E!H 58b= !QXO 3AaD 1a[H1Q4<1u9q1u!a%1H"IIauQx%A,./1t8qQx2GG!H Qr{QqW$DH 9ryyIM2QrU: :AaD d(C !QT !!r8ceZdZedZdZdZejjdddZ dZ d Z ejjd Zd Zd Zd ZdZy)TestHoltWinterscgd}gd}tj||}tj|jtj|j|_||_gd}gd}tj||}tj|jtj|j|_||_gd}gd}tj||}tj|jtj|j|_||_t|_gd|_ y) 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_classs    iie, ,,   BMM+2C2C$D & " "YYtU+ ++   2==1A1A#B $  B  B "yyu5%'%5%5 & &399:& "$8   r8cPt|jddddj|j}t|jddddj|j}t |j ddgd d t |j d gd d y) 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# II"-  #4,,# - $ II"-  #4,,# -   LL.0E F 0   LL2L 3 0  r8ct|jjddddj|j}t |j dgddy) Nrjrrrkrlrqrrrs)r rdvaluesrurerforecastrwrxs r6 test_ndarrayzTestHoltWinters.test_ndarrayZsW# II  "-  #4,,# -   MM! BA r8zOptimizer does not convergeF)reasonstrictct|jdddjdd}t|j dgdd y) Nrjr)rmrnrobhT)method use_brutesteps)g9#zN@g/lB@gaG@gHI@rsr rdrurr}r~s r6 test_forecastzTestHoltWinters.test_forecastgsL# II  #TT# *  MMM "$H! r8c6t|jdjdd}t|jdjdd}t|jdj}t|j dd gd t|j gd d t|j dd gd t|j dd gd t|j ddd t|j dddy)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_smoothingss!   4F #cU# # "   4F #cU# # "   K #%  DMM!,zlA> JJ  ! $ DMM!,zlA>DMM!,zlA>DKK(9:HaHDKK8*aHr8ct|jdjddd}t|jddjddd}t|jdd jdd }t|j d gd dt|j gddt|j gddt|j d gddt|j d gddy)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  OO3E #c3%# H  OO"4  #c3%# H  OO"-  #c3# 7  MM! A1   JJ " ' *     $ ) ,  MM! A1   MM! A1 r8ct|jdj}t|jdd}|jdd}t|jddd}|j}t |j d d d t |j d t jd t |j d t jd t |j dddt |j dt jd t |jdd t|jjtsJt |j d dd t |j d dd t |j d dd t |j ddd t |j ddd t |jdd t|jjtsJt |j d dd t |j d dd t |j d dd t |j dddt |j ddd t |jddt|jjtsJy)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_fitsU"  % %[ #%   % %"-  xxd?xC  % %"-   xxzDKK(9:D!DDKK(9:BFFAFDKK8"&&!DDKK8&!DDKK8"&&!DDHHgq1$,,.002C888DKK(9:D!DDKK(9:D!DDKK8$BDKK8&!DDKK8$BDHHgq1$,,.002C888DKK(9:D!DDKK(9:D!DDKK8$BDKK8&!DDKK8$BDHHgq1$,,.002C888r8c t|jdd}dddddd }|j|5|jd }ddd|j D]}t j ||| |j|5|jd }dddt jjt |j d |d |jDcic] \}}d |vs ||}}}|j|5|jd } dddtj jjd|jjdrJt |j|jz ddgd} t tj|j d|jf| gd} t tj|j d |jf| dgd} t |j | dgd} t |j#d| dy#1swYxYw#1swYxYwcc}}w#1swYIxYw)NTrkrgl9+?gG:?rg<΍o@g"&R@)rrrrrFrrrrgXnh@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>ı@g^ F* r8rLtrend_seasonal))rN)Nr)rrc|\}}tjd }tjt5t |||ddddy#1swYyxYw)Ndr)rnrorm)r)onesrraises ValueErrorr )rrnror's r6test_negative_multipliativersN%OE8  A z "  UX    s AAroctjt5ttd|dddddy#1swYyxYw)NFTr)rnrorrm)rr TypeErrorr housing_data)ros r6test_dampen_no_trendrs: y !       s 8A)rrctjtjd tjddd}t j t5t||ddddy#1swYyxYw)Nrr$rr%rr!)rorm) r/r2r)rr0rrrr )ror's r6test_invalid_seasonalr+s_  R]]:sN A z "GQAFGGGs A33A<ctjt5tt j t t gdjdddy#1swYyxYw)Nr!axis)rrrr r/concatrrurr8r6 test_2d_datar4sE z " II|\2 ; #%s 9AA%c<tj}t|j|_t j d5}t |ddd}t|dk(sJdt|dvsJ dddjd k(sJy#1swYxYw) NT)recordrrkrnrorpr!rr ) rr-listrwarningscatch_warningsr lenrrm)hd2wrs r6test_infer_freqr;s    CSYYCI   -+" uuK 1v{{QqT*** +   2 %% % ++s 2BBrncHtt||d}|j}|jddddii}t|j j t sJ|jd|jzksJt|jtsJy) Nrkr basinhoppingminimizer_kwargsrSLSQPrminimize_kwargsg)\(?) r rrurrrrrrdict)rnrorresres2s r6test_start_paramsrGs )  C '')C 77+h-@A  D ckkm++-s 33 3 88dSWWn $$ $ dkk4 (( (r8cntttjdd}|jdy)Nrknownrrp?r)r rrrurs r6test_no_params_to_optimizerZs0 "''*% C GGCG r8cttd}tjt5|j t jdgdddy#1swYyxYw)Nrkrrrq)r rrrrrur)arrayrs r6test_invalid_start_param_lengthrcsG |; OC z ". RXXse_-...s 'AA$cttdd}|j}|jd}t|j j t sJt|j j t sJd}|j|j|zksJ|jd}t|j|jdy) Nrrkrnrprrgh㈵>r)decimal) r rrurrrrrr)reset_randomstaterr r tolres3s r6test_basin_hoppingr is E C '')C 77.7 )D ckkm++-s 33 3 dlln,,. 44 4 C 88sww} $$ $ 77.7 )D$((A6r8cttdd}|j}|jd}tj|j |j k7sJtj dddf|j z }t|jdt|jjtsJt|jjtsJy)NrrkrT) remove_biasrr) r rrur)anyrrrmeanrrrr)rr r err2s r6 test_debiasedr&xs E C '')C 77t7 $D 66#""d&7&77 88 8   QT "T%6%6 6D S) ckkm++-s 33 3 dlln,,. 44 4r8ctt||ddj}t|jddtt||dj}t|jddy)Nrkr)rnrorprr)rnror)r rrurr)rnror s r6test_float_boxcoxr(su )    ce CJJ|,c2     ce  CJJ|,c2r8c tt||d}|j}t|j j t sJ|j}tjd}tjtjt}|rdnd}tjd|z}|d} |d} |d} |d } tj| rd n| } |d } |d }| | | | || f|dd|r|d |ddtj|jtj }|j#}tj$dd ggdz}t&||f}t(||f}t+||||||d}t-||||||d}|||}|||}t/|||||}t/||y)Nrkrrrrrsmoothing_seasonalrrrrinitial_seasonsrrsF)r rrurrrrrr1r)squeezeasarrayr;isnan ones_likeastypeint64r-rr r PyHoltWintersArgsrr)rnrorr rrr'mpr>r?gammaphir@rAxip_copyboundspy_funccy_func py_hw_args cy_hw_argssse_cysse_pys r6test_equivalence_cython_pythonrAs )  C '')C ckkm++-s 33 3 ZZF   a D 2::l+,AAA QA $ %E # $D ' (E  !C##CC  B  B 4B +AbqE()!" a   )B VVXF XXSzlQ& 'FHe,-G5)*G"2vvq!T5IJ VVQ4GJ Q #F Q #FFF# Q #FFF#r8c ttd}|j}t|j j t sJtjtj|j}|jd}t||d|jddd\}}}}}t||jt||jjdtj dzt||j#dt%td d }|j}tjtj|j}|jd}|jd } t||| |jd|jd d\}}}}}t||j&t||z|j|j(zt||jt||j(t||jjd|j(jdtj*gd zzt||j#dt|j j t sJy)Nrkrrrrr)r?r@rArBr:rrrr)r!rrsrjr)r rrurrrrr)r-r.endogrrJrrrrr}r rrnr) rr r=r>lrErG_rFr?s r6test_direct_holt_addrFs \ MC '')C ckkm++-s 33 3 2::cii()A JJ( )E/ 5sszz/:saAq!QAsyy!Asyy~~b)BGGAJ67As||A' E C '')C 2::cii()A JJ( )E ::' (D/   ::o & ::o & Aq!QD#**+AE399syy01Asyy!Asyy! 399>>"  r 2RXXo5N N NAs||A' ckkm++-s 33 3r8cDtjjd}d|jdz}tj|dddf}||dddfz}|j t }t|ddj}|jd d k7sJy) Ni90r)rrr!rrkrrr) r)r*r+r,cumsumr1intr rur)rr3r4y_starr'r s r6test_integer_arrayrLs   u %B R   **A YYqAw F1a4A  A  k  ce ::' (C // /r8ctjd}t|ddd}|jddd }|j d }t |tj dtjd fd |jddd }|j d }t |tj dtjd fd t ||d y)NrrTrkrnrrpr!rg#B ;)rrrr)start g|=r)r)r<r rurvrr)rCrres1pred1r pred2s r6test_damping_trend_zerorTs IIbME  )  C 771c7 OD LLqL !EE255biil!23%@ 771c7 KD LLqL !EE255biil!23%@E5u-r8cgd}tjdt|}tj||}gd}tjdt|}tj||}t |dj }t |dj }t |dj }t |dj } t |jdgd t |jd d gd d t |jdgd t |jdd gd d t |jdgd t |jd d gdd t | jdgd t | jdd gdd y)N)r(2r$)rOr&)rrjr r)rnr<r!r)rOrt)r\FPz2000-1-6z2000-1-8@)r`)r/r0rr2r rurrv) array_input_addseries_index_addseries_input_addarray_input_mulseries_index_mulseries_input_mulrxryrrs r6test_different_inputsrisx*O}}#o"6yy2BC'O}}#o"6yy2BC u = A A CD  0 > B B DD u = A A CD  0 > B B DD a0 1! 4lAF a0 :: 6 a a0 1! 4naH a0 :: 6r8cfgd}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?r6r7rDrEs1s2s3s4g`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(?rogCaj?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?r6r7F)rrr+rrrDr:rErrrr)g@9?gY,?gOQ?g[iY@r!) repetitionsr random_errorsr)r)r.r ru_level_trend_seasonr#r/rTsimulaterr|)rnrorrrmrr short_name model_nameexpectedstateruinnovsims r6test_simulate_expected_rrs{(S4JE"Z%66H9MMJ088::9&A*MN$V,Z8     cgf  >El   3ZCJJrN3ZCJJrNDkCKKODkCKKODkCKKODkCKKO vvbhhs''() JJGH I K KE ,,qauE, JC#**a0r8ct|dddddj}t|jdddj|jdddjt|jdd dj|jdd djt|jdd dj|jdd dj|jdd t j j|jdd t j j|jdd tjjdd |jdd d|jdd d j}|jdd tjjd j}tj||k(sJy)zS check whether all keywords are accepted and work without throwing errors. rjrTrkrmrnrorrpr)anchor random_staterOr:rkrtz 2016-03-01r)rr bootstraprrN) r rurrr|scipystatsnormr)r*randnr+all)rmrur r s r6test_simulate_keywordsrs )   ce Qqq 188 QwQ 7>> Qr 299 Q|! <CC Qu1 5<< Q|! <CC LL%++2B2BLCLL%++2B2B2DLELL"))//!R2HLILL+L> ,,qbr, : A AC << r (=(=b(A   f  66#+  r8c Dt|ddddddj}|jdj}|j ddd j}t j |d }t jt j||z d ksJy)zE check if simulation results with boxcox fits are reasonable rjrrFrkT)rmrnrorrprrrrr!rrN) r rur}r|rr)r$rabs)rmrurr r$s r6test_simulate_boxcoxrFs )  ce||A%%H ,,qbq, 9 @ @C 773Q D 66"&&)A- .. .r8ix)rrrHcPtjgdt||dz}tjt 5t |ddj}dddjdj}|d|dzk(sJ|d |d zk(sJy#1swY@xYw) N) iaNi>^iLiIi1iIiiiirrrF)rnrrrr:) r/r2r.rwarnsrr rur}r)rts_1modelrs r6test_forecast_indexr[s 99NBR  D ( )$ E #%  NNN $ * *E 8rBw   9R   s BB%ctjtd5tt j dddddy#1swYyxYw)NzCan only dampen thematchrT)rrrrr r)rrr8r6test_error_dampenrks8 z)> ?>RWWS\=>>>s !AActjjd}tjt d5t |dddddt |dzd }t|t sJt |dzdd }tjtd 5|jd dddy#1swYsxYw#1swYyxYw) Nrzuse_boxcox must be TruerarrrprTrrzuse_boxcox was setF) r)r*r,rrrr rrru)r'rs r6test_error_boxcoxrps !!#&A y(A BOQ3gNO q!t 5C c/ 00 0  15G C z)= >" 5!""OO""sB< C<CCc tjtd5t|ddddtjtd5t|ddddddtjtd5t|ddgd  dddtjt5t|d dddtjt5t|d dddtjt5t|gd dddtjt5t|ddddddtjt5t|ddddddddt|dd}tjt5|j d dddtjt5|j dddddy#1swYxYw#1swYxYw#1swYxYw#1swYmxYw#1swYKxYw#1swY'xYw#1swYxYw#1swYxYw#1swYxYw#1swYyxYw)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 A S@A z)F G  ")    z)I J  ")1    z "5S45 z "5S45 z "HS3FGH z "  ")    z "  ")     7# C z "# c"# z "? o>??[AA    5555HH    ##??swH HH3H,"H9II3I 4I,(I8HHH),H69III I),I58Jr)rrzL-BFGS-BTNCrPowellz trust-constrc@tjddg}i}t|d}|j|||}t |j ddd t |j d d d t |jjtsJy) Ngp= ף?g&@rkr)rrer rgʳ?rrtolrgnt&@) r)rr rurrrrrr)rr7svr rr s r6test_alternative_minimizersrs 4- BO s+ FC ''B  CCJJ01:DICJJ/4H ckkm++-s 33 3r8c|t|d}ddi}tjt5|j |dddtjt5|j d|ddddddii}tjt5|j d |ddddd d ii}|j d |}t |j tsJt |jjtsJy#1swYxYw#1swYxYw#1swYxYw) Nrkrargsanything)r rr rrrr) r rrrrurrr rrr)r7rkwargsr s r6test_minimizer_kwargs_errorrs s+ FCj !F z "( '( z "@ ?@ 6:"6 7F z "? ~v>? 8W"5 6F ''' @C cjj$ '' ' ckkm++-s 33 3((@@??s#DD&D2D#&D/2D;r)rr?)rrr")rrrc tj|}tjdgdz}tjddggdz}tt|||t ||j tj |yNTrsrr)r)rrrrr1r2rselr:s r6test_to_restricted_equivrslXXf F ((D6A: C XXSzlQ& 'Ffc6*FCJJrxx$8&Ar8)rr皙?)rr"r)rrrc tj|}tjdgdz}tjddggdz}t|tt |||||yr)r)rrrrrs r6test_restricted_round_tiprs_XXf F ((D6A: C XXSzlQ& 'F fc6:CHr8c~ddi}tjt5t||ddddddgi}tjt5t||dddddi}tjt 5t||dddddi}tjt 5t||ddddd i}tjt 5t||dddy#1swYxYw#1swYxYw#1swYxYw#1swYaxYw#1swYyxYw) Nbad_key)rr)r:rrr)rrr)rr)gr)rrKeyErrorr rrr7r:s r6test_bad_boundsrs $F x 1S01#s ,F y !1S01 1F z "1S01 ,F z "1S01 -F z "1S0111111111111s;DDD9D',D3D DD$'D03D<cddi}t||djd}t|djd}t|jd|jdd y) Nr)rrrk)r:rprrrrr)r rurr)r7r:r r s r6test_valid_boundsrsu ,F  F+  cc! ; G K K L D $% %& r8cbt|d}|jddi5|j}dddjddk(sJt |j j tsJt|ddd}|jdd i5|j}ddd|jdd k(sJt |j j tsJt|ddd }|jd d d 5|j}ddd|jdd k(sJ|jdd k(sJt |j j tsJy#1swY\xYw#1swYxYw#1swYyxYw)NrkrrrrTrNrrrrr")r+rr+)r rrurrrrrr7rr s r6test_fixed_basicrs s+ FC *C0 1ggi ::' (C // / ckkm++-s 33 3  5t; C $/ 0ggi ::o &$ .. . ckkm++-s 33 3  55  C ssK Lggi ::* +s 22 2 ::' (C // / ckkm++-s 33 3)s#F FF% FF"%F.ct|d}tjt5|j ddi5 ddddddtjt 5|j ddi5 ddddddt|dd}tjt 5|j dd d 5 ddddddt|ddd }tjt 5|j dd d 5 ddddddddd }t|dd|d}tjt d5|j ddd5|j ddddddy#1swYUxYw#1swYZxYw#1swY0xYw#1swY5xYw#1swYxYw#1swYxYw#1swYxYw#1swYxYw#1swYvxYw#1swYyxYw)Nrkrrrrg333333ӿrrrrrrrr+)rr)ffffff?r)rnror:rpz!After adjusting for user-providedrr)rr+)r rrrrrru)r7rr:s r6test_fixed_errorsr-s s+ FC x  ^^.4 5    z " ^^.5 6     5  C z " ^^L M     55  C z " ^^ #3 ?     ",: NF  )  C z)L M ^^ #3 ?   GGI 9          sFFF&F7:F*<F73GG GG(GG(H.G4?HF FF'*F4 /F77GG GGG% !G((G14G= 9HH rcvt|||d}|jd}|jjsJ|j ddi5|jd}ddd|jjsJt |j jtsJy#1swYPxYw)NrkrT)rrr) r ru mle_retvalssuccessrrrrr)r7rnrorr s r6 test_bruterSs  58; C ''D' !C ?? " "" " *C0 1&ggg%& ?? " "" " ckkm++-s 33 3&&s B//B8ctjt5t|dddddtjt5t|ddddddtjt5t|dt j dd dddy#1swYxYw#1swYZxYw#1swYyxYw) Nr heuristicrrr)rnrrprrk)rorrpr)r7s r6test_fix_set_parametersrbs z "  s+   z "  "4    z "  WWR["-          s#B' B3;#B?'B03B<?Cct|d}tjtd5|j ddi5|j dddddddy#1swYxYw#1swYyxYw) NrkrzCannot fix a parameterrrg?r"r)r rrrrrurs r6test_fix_unfixablerwsl s+ FC z)A B) ^^.5 6 ) GGCG ( ))) ) )))s"A-A!A-!A* &A--A6cLddd}tjtd5t|d|dj dddd d d }tjtd 5t|d|d j dddy#1swYPxYw#1swYyxYw)N)rr")rrrzThe bounds for smoothing_trendrrrk)rnr:rp)rr)rrrz!The bounds for smoothing_seasonal)ror:rprrs r6test_infeasible_boundsr~s!+ KF z)I J uV; #%",: NF z)L M "-  #%   sB'BBB#)rkrrct||||}|j}|jjsJt |j j tsJy)Nr)r rurrrrrr)r7rrnrorr s r6test_initialization_methodsrsV  586 C '')C ?? " "" " ckkm++-s 33 3r8ct|dj}|jdkDsJ|jjd|jdk(sJt |j |jddy)Nrkrrr:)r rurresidr1r fcastvaluesr)r7r s r6test_attributesrsc s+ F J J LC 55199 99??1 1 -- -COOS__RS%9:r8ct|dzdd}tjtd5|j dddd|j }t |j }tjd|sJt|j jt sJy#1swYyxYw) NrTrrzuse_boxcox was set at modelrrzBox-Cox:[\s]*True) r rrrrurrrefindallrr)r7rr summs r6test_summary_boxcoxrs  Q4{ C z)F G! 4 ! '')C s{{} D ::*D 11 1 ckkm++-s 33 3 !!s B88Ccttj|d}|j}t |j j tsJtjtd5|jdddddtjtd5|jdd tjd  ddd|jdd d tjtd5|jdd d dddtjtd5|jdd d ddddtjtd5|jdd d dddy#1swYxYw#1swYxYw#1swYxYw#1swYgxYw#1swYyxYw)Nrrz error must berrunknown)rz If randomr)rVrV)rrr)rrz Cannot anchorrzArgument random_state bad_value)rrrzArgument random_errors bad_values) r r)r.rurrrrrrrremptyrs r6 test_simulaters  3{ C '')C ckkm++-s 33 3 z 9* Ry )* z 5M Rz(9K LMLL:cL2 z 98 Rz$ 78 z)@ A  j;    z)A BG Rz FGG**MM88  GGs<:F1(F,F8GGF),F58GG G index_typ)r0periodr. irregularc |jd}i}d}d}|dk(rtjd|dzd}n|dk(r tjd|dzt}n|dk(rtj |dz}d |d <n|d k(rt jjd}tjt j|jdd |dz }t}d |d <tj ddd}|dd}|j}dd|_t|5t!|fdddd|j#}dddt|5j%d} dddt' tj(sJtj*j-| j|y#1swYuxYw#1swY\xYw)Nrr r$$rr%r0r.rrmr rj)sizerHi r!)rOstopstepirrr)r1r/ period_ranger0r RangeIndexr)r*r+IndexrIrandintrr-rrr rur}rr2testingassert_index_equal) r7rrrwarning fcast_indexrr3r fcasts r6test_forecast_index_typesrs 99QV ## # MM,AGGAJ ?E II  UA  FE 5MEDJ djj""2"2 33 3 ::?? "" " :: # # ++ + l*J K $ "-   YYY 6F l. 9&f%&&  &&s)G 0G GG!cgd}t|dd}|j}|jt|jddy)N)rrrgY@rrrrg4@rr)r rurr)serieseses_fits r6test_initial_levelr5Hs? 2F gT BVVXF MMFMM/2D9r8c ttddddj}tj|j d}tj|j dd}tj|j d d}ttdddd ||| j}t |j d|t |j d|t |j d |y) Nrjrrrkrlrrr!r,r)rmrnrorprrr)r rdrur)roundrr)rxlvlrnseasrys r6test_all_initial_valuesr:Ss  )    ce   ((4;;/ 0C HHT[[11 5E 88DKK 12A 6D  %  ce DKK0#6DKK0%8DKK 12D9r8)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._smoothersr3rrpathsplitabspath__file__baserEread_csvjoinrasfreq SEASONALSTRENDSrgr to_datetimeidxr2r_r`rdfixturer7rJrLr parametrizerrrrrrrrr r&rr(rArFrLrTrirmrrrrrrrrrrrrrrrrrrrrrrrrrrr!r$r&r0r5r:rr8r6rUs02 > L ''--1 2ar{{GGLLy"45 ""4( 4 4bnnUryy-R--c  c8JKLh/ /"4B  B  JD  Y/ 0 ^4G5G &&)Y/)0*)"!. 7 5&)Y/30*3$&)Y/)$0*)$X&4R 0.&@J"J"Z2+2+jg+g+T &)Y/=1.161220*61r'T/*45  6  > "/?d   4  4 4  A A1$ 42#L5$-0eT]3 441 4 *)   <4-0dE]3441 4; 4G*?"<"'?1'.4-0dE]3!R) #*41 ##&L::r8