i p dZddlmZddlmZddlZddlZddlZddlZ ddl m Z m Z ddl ZddlZddlZddlmZddlmZmZmZmZmZddlmZmZdd lmZmZm Z ejBjEejBjGe$\Z%Z&ejNejBjQe%d d d d Z)e)jUdZ)dZ+dZ,gdZ-gdZ.ej^e.Z0ejbe-ejde0ejfe0Z4ejjddZ6dkdZ7GddZ8ejrjudddZ;ejrjude+dZd#Z?ejrjud$d%gejrjudd%gd&Z@d'ZAd(ZBd)ZCd*ZDejrjejrjud$e,ejrjude+d+ZFejrjud$e,ejrjude+d,ZGd-ZHd.ZId/ZJd0ZKejjd1ZLejjd2ZMejjd3ZNejrjud$e,ejrjude+ejrjud4d5ejrjud6d d7ZOd8ZPd9ZQejrjud:gd;d<ZRd=ZSd>ZTd?ZUejrjud@gdAdBZVdCZWejrjudDgdEgdFgdGgdHZXejrjudDgdIgdJgdKgdLZYdMZZdNZ[dOZ\dPZ]ejrjud$dQdgejrjuddQdgdRZ^dSZ_dTZ`dUZaejrjud@gdVejrjud$ddQgejrjudddQgdWZbdXZcdYZddZZeejrjud[gd\d]Zfd^Zgejrjud_d`dagejrjudbddcgddZhejrjud$ddQgejrjudddQgejrjudedfdagdgZidhZjdiZkdjZly)lz4 Author: Terence L van Zyl Modified: Kevin Sheppard ) MONTH_END) pytest_warnsN)assert_allcloseassert_almost_equal) 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 x/home/ubuntu/crypto_trading_bot/.venv/lib/python3.12/site-packages/statsmodels/tsa/holtwinters/tests/test_holtwinters.pysesr6is   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)r0r(zerosr-arange) xalphabetal0b0 nforecastnlvalsbxhatfterrs r5_simple_dbl_exp_smootherrIus  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 !!r7ceZdZedZdZdZejjdddZ 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:00rNrOrPrQrRrSrTrUrV)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:00rXrYrZr[r\r]rNrOrPrQrR) g^>g{%jCg&HI@gV{6?gҏB?g*| W?gfv?g?) r.r1 DatetimeIndexr infer_freq oildata_oil air_ausairlivestock2_livestockaust start_params)clsdatarr`rarbs r5 setup_classzTestHoltWinters.setup_classs    iie, ,,   BMM+2C2C$D & " "YYtU+ ++   2==1A1A#B $  B  B "yyu5%'%5%5 & &399:& "$8   r7cPt|jddddj|j}t|jddddj|j}t |j ddgd d t |j d gd d y) Nrr estimatedseasonal_periodstrendseasonalinitialization_methodrdz2011-03-01 00:00:00z2011-12-01 00:00:00g_vN@g9vB@g q{G@gڊeI@)end)r rcfitrdrpredict)selffit1fit2s r5 test_predictzTestHoltWinters.test_predict:s# II"-  #4,,# - $ II"-  #4,,# -   LL.0E F 0   LL2L 3 0  r7ct|jjddddj|j}t |j dgddy) Nrirrrjrkrprqrr)r rcvaluesrtrdrforecastrvrws r5 test_ndarrayzTestHoltWinters.test_ndarray[sU# II  "-  #4,,# -  DMM!,.RTUVr7zOptimizer does not convergeF)reasonstrictct|jdddjdd}t|j dgdd y) Nrir)rlrmrnbhT)method use_brutesteps)g9#zN@g/lB@gaG@gHI@rrr rcrtrr|r}s r5 test_forecastzTestHoltWinters.test_forecastfsL# II  #TT# *  MMM "$H! r7c6t|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-heuristicror!F optimized333333?rjr gbL~@ri) rMgO~|@g>V|@gAV{@g=0{@g{@g9D{@g2\=$|@gV|@gsY}@g@{`4'~@gsL~@gj=f]@gBX@smoothing_levelgc?? initial_levelg{@rr)r r`rtrr|levelparamsrvrwrxfit3s r5test_simple_exp_smoothingz)TestHoltWinters.test_simple_exp_smoothingrs!   4F #cU# # "   4F #cU# # "   K #%  DMM!,zlA> JJ  ! $ DMM!,zlA>DMM!,zlA>DKK(9:HaHDKK8*aHr7ct|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) exponentialrorj damped_trendrorr)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?ri)rWgoDI6@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 rartrr|rm fittedfcastrs r5 test_holtzTestHoltWinters.test_holtsDOO;MNRRS  OO"4  #c3%# H  OO"-  #c3# 7 DMM!,.QSTU JJ " ' *     $ ) , DMM!,.QSTUDMM!,.QSTUr7ct|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)NrjrTr\(\? least_squares) damping_trendr)rrror?rrrrg(\p@r initial_trendgYi@g(\p@g(\@g(\@g ףp= ?g33333/p@gp= ף?g·@r) r rbrtr rrr(nansse isinstancesummaryas_textstr)rvrwmod4fit4mod5fit5s r5test_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$,,.002C888r7c 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)NTrjrgl9+?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>ı@gZ F* r7rKtrend_seasonal))rN)Nr)rrc|\}}tjd }tjt5t |||ddddy#1swYyxYw)Ndr)rmrnrl)r(onesrraises ValueErrorr )rrmrnr&s r5test_negative_multipliativer sP%OE8  A z "UQehQSTUUUs AArnctjt5ttd|dddddy#1swYyxYw)NFTr)rmrnrrl)rr TypeErrorr housing_data)rns r5test_dampen_no_trendrs: y !       s 8A)rrctjtjd tjddd}t j t5t||ddddy#1swYyxYw)Nrr#rr$rr )rnrl) r.r1r(rr/rrrr )rnr&s r5test_invalid_seasonalr s_  R]]:sN A z "GQAFGGGs A33A<ctjt5tt j t t gdjdddy#1swYyxYw)Nr axis)rrrr r.concatrrtrr7r5 test_2d_datar)sH z "TRYY l'C!LMQQSTTTs 9AA%c<tj}t|j|_t j d5}t |ddd}t|dk(sJdt|dvsJ dddjd k(sJy#1swYxYw) NT)recordrrjrmrnror rr ) rr,listrwarningscatch_warningsr lenrrl)hd2wrs r5test_infer_freqr.s    CSYYCI   -+" uuK 1v{{QqT*** +   2 %% % ++s 2BBrmrcHtt||d}|j}|jddddii}t|j j t sJ|jd|jzksJt|jtsJy) Nrjr basinhoppingminimizer_kwargsrL-BFGS-Brminimize_kwargsg)\(?) r rrtrrrrrrdict)rmrnrresres2s r5test_start_paramsr :s )  C '')C 77+h -CD  D ckkm++-s 33 3 88dSWWn $$ $ dkk4 (( (r7cntttjdd}|jdy)Nrknownrro?r)r rrrtrs r5test_no_params_to_optimizerMs0 "''*% C GGCG r7cttd}tjt5|j t jdgdddy#1swYyxYw)Nrjrrrp)r rrrrrtr(arrayrs r5test_invalid_start_param_lengthrVsG |; OC z ". RXXse_-...s 'AA$cttdd}|j}|jd}t|j j t sJt|j j t sJd}|j|j|zksJ|jd}t|j|jdy) Nrrjrmrorrgh㈵>r)decimal) r rrtrrrrrr)reset_randomstaterr r tolres3s r5test_basin_hoppingr\s E C '')C 77.7 )D ckkm++-s 33 3 dlln,,. 44 4 C 88sww} $$ $ 77.7 )D$((A6r7cttdd}|j}|jd}tj|j |j k7sJtj dddf|j z }t|jdt|jjtsJt|jjtsJy)NrrjrT) remove_biasrr) r rrtr(anyrrrmeanrrrr)rr r err2s r5 test_debiasedr%ks E C '')C 77t7 $D 66#""d&7&77 88 8   QT "T%6%6 6D S) ckkm++-s 33 3 dlln,,. 44 4r7ctt||ddj}t|jddtt||dj}t|jddy)Nrjr)rmrnrorr)rmrnr)r rrtrr)rmrnr s r5test_float_boxcoxr'xsu )    ce CJJ|,c2     ce  CJJ|,c2r7c 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)Nrjrrrrrsmoothing_seasonalrrrrinitial_seasonsrrrF)r rrtrrrrrr0r(squeezeasarrayr:isnan ones_likeastypeint64r,rrr PyHoltWintersArgsr r)rmrnrr rrr&mpr=r>gammaphir?r@xip_copyboundspy_funccy_func py_hw_args cy_hw_argssse_cysse_pys r5test_equivalence_cython_pythonr@s )  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#r7c 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)Nrjrrrrr)r>r?r@rAr9rrrr)r rrrrir)r rrtrrrrr(r,r-endogrrIrrrrr|r rrmr) rr r<r=lrDrF_rEr>s r5test_direct_holt_addrEs \ 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 3r7cDtjjd}d|jdz}tj|dddf}||dddfz}|j t }t|ddj}|jd d k7sJy) Ni90r)rrr rrjrrr) r(r)r*r+cumsumr0intr rtr)rr2r3y_starr&r s r5test_integer_arrayrKs   u %B R   **A YYqAw F1a4A  A q[ Q U U WC ::' (C // /r7ctjd}t|ddd}|jddd }|j d }t |tj dtjd fd |jddd }|j d }t |tj dtjd fd t ||d y)NrrTrjrmrror rg#B ;)rrrr)start g|=r)r(r;r rtrurr)rBrres1pred1r pred2s r5test_damping_trend_zerorSs IIbME  )  C 771c7 OD LLqL !EE255biil!23%@ 771c7 KD LLqL !EE255biil!23%@E5u-r7cgd}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#)rNr%)rrir r)rmr<r r)rNrs)r[FPz2000-1-6z2000-1-8@)r_)r.r/rr1r rtrru) array_input_addseries_index_addseries_input_addarray_input_mulseries_index_mulseries_input_mulrwrxrrs r5test_different_inputsrhsj*O}}:s??STyy2BC'O}}:s??STyy2BC u = A A CD  0 > B B DD u = A A CD  0 > B B DD a0 1! 4lAF a0 :: F VWX a0 1! 4naH a0 :: 6r7cfgd}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>r5r6rCrDs1s2s3s4g`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(?rngCaj?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 @gEr5r6F)rrr*rrrCr9rDrrrr)g@9?gY,?gOQ?g[iY@r ) repetitionsr random_errorsr)r(r-r rt_level_trend_seasonr"r.rTsimulaterr{)rmrnr~rrlrr short_name model_nameexpectedstatertinnovsims r5test_simulate_expected_rrs{(S4JE"Z%66H9MMJ088::9&A*MN$V,Z8     cgf  >El   3ZCJJrN3ZCJJrNDkCKKODkCKKODkCKKODkCKKO vvbhhs''() JJGH I K KE ,,qauE, JC#**a0r7ct|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. rirTrjrlrmrnrror)anchor random_staterNr9rjrsz 2016-03-01r)rr bootstraprrN) r rtrrr{scipystatsnormr(r)randnr*all)rlrtr r s r5test_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#+  r7c 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 rirrFrjT)rlrmrnrrorrrrr rrN) r rtr|r{rr(r#rabs)rlrtrr r#s r5test_simulate_boxcoxr1s )  ce||A%%H ,,qbq, 9 @ @C 773Q D 66"&&)A- .. .r7ix)rrrGctjgdt||dz}t|ddj }|j dj }|d|dzk(sJ|d |d zk(sJy) N) iaNi>^iLiIi1iIiiiirrrF)rmrrrr9)r.r1r-r rtr|r)rts_1modelrs r5test_forecast_indexrFs~ 99NBR  D !U G K K ME NNN $ * *E 8rBw   9R  r7ctjtd5tt j dddddy#1swYyxYw)NzCan only dampen thematchrT)rrrrr r(rrr7r5test_error_dampenrSs8 z)> ?>RWWS\=>>>s !AActjjd}tjt d5t |dddddt |dzd }t|t sJt |dzdd }tjtd 5|jd dddy#1swYsxYw#1swYyxYw) Nrzuse_boxcox must be TruerarrrorTrrzuse_boxcox was setF) r(r)r+rrrr rrrt)r&rs r5test_error_boxcoxrXs !!#&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)rorrzinitial_seasonal set but model)r!333333?皙?r)rorinitial_seasonal)r)r)rr!g?ri)rr)rmror@)rmrnrorrrorTr)use_basinhoppingrrrrr rtr6rs r5test_error_initializationrgs z)V WAS@A z)F G  ")    z)I J  ")1    z "5S45 z "5S45 z "HS3FGH z "  ")    z "  ")    s'QT UC z "# c"# z "? o>??SAA    5555HH    ##??swH HH3H,"H9II3I 4I,(I8HHH),H69III I),I58Jr)rrrTNCSLSQPPowellz trust-constrc@tjddg}i}t|d}|j|||}t |j ddd t |j d d d t |jjtsJy) Ngp= ף?g&@rjr)rrdr rgʳ?rrtolrgnt&@) r(rr rtrrrrrr)rr6svr rr s r5test_alternative_minimizersrs 4- BO s+ FC ''b/' RCCJJ01:DICJJ/4H ckkm++-s 33 3r7c|t|d}ddi}tjt5|j |dddtjt5|j d|ddddddii}tjt5|j d |ddddd d ii}|j d |}t |j tsJt |jjtsJy#1swYxYw#1swYxYw#1swYxYw) Nrjrargsanything)r rrrrrr) r rrrrtrrr rrr)r6rkwargsr s r5test_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 |yNTrrrr)r(rrrrr0r1rselr9s r5test_to_restricted_equivrsj XXf F ((D6A: C XXSzlQ& 'Ffc6*FCJJrxx$8&Ar7)rr皙?)rr!r)rrrc tj|}tjdgdz}tjddggdz}t|tt |||||yr)r(rrrrrs r5test_restricted_round_tiprs] XXf F ((D6A: C XXSzlQ& 'F fc6:CHr7c~ddi}tjt5t||ddddddgi}tjt5t||dddddi}tjt 5t||dddddi}tjt 5t||ddddd i}tjt 5t||dddy#1swYxYw#1swYxYw#1swYxYw#1swYaxYw#1swYyxYw) Nbad_key)rr)r9rrr)rrr)rr)gr)rrKeyErrorr rrr6r9s r5test_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)rrrj)r9rorrrrr)r rtrr)r6r9r r s r5test_valid_boundsrsu ,F  F+  cc! ; G K K L D $% %& r7cbt|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)NrjrrrrTrMrrrrr!)r*rr*)r rrtrrrrrr6rr s r5test_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)Nrjrrrrg333333ӿrrrrrrrr*)rr)ffffff?r)rmrnr9roz!After adjusting for user-providedrr)rr*)r rrrrrrt)r6rr9s r5test_fixed_errorsr s s+ FC x  ^^.4 5    z " ^^.5 6    s%{ SC z " ^^L M     55  C z " ^^3O P   ",: NF  )  C z)L M ^^3O P  GGI 1           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)NrjrT)rrr) r rt mle_retvalssuccessrrrrr)r6rmrnrr s r5 test_bruter+s  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)rmrrorrj)rnrror)r6s r5test_fix_set_parametersr:s z "XS;WX z "  "4    z "  WWR["-    XX    s#B' B3;#B?'B03B<?Cct|d}tjtd5|j ddi5|j dddddddy#1swYxYw#1swYyxYw) NrjrzCannot fix a parameterrrg?r!r)r rrrrrtrs r5test_fix_unfixablerMsl 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_trendrrrj)rmr9ro)rr)rrrz!The bounds for smoothing_seasonal)rnr9rorrs r5test_infeasible_boundsrTs!+ KF z)I J uV; #%",: NF z)L M "-  #%   sB'BBB#)rjrrct||||}|j}|jjsJt |j j tsJy)Nr)r rtrrrrrr)r6rrmrnrr s r5test_initialization_methodsrdsV  586 C '')C ?? " "" " ckkm++-s 33 3r7ct|dj}|jdkDsJ|jjd|jdk(sJt |j |jddy)Nrjrrr9)r rtrresidr0r fcastvaluesr)r6r s r5test_attributesrpsc s+ F J J LC 55199 99??1 1 -- -COOS__RS%9:r7ct|dzdd}tjtd5|j dddd|j }t |j }tjd|sJt|j jt sJy#1swYyxYw) NrTrrzuse_boxcox was set at modelrrzBox-Cox:[\s]*True) r rrrrtrrrefindallrr)r6rr summs r5test_summary_boxcoxrws  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)rUrU)rrr)rrz Cannot anchorrzArgument random_state bad_value)rrrzArgument random_errors bad_values) r r(r-rtrrrrrrrremptyrs r5 test_simulatersx rzz#k RC '')C ckkm++-s 33 3 z 9* Ry )* z 5M Rz(9K LMLL:cL2 z 98 Rz$ 78 z)@ AQ Rz#K PQ z)A BG Rz FGG**MM88QQGGs<:F1(F,F8GGF),F58GG G index_typ)r/periodr- 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)Nrrr#$rr$r/r-rrlr ri)sizerGi r )rNstopstepirrr)r0r. period_ranger/r RangeIndexr(r)r*IndexrHrandintrr,rrr rtr|rr1testingassert_index_equal) r6rrrwarning fcast_indexrr2r fcasts r5test_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 rtrr)serieseses_fits r5test_initial_levelr5s; 2F fGSW XB VVXF MMFMM/2D9r7c ttddddj}tj|j d}tj|j dd}tj|j d d}ttdddd ||| j}t |j d|t |j d|t |j d |y) Nrirrrjrkrrr r+r)rlrmrnrorrr)r rcrtr(roundrr)rwlvlrmseasrxs r5test_all_initial_valuesr:s  )    ce   ((4;;/ 0C HHT[[11 5E 88DKK 12A 6D  %  ce DKK0#6DKK0%8DKK 12D9r7)r)m__doc__statsmodels.compat.pandasrstatsmodels.compat.pytestrosrrnumpyr( numpy.testingrrpandasr.r scipy.statsrstatsmodels.tools.sm_exceptionsrstatsmodels.tsa.holtwintersrr r r r 2statsmodels.tsa.holtwinters._exponential_smoothersr r&statsmodels.tsa.holtwinters._smoothersr2rrpathsplitabspath__file__baserDread_csvjoinrasfreq SEASONALSTRENDSrfr to_datetimeidxr1r^r_rcfixturer6rIrKr parametrizerrrrrr rrrr%rr'r@rErKrSrhrlrrrrrrrrrrrrrrrrrrrrrrrrrrr r#r%r0r5r:rr7r5rUs 02 > 8 ''--1 2ar{{GGLLy"45 ""4( 4 4bnnUryy-R--c  c8JKLh/ /"4x  x  vDUUY/ 0 ^4G5GT &5'*eW-).+)"!. 7 5&)Y/30*3$&)Y/)$0*)$X&4R0.&4J"J"Z2+2+jg+g+T &)Y/=1.161220*61r'T/*45  6  > "+?\  4 4 4 O_o#VWXO_o#VWX1$ 42@5$-0eT]3 441 4 &)  #QR4-0dE]3441S4; 4G"&TU"<V"'?1'.4-0dE]3!R) #*41 ##&L::r7