Џkh[SZddlmZddlmZddlZddlZddlmZddl 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(Z2ejDjgdd)Z4ejDjGd*ed+Z5ejDjmd,d-Z7d.Z8y)/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 a/var/www/teggl/fontify/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_casect|tr,|jdr|dkDrtjd t t |}tjjd}|j|d|d}tj|}|j |\}} t||||d zt|||t||||d zt!||d } t#|||| |d z|rd } |j$|j&|j(|j*|j,g} ddddd} | j/|d}t1|||| | |dk7r t3||dk(rt5|||dvr t7||t9||t;||||j<j>t j@j>k7rtC||t j@yyy#t$r|}d}YwxYw)N nchypergeomrz)nchypergeom has a global random generator)reasonsample distributioniPisize random_statez cdf_ppfz oth{Gz?z chisquare)r rr)randint hypergeom bernoullinchypergeom_wallenius> poisson_binom)" isinstancestr startswithr'skipgetattrr TypeErrornprandom RandomStatervsunique check_cdf_ppf check_pmf_cdf check_othrcheck_discrete_chisquarepmflogpmfcdflogcdflogsfgetrcheck_scale_docstringrrrr __class___entropy rv_discreter )r-r. first_casenum_parallel_threadsdistfnrngrIsuppmvalphalocscale_defaultsmethsspec_kks r/test_discrete_basicre&s8S!h&9&9-&H$q( FG)) ))   (C &**c3 7C 99S>D 6<< DAq&#tX %:;&#x( fc4F!23vs# EVS#u% 46 V]]FJJ  aa+,. JJx #C):EB , , !& ) 1 $ ' 400vs+vs# fc8,    $ $(9(9(B(B B !&#u/@/@ A C)# )()sG44 HHz distname,argc  tt|}|j|ddi\}}}}t|||t |||||t ||||t |||||t||||||tjj5}|dvr|jtt||||||dddt|||dt|||||zzdy#t$r|}d}YwxYw#1swY?xYw)Nr4momentsmvsk)zipf betanbinomrr)rDrrErrrr r rFtestingsuppress_warningsfilterRuntimeWarningr check_moment_frozen)r-r.r[r^r_srdsups r/ test_momentsrrWs))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) betabinomrjskellam yulesimondlaplacenchypergeom_fisherr>r?zrv_discrete(values=(z, z))rr?)rr ?)rrr)rrr)rDrrExkpkrFzerosnumargsfullrdtypeintrangeappendinsertr6) dist shape_args shape_onlydistfunclocnargsallargsbshapeprdshp param_vals r/test_rvs_broadcastrpsfDDJA5$' ((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) NMrrryr?) rDrrErFrGseedr;supportnptassert_array_equalppf)rargsr[re_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) Nrrrrrryrrr)rr6) rDrrErFrGrr;risfrrr}) rrr[rrrrexpectedress 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>r{z ppf-cdf-next) rQroundtrip_cdf_ppf_exceptionsrTnamerFspacingrrrhasattrrinc) r[r.r]msgcdf_suppn_ulps cdf_supp0rrsupp1s r/rKrKs/vzz$%%H* - -fkk2 >F6"**X"666I:6::i6#6|!35:6::jfjj&<&rratolrtolrrrzr)rr{rrlistrrQrOcumsumrrrFasarraydiff assert_equal) r[r.r-indexstartindcdfspmfs_cumrrrd k_shiftedrrs r/rLrLsvvt 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)r[r.r^rds r/roros( ++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) Nrrrr8Gz?rzr) rrsfrQrFlinspacerrassert_)r[r.r]rq median_sfs r/rMrMs  $--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)rrmaxminrrQrrFarray histogramr chisquarelenrrrA)r[r.rIr`rwsupprrlohigh distsupportlastdistsuppdistmassiicurrenthistsuppfreqhsuppchispvals r/rNrNs$ 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)r[s r/rUrULs& ~~! G6>>12"rmethod)rOrPrQrRrrSrrzdistname, 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)rDrrErrzip) rr-rr dist_methodzrp2rresultr_s r/test_methods_with_listsrRs 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)rDrrFarangerQnanrr)r-rrxvalsrs r/test_cdf_gh13280_regressionrisN 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!rj) r%rr+rDrrE _shape_info enumeraterrT integrality)integrality_exceptionsr,r-shapesr shape_infoishapes r/"cases_test_discrete_integer_shapesrss*-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)Nrzrrrr)rDrrrrrrrFroundrrOanyisnanall)r- shapenamerrrr shape_namesr shapes_copy valid_shape invalid_shapenew_valid_shapeabrrOs 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)matchryrzherringpdf) r'raisesAttributeErrorrbinomrlogpdfr@rdelattr)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) r@rArDrrrrrrinterval)r-rrrrs 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[ryrrrrrignore)divider)rr8rrzrg?rrh)rgcy)Nrrrs r/z test_rv_sample..+src |dzS)Nrrr$s r/r%z test_rv_sample..-s !Q$rc&j|S)N)rP)rrs r/r%z test_rv_sample..1st{{1~rdr5r)"rFrG default_rngsortsumrrXrtolistrrrrOerrstaterPloginfrQrRrrSrrrrmedianmeanstdsqrtvarrexpectentropyrI)r\r{r|xk_leftxk_rightcdf2 cdf2_left cdf2_rightrrsf2psint2med2mean2var2skew2kurt2mom3rIrvs0rs @r/test_rv_samplerFsc ))   ,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._pmf?sQU# #rN)__name__ __module__ __qualname__rLrrr/rv_exponentialrI>s $rrPrr/)rr*)r7r)rrXfloatrIr)rPrvrIs r/test__pmf_float_inputrT:s@$**$ #u .B &&b& !CCr)9 numpy.testingrkrrnumpyrFr'scipyr common_testsrrrr r r r r rrrrrrscipy.stats._distr_paramsrr!scipy.stats._distn_infrastructurerrrXr&rr0r) parametrizererrrrrrKrLrorMrNrUrrrrparallel_threadsrrxfail_on_32bitrFrTrrr/r^sp) ,,,,D@*+ #%##40"566  | $!)"~24M4OP-BQ-B`6/7/0*L9-:-`l3 4  |4.5.0J4360 76r3 $AB)<809B0()?;%<%/46;=?+?+,a   !  +\: :; :;<s=sj r