Q(hRddlmZddlmZddlZddlZddlmZddl 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 c/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/stats/tests/test_discrete_basic.pycases_test_discrete_basicr0se 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+|||dvr t-||t/||t1||||j2j4tj6j4k7rt9||tj6yyy#t$r|}d}YwxYw)Nsample distributioniPsizeiz cdf_ppfz oth{Gz?z chisquare)r rr)randint hypergeom bernoullinchypergeom_walleniusr> poisson_binom)getattrr TypeErrornprandomseedrvsunique check_cdf_ppf check_pmf_cdf check_othrcheck_discrete_chisquarepmflogpmfcdflogcdflogsfgetrcheck_scale_docstringrrrr __class___entropy rv_discreter ) r-r. first_casedistfnrAsuppmvalphalocscale_defaultsmethsspec_kks r/test_discrete_basicr[&s))IINN7 &** %$ %C 99S>D 6<< DAq&#tX %:;&#x( fc4F!23vs# EVS#u% 46 V]]FJJ  aa+,. JJx #C):EB , , !& )#FC0 , , 63 'vs# fc8,    $ $(9(9(B(B B !&#u/@/@ A C'# )()sF22 GGz 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.rRrTrUsrZsups r/ test_momentsrhRs))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}|dk(rEt jdd}||f}d }t|||||t jtgyt|D]P} | d zfd | d zzz} || } |jt j| | |jd | d R|j||j|jt|||||t jtgy#t$r#|}d|jd|jd}YMwxYw)N) betabinomr`skellam yulesimondlaplacenchypergeom_fisherr:r;zrv_discrete(values=(z, z))rr;)rr ?)rrr)rrr)r<rr=xkpkr>zerosnumargsfullrdtypeintrangeappendinsertr3) dist shape_args shape_onlydistfunclocnargsallargsbshapeprZshp param_vals r/test_rvs_broadcastrksfDDJA5$' ((1+C   EG F  GGJ $c(HdG"J# A5\!1uhq1u%qM rwwsI./ aQ !  NN3 MM#(($bhhsm_7 A%dgg[477+R@AsD88(E$#E$z dist,argsc  tt|}tjj dtjj dddtjj ddg}|j|\}}|D]M}tj|dz |z||zg|jdg|d|i|jd g|d|igOy#t$r|}YwxYw) NMrrror?) r<rr=r>r?r@r7supportnptassert_array_equalppf)r{argsrRre_locs_a_brs 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) Nrrrrrrorrr)rr3) r<rr=r>r?r@r7risfrrrs) r{rrRrrrrexpectedress r/test_isf_with_locrs%IINN7yy  b)1bii.?.?2.FGG V^^T "FB.8R!Vc\)fjj,d,,jfjj.Ld.L.LL x-. yy  bv 6xxyy  BV 46GV^^T "FB.8R!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>rqz ppf-cdf-next) rIroundtrip_cdf_ppf_exceptionsrLnamer>spacingrrrhasattrrinc) rRr.rSmsgcdf_suppn_ulps cdf_supp0rrsupp1s r/rCrCs/vzz$%%H* - -fkk2 >F6"**X"666I:6::i6#6|!35:6::jfjj&<&rratolrtolrrrpr)rrqrwrlistrxrIrGcumsumrrr>asarraydiff assert_equal) rRr.r-indexstartindcdfspmfs_cumrrrZ k_shiftedrr{s r/rDrDsvvt 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)rRr.rTrZs r/reres( ++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?rpr) rrsfrIr>linspacerrassert_)rRr.rSrq median_sfs r/rErEs  $--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)rrwmaxminrxrIryr>array histogramr chisquarelenrrstr)rRr.rArVrwsupprrlohigh distsupportlastdistsuppdistmassiicurrenthistsuppfreqhsuppchispvals r/rFrFs$ 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)rRs r/rMrMGs& ~~! G6>>12"rmethod)rGrHrIrJrrKrrzdistname, argsc < tt|}t||}|dvrddg}nddg}|Dcgc]}|gdz }}ddg}||g|d|i} tj| t |g||D cgc]} ||  c} dd y#t$rYywxYwcc}wcc} w) N)rrrrrrrrV瞯<)rr)r<rr=rrzip) rr-rr{ dist_methodzrp2rresultrUs r/test_methods_with_listsrMs uh'$'K  #J F A1#a% B  a&C  )R )S )F25a2B"2Bc2BCQaC"0  DsB B. B  BBctt|}tjdd}|j|g|}tj }t j||y)Nr)r<rr>arangerInanrr)r-rr{xvalsrs r/test_cdf_gh13280_regressionrdsN 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 enumeraterrL integrality)integrality_exceptionsr,r-shapesr{ shape_infoishapes r/"cases_test_discrete_integer_shapesrns*-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)Nrprrrr)r<rrrrrrr>roundrrGanyisnanall)r- shapenamerr{rr shape_namesr shapes_copy valid_shape invalid_shapenew_valid_shapeabrrGs r/test_integer_shapesrs6 5( #D!!#J+56%5::6K6)$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)matchrorpherringpdf) r'raisesAttributeErrorrbinomr logpdf isinstancerdelattr)message frozen_binoms r/test_frozen_attributesrs=G ~W 5! B  ! ~W 5$ B##$EKKO;;r3'L l$6 77 7 EKK!!$$s!C#"!C/#C,/C8zdistname, shapesc~t|trtt|}n|}|j|\}}t j |jddgg||dz |ft j |jddgg||dz |ft j |jdg||dz |fy)Nrr) rrr<rrrrrrinterval)r-rr{rrs 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[rorrrrrignore)divider)rr4rrprg?rr^)r]cy)Nrrrs r/z test_rv_sample..%src |dzS)Nrrrs r/rz test_rv_sample..'s !Q$rc&j|S)N)rH)rr{s r/rz test_rv_sample..+st{{1~rd)r3 random_stater)"r>r? default_rngsortsumrrPrtolistrrrrGerrstaterHloginfrIrJrrKrrrrmedianmeanstdsqrtvarrexpectentropyrA)rngrqrrxk_leftxk_rightcdf2 cdf2_left cdf2_rightrrsf2psint2med2mean2var2skew2kurt2mom3rArvs0r{s @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 %7 B4 H-w7 G,rvvg67 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 %P%=>> OP ))   ,C ((#( .C ))   ,C 88CJJCJ( )DCk77$$$$lPPs3f.&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._pmf9sQU# #rN)__name__ __module__ __qualname__rErrr/rv_exponentialrB8s $rrIrr')rr*)r r)rrPfloatrAr)rIrvrAs r/test__pmf_float_inputrM4s@$**$ #u .B &&b& !CCr)8 numpy.testingrarrnumpyr>r'scipyr common_testsrrrr r r r r rrrrrrscipy.stats._distr_paramsrr!scipy.stats._distn_infrastructurerrrPr&rr0r) parametrizer[rhrrrrCrDrerErFrMrrrrrrxfail_on_32bitr?rMrrr/rVsR) ,,,,D@*+ #%##40"566  | $!)"~24M4OP(BQ(BV6/7/0*L9-:-`l3 4  |4.5.0J4360 76r3 $AB)<809B0()?;%<%/46;=?+?+,  +\: :; :;<s=sj r