g-&<ddlZddlmZmZddlZddlmZddlm Z ddl m Z m Z m Z mZ ddlmZe j&Zej*ej,e_ej,dddfj/eZej,ddddfZej4j7edej4j8d Zej4j<ej4j8d Zej4j7edej4j<ej4j8d Z ej4j7edej4j<ej4j8d Z!ej4j7edej4j<ej4j8d Z"ej4j7edej4j<ej4j8dZ#ej4j7edej4j8dZ$ej4j7edej4j<ej4j8dZ%dZ&dZ'ej4j8dZ(y#e$rYwxYw)N)assert_almost_equal assert_equal)PYTHON_IMPL_WASM)elnino) banddepthfboxplot hdrboxplot rainbowplotz0Multiprocessing is not supported in WASM/Pyodide)reasonc  tttd\}}t|jdk(sJgd}t |j |dtj|j|j|jg}tjgdgdgd gd gd gd gd gdg}t ||dtjtjt|jjtjd}t|}t!gd|t!t|j"|y#t$$rt'j(dYywxYw)N90)labelsseedr gZd;?8@g9@gX99@g9v8@gS7@gZd;OM6@gB`";5@gV-]4@g^I +4@gvo4@g#~j5@gCL6@decimal) g\(\8@gQk9@ffffff9@g(\8@g(\58@gY7@g6@gp= 5@g ףp= 6@gzG6@gR8@gp= 9@) g@;@g{G:<@g<@R<@gQ^<@gGzn;@g{G9@gzG7@(\B6@gQ86@g(\56@gq= ף07@) g333337@gGz:@gQ+;@g= ףp:@gQ:@gfffff&:@gףp= 9@g333338@gq= ף8@gp= ף8@g9@gGz;@) gQ<@rg= ףp==@g33333s<@g\(\;@gq= ף09@g\(7@gQE6@(\O5@gQ^5@g5@g(\6@) g{Gz9@g= ףp:@g(\;@g ףp= ;@{G::@gq= ף8@gq= ף7@gQ6@rp= ף6@gGzT7@gq= ףp8@) g\(7@g8@g)\8@gq= ףp7@g= ףp5@gGz4@gGz3@gףp= 2@g(\2@g= ףp2@g)\3@g\(4@) gףp= 8@rgQ:@gGz9@gQ8@gR7@g(\u6@g33333s5@g(\B5@gR5@gp= #6@g)\h7@) gQ7@皙9@gףp= W9@gQ88@g= ףp6@gQ5@gffffff4@g(\3@gzG3@gQ3@gQ^4@g(\5@r axis)0Multiprocess randomly crashes in Windows testing)r datarlenextra_quantilesrmediannpvstackoutliershdr_90hdr_50allisinreshapeshaper outliers_idxOSErrorpytestxfail) close_figures_hdrmedian_tquantquant_t labels_posr)s ]/var/www/html/venv/lib/python3.12/site-packages/statsmodels/graphics/tests/test_functional.pytest_hdr_basicr<s: $IDe<33&&'1,,,D CJJ!< 3<<SZZ@A))HHHHHHHHIJ" E7A6VVBGGD#,,7?? K!"$ *%-x8VC,,-x8 I GHIsE E E.-E.c ttdtd\}}t|jdk(sJgd}t |j |dy#t$rtjdYywxYw)NrT)ncompr use_bruterrrr") r r#rr$r%rr&r1r2r3)r4reset_randomstater5r6r7s r;test_hdr_basic_bruterAGsmID&DI33&&'1,,,D CJJ!< I GHIAAA43A4ctj}|jd} ttt j |dd|jd|jd|jtjdd dz |jgd |jd d gy#t$rtj d YywxYw)Nor r)rax thresholdrzMonth of the yearzSea surface temperature (C) )step)MarJunSepDecgɿgffffff&@r")pltfigure add_subplotr r#rtolist set_xlabel set_ylabel set_xticksr'arangeset_xticklabelsset_xlimr1r2r3)r4figrEs r; test_hdr_plotrZUs **,C  B I4 B!  )* 34 bii+a/0 ;< T4L! I GHIsBB??C C c ttdgd\}}tjgdgdg}t |j |dy#t $rtjdYywxYw) Ngffffff?ralphar) rg:@g;@gffffff:@rg8@g7@g6@g333335@g6@g333336@g7@) gffffff7@g8@g9@gfffff7@ffffff6@g5@g4@gL3@g3333333@gffffff3@g4@gL5@rrr") r r#r'r(rr%r1r2r3r4r5r6 extra_quant_ts r;test_hdr_alpharalso IDE:3 $H$H#IJ  C//J I GHIsA A A-,A-c ttddgd\}}gdgdgdgdg}t|jt j |d y#t $rtjd YywxYw) Ng?gq= ףp?rr\) gʡE9@gZd;O ;@gt;@g333333;@gx):@gh|?8@g$7@gT㥛`6@g@6@rgFx7@g/$8@) g +6@g-阮8@gS㥛8@g+W7@g r5@gHz4@g3@g 3@gZd;2@gw/2@gQ3@g(\4@) gn8@g r:@g"~j:@g/$9@g+N8@gMbX6@gv5@g 4@gS{4@g/5@gK7A5@gp= 6@) gsh|7@g"^9@gn9@g/ݤ8@gZd;O-7@g?5^I5@g"~4@gp= 4@gOn3@g~jt3@gZd;4@gS5@rrr") r r#rr%r'r(r1r2r3r_s r;test_hdr_multiple_alpharcswIDd %@3:::: ;  C//=1I$% ' I GHIrBcV ttdgdd\}}tjtjt|j j tjd}t|}tgd|y#t$rtjdYywxYw) Ng?g(\?r)r]rFrr r)irrr r!r") r r#r'r,r-r)r.r/rrr1r2r3)r4r5r6r:r)s r;test_hdr_thresholdresID!&(3VVBGGD#,,7?? K!"$ *%3X> I GHIsBBB('B(c ttdd\}}gd}t|j|dy#t$rt j dYywxYw)Ncv_mlr)bwr) g@8@gp= ף9@g= ףp9@g ףp= 9@g(\µ7@zGa6@rgq= ףp4@g= ףp=4@g(\4@gq= ף05@rirrr"r r#rr&r1r2r3r4r5r6r7s r; test_hdr_bwrlsQ IDW593> CJJ!< I GHI14AAc ttdd\}}gd}t|j|dy#t$rt j dYywxYw)NrHr)r>r) gGzT8@g(\µ9@g ףp= :@gGz9@g= ףp7@r^gRQ5@g33333s4@g@4@gHz4@g3333335@gp= c6@rrr"rjrks r;test_hdr_ncomprosOID63>CJJ!< I GHIrmctjddz }ddtj|zz}dtj|tjdz zz}dtj|tjdz zz}ddtj|tjdz zzz}tj ||||g}t |d }gd }t||y) Nigb@r ?g333333?gBD2method)rq竪?rwrq)r'rVsinpicosasarrayrrxxy1y2y3y4r#depthexpected_depths r;test_banddepth_BD2rs 3$ B S266":  B rvvb2557l# #B rBEE!G|$ $B cBFF2a<(( (B ::r2r2& 'D d5 )E+N~.cztjddz }tj|j}d|zdz }tj|jdz}tj|jdz}tj ||||g}t |d}gd }t||d y) Nig@rr rqgпMBDru)rw?g?rr)r'rVzerosr/onesr{rrr|s r;test_banddepth_MBDrs 45 B "(( B R!B  S B  U "B ::r2r2& 'D d5 )EEN~q9rcld}tjjdtjddtjzd}t dDcgc] }|| }}t j}|jd}t|d|\}}}} tjgd } t|| tjgd } t| | tj|dj} t|| |t jj }ycc}w) NcFttjjdkD}tjjdz}tjjdz}dtjjzdz}dtjjzdz}d|z |tj|z|tj|zzz||tj|z|tj|zzzz}|S)z3Test function, combination of a few harmonic terms.g?g?g?g?r )intr'randomrxrz)tcia1ia2ib1ib2ifuncs r;harmfuncz+test_fboxplot_rainbowplot..harmfuncs!!#c) *ii 4'ii 4'RYY--//#5RYY--//#5B3?S266!9_<= #q /C"&&)O3 45 rirrrD)wfactorrE)rGrrrrH r r   r)rrrr)xdatarcmap)r'rrlinspaceryrangerOrPrQrarrayrrVsizer cmrainbow) r4rrr5r#rYrErix_depth ix_outliers ix_expected ix_expected2rs r;test_fboxplot_rainbowplotrs IINN7 Aq255y#&A!&r +AHQK +D + **,C  B&.tQ2&F#Auh ((78K;'88O,Ll+ IId1gll #E d%u366>> JC ,sD1))numpyr' numpy.testingrrr2statsmodels.compat.pythonrstatsmodels.datasetsrstatsmodels.graphics.functionalrrr r matplotlib.pyplotpyplotrO ImportErrorloadr#r{raw_dataastyperrmarkskipif matplotlibr<slowrArZrarcrerlrorrrrr;rs$; 6' # v{{} 4==) q!t  # #C ( }}QU =%I  %IP I I =I  I" = I  I =I  I, = I  I = I  I =I  I /, :"K"KE  sLLL