5[gWPddlmZddlmZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZddlmZgdgd fZeej<e d ggz Zd d hZddiZ dZ!ejDjGde!dZ$ejDjGdedZ%ejDjGdedZ&ejDjGdedZ'ejDjGdedZ(dZ)dZ*dZ+dZ,dZ-d Z.ejDjGd!gd"ejDjGd#ed$Z/ejDjGd#ed%Z0d&Z1ejDjGd'e1d(Z2d)Z3ejDjGd*ed+Z4ejDjkd,d-Z6d.Z7y)/N)assert_allclose)stats)check_normalization check_momentcheck_mean_expectcheck_var_expectcheck_skew_expectcheck_kurt_expect check_entropycheck_private_entropycheck_edge_supportcheck_named_argscheck_random_state_propertycheck_picklingcheck_rvs_broadcastcheck_freezing) distdiscreteinvdistdiscrete)rv_discrete_frozen)r)皙?皙?g333333?g?valueszipfian nhypergeomnbinomc#Kt}tD][\}}|tvr4tj|||tj j n ||||vf|j|]yw)N)marks)setrdistslowpytestparammarkslowadd)seendistnameargs `/var/www/html/bid-api/venv/lib/python3.12/site-packages/scipy/stats/tests/test_discrete_basic.pycases_test_discrete_basicr0s` 5D% # x ,,xhfkk>N>NO OC!55 5  &sA0A2zdistname,arg,first_casec tt|}tjj d|j |ddi}tj|}|j|\}}t||||dzt|||t||||dzt||d}t|||||dz|rd } |j|j|j|j |j"g} d d d d d} | j%|d} t'|| || | |dk7r t)|t+||t-||t/||t1||||j2j4tj6j4k7rt9||tj6yyy#t$r|}d}YwxYw)Nsample distributioniPsizeiz cdf_ppfz oth{Gz?z chisquare)r rr)randint hypergeom bernoullinchypergeom_walleniusr)getattrr TypeErrornprandomseedrvsunique check_cdf_ppf check_pmf_cdf check_othrcheck_discrete_chisquarepmflogpmfcdflogcdflogsfgetrcheck_scale_docstringrrrr __class___entropy rv_discreter ) r-r. first_casedistfnr@suppmvalphalocscale_defaultsmethsspec_kks r/test_discrete_basicrZ&s))IINN7 &** %$ %C 99S>D 6<< DAq&#tX %:;&#x( fc4F!23vs# EVS#u% 46 V]]FJJ  aa+,. JJx #C):EB , , !& )#FC0vs#vs# fc8,    $ $(9(9(B(B B !&#u/@/@ A C%# )()sF.. F?>F?z distname,argc  tt|}|j|ddi\}}}}t|||t |||||t ||||t |||||t||||||tjj5}|dvr|jtt||||||dddt|||dt|||||zzdy#t$r|}d}YwxYw#1swY?xYw)Nr2momentsmvsk)zipf betanbinomrr)r;rr<rrrr r r=testingsuppress_warningsfilterRuntimeWarningr check_moment_frozen)r-r.rQrSrTsrYsups r/ test_momentsrgQs))s3F3JAq!QX.aH-fc1h/VS!Q1fc1aH5  % % '3 - - JJ~ &&#q!Q9 ( Q*QqsUA.% )() ( 'sC%*C8% C54C58Dzdist,shape_argsc :|dv} tt|}t j d}|j}g}g}t|D]P}|dzfd|dzzz} ||} |jt j| | |jd | d R|j||j|jt|||||t jtgy#t$r#|}d|jd|jd}YwxYw) N) betabinomr_skellam yulesimondlaplacenchypergeom_fisherr:zrv_discrete(values=(z, z))rr)rrr)r;rr<xkpkr=zerosnumargsrangeappendfullinsertr3rdtypeint) dist shape_args shape_onlydistfunclocnargsallargsbshaperYshp param_vals r/test_rvs_broadcastrjs33JA5$' ((1+C   EG F 5\1uhq1u%qM rwwsI./ aQ   NN3 MM#(($bhhsm_# A%dgg[477+R@AsC..(DDz dist,argsc  tt|}tjj dtjj dddtjj ddg}|j|\}}|D]M}tj|dz |z||zg|jdg|d|i|jd g|d|igOy#t$r|}YwxYw) NMrr r|?) r;rr<r=r>r?r7supportnptassert_array_equalppf)rxargsrQre_locs_a_br|s r/test_ppf_with_locrs%IINN7yy  b)1bii.?.?2.FGG V^^T "FB  T#Xr#v  VZZ ,d , ,jfjj.Mt.M.M N  sC C%$C%z dist, argscZ tt|}tjj dtjj dddtjj ddg}|j|\}}|D]Q}||z|dz |zf}|jdg|d|i|jd g|d|if}tj||Stjj ddd tjd tjj ddd g}|j|\}}|D]Q}||z|dz |zf}|jdg|d|i|jd g|d|if}tj||Sy#t$r|}YwxYw) Nrrrrrrrr|r)rr3) r;rr<r=r>r?r7risfrrrp) rxrrQrrrr|expectedress r/test_isf_with_locrs%IINN7yy  b)1bii.?.?2.FGG V^^T "FB8R!Vc\)fjj,d,,jfjj.Ld.L.LL x- yy  bv 6xxyy  BV 46GV^^T "FB8R!Vc\)fjj,d,,jfjj.Ld.L.LL x- sF F*)F*cZ|j|g|}tj|jd}||t j |zz }t j|j|g|||dzt j|j|j|g|dz g|||dzt|dsg|j|\}}|||k} t j|j|j| g|dzg|| |jz|dzyy)Nz -roundtrip:0yE>rnz ppf-cdf-next) rHroundtrip_cdf_ppf_exceptionsrKnamer=spacingrrrhasattrrinc) rQr.rRmsgcdf_suppn_ulps cdf_supp0rrsupp1s r/rBrBs/vzz$%%H* - -fkk2 >F6"**X"666I:6::i6#6|!35:6::jfjj&<&rratolrtolrr?r|)rrnrwrlistrrrHrFcumsumrrr=asarraydiff assert_equal) rQr.r-indexstartindcdfspmfs_cumrrrY k_shiftedr|rxs r/rCrCsvvt zvzz$--12U8X]34 6::e "c "Dvzz%&#&--/HJD$9 dtAw8A;(>!. 5A#2A%IZVZZ 0C0!4 C C c D12-rwwtxxC7H/IJTXXi#o.2rcZtj||j||ddy)Nrr)rrmoment)rQr.rSrYs r/rdrds( ++A."0rctj|j|g|d|j|g|z ddt j ddd}tj|j |g||jd|z g|dd|j dg|}tj|j|dz g|dkDtj|j|dzg|dkDy) Nrrrr4Gz?rr) rrsfrHr=linspacerrassert_)rQr.rRrq median_sfs r/rDrDs  $--rJFJJt4Jc4J/J"0 D$#A  1+s+ZVZZQ-E-E"0 3%%IKK  )a-.#.45KK  9q=/3/#56rc &d}|j|\}}tt|d}tt|ddz} t || } d} |g} g} | D]R}|j |g|}|| z |dz k\s"| j || j || z |} |d|z kDsRn| d|kr%| j || j d| z tj| } tj| } | dz}||d<tj||\}}tjtj|t|| z\}}tj||kDd |d t|d t|y ) asPerform chisquare test for random sample of a discrete distribution Parameters ---------- distname : string name of distribution function arg : sequence parameters of distribution alpha : float significance level, threshold for p-value Returns ------- result : bool 0 if test passes, 1 if test fails g?iirrg+=rrzchisquare - test for z at arg = z with pval = N)rrwmaxminrrrHrsr=array histogramr chisquarelenrrstr)rQr.r@rUrwsupprrlohigh distsupportlastdistsuppdistmassiicurrenthistsuppfreqhsuppchispvals r/rErEs$ EV^^S !FB SU^ B s2t}  !DD/K DtHH&**R&#& T>UU] * OOB  OOGdN +D!e)$|bD!xx!Hxx!H$HHQK,,sH-KD%$S(1BCJD$KK u uJs3xj c$i[Qrc`|j"tjd|jvyy)Nscale)__doc__rr)rQs r/rLrL=s& ~~! G6>>12"rmethod)rFrGrHrIrrJrrzdistname, argsc L tt|}|dvrddg}nddg}|Dcgc]}|gdz }}ddg}|j|g|d|i}t j |t |g||D cgc]} |j| c} dd y#t$rYywxYwcc}wcc} w) N)rrrrrrrr|V瞯<)rr)r;rr<rFrrzip) rr-rrxzpp2r|resultrTs r/test_methods_with_listsrCs uh' #J F A1#a%B  a&C TXXa &" &# &F/21/?r/?3/?@/?!1/?@"0  AsB B,B! BBctt|}tjdd}|j|g|}tj }t j||y)Nr)r;rr=arangerHnanrr)r-rrxxvalsrs r/test_cdf_gh13280_regressionrYsN 5( #D "bA 488A  DvvHT8$rc#Kdhdhd}t}tD]\}}||vr |j| tt|}|j }t|D]I\}}|j|j|tvs |js9||j|fKy#t $rYwxYww)Nn)r!r_) r%rr+r;rr< _shape_info enumeraterrK integrality)integrality_exceptionsr,r-shapesrx shape_infoishapes r/"cases_test_discrete_integer_shapesrcs*-cUC 5D(& t    5(+D%%' !*-HAu 48835II))EJJ. . .)   s)4CB1A*C1 B=:C<B==Czdistname, shapename, shapesctt|}|j}|Dcgc]}|j}}|j |}t |}||} | dz } | dz } | g| g| gg||<|j |\} } tjtj| | d}|j|g|}tjtj|dddfrJtjtj|dddfsJtjtj|dddfrJycc}w)Nrrrrr)r;rrrrrrr=roundrrFanyisnanall)r- shapenamerrxrr shape_namesr shapes_copy valid_shape invalid_shapenew_valid_shapeabrrFs r/test_integer_shapesr}s7 5( #D!!#J+56:%5:::K6)$Av,K)K#%M!AoO"mm_6GHKN 4<< DAq Q1%&A $((1 #{ #Cvvbhhs1a4y)** * 66"((3q!t9% && &vvbhhs1a4y)** **!7sEcd}tjt|5tjddj dddtjt|5tjddj ddddtj_tjdd}t|tsJttjdy#1swYxYw#1swYmxYw)Nz,'rv_discrete_frozen' object has no attribute)matchrrherringpdf) r'raisesAttributeErrorrbinomrlogpdf isinstancerdelattr)message frozen_binoms r/test_frozen_attributesrs=G ~W 5 B   6 ~W 5 B## 6EKKO;;r3'L l$6 77 7 EKK 6 5 5 5s!C#"!C/#C,/C8zdistname, shapesc~t|trtt|}n|}|j|\}}t j |jddgg||dz |ft j |jddgg||dz |ft j |jdg||dz |fy)Nrr) r rr;rrrrrrinterval)r-rrxrrs r/ test_intervalrs (C uh' 4<< DAqXTXXq!f.v.1a9XTXXq!f.v.1a9]T]]1.v.1a9rzSensible to machine precisionc tjjd}tj|jddz}|jd}|tj|z}t j ||ftjdg|ddjz|zdz }tj|ddj|ddzgz|zdz }tj|}tjdg|ddjz|zdz }tj|ddjdgz|zdz }j\}} t||dt| |dtj||tj|dtj|dtjd 5tj|tj|tj|tj tj|tj dddtj#||tj#||tj#|dg|ddjztjd 5tj%|tjj#|d tj%|tjj#|d tj%|tjj#|d dddtj'|dj#|z tj'|dj#|z tj'|dj#|z tjd 5tj)|tjj'|d tj)|tjj'|d tj)|tjj'|d dddtj+||tj+||tj+|dd|ddtj+d|dz tj+d| j'|} tj-| |tj-d|z j+|tj-d|z j+|tj-d| tj-d|dz tj.d d d} j+| dz j-| dz f} tj1d| z | tj1dj3tj1d|dz | fj+d} tj3| tj||z}tj||z dz|z}tj||z dz|z|dzz }tj||z dz|z|dzz dz }tj5|tj7tj8|tj;|tj d||||ftj|dz|z}tj=d|tj?ddtj?|tj?d|tjd 5tj?fd jAdddtjjd}jCd|}tjjd}j+|jd}t||y#1swY$xYw#1swYxYw#1swYxYw#1swYxYw)NlKU[rrrrrrignore)divider)rr4rrrg?rr])r\cy)Nrrrs r/z test_rv_sample..s!rc |dzS)Nrrrs r/rz test_rv_sample..s!Q$rc&j|S)N)rG)rrxs r/rz test_rv_sample.. s t{{1~rd)r3 random_stater)"r=r> default_rngsortsumrrOrtolistrrrrFerrstaterGloginfrHrIrrJrrrrmedianmeanstdsqrtvarrexpectentropyr@)rngrnroxk_leftxk_rightcdf2 cdf2_left cdf2_rightrrsf2psint2med2mean2var2skew2kurt2mom3r@rvs0rxs @r/test_rv_sampler=sC ))   ,C B"$ %B BB"&&*B   RH -Dxxb"gnn../"4a7GABBrF1H:56;Q>H 99R=D1#Sb 0 0 223d:A=I((48??,s23d:A=J <<>DAqAr!uAr"vDHHRL"%DHHX&*DHHW%q) H % B4 H-w7 G,rvvg6 & DHHRL$'DHHX&-DHHW%s49+;+;+='=> H % B )=" $ H-rvvdhhx6H/I" $ G,bffTXXg5F.G" $ &DGGBK488B<0DGGH%q();';<DGGG$a(9&9: H % 2twwr{(;" $ 8,bffTWWX5F.G" $ 7+RVVDGGG4D-E" $ &DHHTNB'DHHY',DHHZ("-r!"v6DHHQKQ'DHHQK# ''"+CDHHSM2&DHHQy[)488I+>?DHHQz\*DHHZ,@ADHHQK#DHHQKQ' T4 $B 88BqD>488BqD> )DDMM!B$'.DMM!$dkkm4DMM!$qsAh/ 88C=DDKKM4( FF2b5ME 662:/B& 'D FFBJ?R' (4#; 6E FFBJ?R' (47 2Q 6EDIIK'DHHJ .DHHJ%DJJvJ.eU0KL 662q5B, DDKKND)DKK ,a0DKKM5)DKK/6 H %%=>> O & ))   ,C ((#( .C ))   ,C 88CJJCJ( )DCk & % & % & %l & %s3f.&B>f;,.g!f+.f8;ggcGddtj}|dtd}|jd}t |dy) NceZdZdZy)-test__pmf_float_input..rv_exponentialcddd|z zzS)NgUUUUUU?rrr)selfrs r/_pmfz2test__pmf_float_input..rv_exponential._pmf.sQU# #rN)__name__ __module__ __qualname__rCrrr/rv_exponentialr@-s $rrGrr%)rr*)rr)rrOfloatr@r)rGrvr@s r/test__pmf_float_inputrK)s@$**$ #u .B &&b& !CCr)8 numpy.testingr`rrnumpyr=r'scipyr common_testsrrrr r r r r rrrrrrscipy.stats._distr_paramsrr!scipy.stats._distn_infrastructurerrrOr&rr0r) parametrizerZrgrrrrBrCrdrDrErLrrrrrrxfail_on_32bitr=rKrrr/rTsR) ,,,,D@*+ #%##40"566  | $!)"~24M4OP'BQ'BT6/7/0*L9$:$Nl3 4  |4.5.0J4360 76r3 $AB)<809B0&)?;%<%/46;=?+?+,  +\: :; :;<s=sj r