Qi+<ddlmZddlmZmZmZmZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddl m!Z!ddl"m#Z#m$Z$ddlm%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBddlCmDZDmEZEmFZFmGZGddlHmIZIddlJmKZKe dZLdZMdZNdZOeDdZPdZQdZReEd ZSd!ZTd"ZUd#ZVd$ZWd%ZXd&ZYd'ZZd(Z[d)Z\d*Z]d+Z^d,S)-)Sum)IRationaloopiS)Symbol)imre)log)floor)sqrt) Piecewise)besseli)beta)zeta) FiniteSet)simplify)lambdify)EqNe)exp)Or)Range) PEvariancedensitycharacteristic_functionwheremoment_generating_functionskewnesscdfkurtosis coskewness) PoissonDistributionGeometricDistribution FlorySchulzPoisson GeometricHermite LogarithmicNegativeBinomialSkellam YuleSimonZeta DiscreteRV)slow nocache_failraisesskip) Expectation)FallingFactorialxcd}t|}t|ddz dksJt|ddz dksJ|t t |ksJ|t dzt |t t dzz |ksJdS)N MbP?g$@)r'absr$evalf expectationr9)lps z/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/sympy/stats/tests/test_discrete_rv.pytest_PoissonDistributionrF"s AAA quuRyy  1$ % % , , , , quuT{{  ""Q& ' '$ . . . . ==A  ! # # # # ==Aq ! !AMM!Q$7$7$: :a ? ? ? ? ? ?cd}td|}t||ksJtd|zd|zksJt||ksJt|t |ksJt t|dt sJt td|zdt sJ|jddksJ ddl }n#t$rtdYnwxYwtd | d }t|d ksJtd | d }t|d ksJd }td |}t||ksJtt|d|dzksJt|dz||dzzksJdS)Nr;r9r?Fevaluater=rznumpy not installedygKPh;x=gUPh;x=z)r*rrrr' isinstancer7pspacecompute_expectationnumpy ImportErrorr6float64r8)rCr9nprKl2rMs rE test_PoissonrV+s AQA Q4419999 QqS66QqS==== A;;!     1::,Q// / / / / aE***K 8 8888 a!e,,,k : :::: 8 ' ' * *a / / / /$ $$$ "#####$RZZ 45566A Q44' ' ' ' 'RZZ 56677A Q44( ( ( ( ( BRA Q442:::: a # # $ $A - - - - QT77b2q5j s*C//D  D c~td}td}td|}t|d|z |z ksJt|dd|z z|dzz z t dksJt |||dz|zd|z |dz zzksJdS)NarMr9r?r=r)r r)rrrr r)rXrMr9s rEtest_FlorySchulzrYFss As ACA Q44AE19     QKK!QU)AqD. ( 2 2 4 4! < < < < 71::a==AqDFAEQU#33 3 3 3 3 3 3rGc,tjdz }t|}|ttd|z ksJ|tdzt|ttdzz d|z |dzz ksJt |ddz dksJt |ddz dksJtdtddz }t |td dz t|td dz kksJtd tdd}td td d}t|||z|d|zz tdtddzksJdS)NrLr=r?i Nr>g33333@G)rDXYr;r<Qi~)r Oner(rBr9r@r$rAr+rrr&rr)rDdr[r^r_s rEtest_GeometricDistributionrdOs  Aa  A ==A  !A# % % % % ==Aq ! !AMM!Q$7$7$: :qsAqDj H H H H quuU||!!##a' ( (4 / / / / quuW~~##%%) * *T 1 1 1 1#1a   A 3q66!A$$q&>>QqAaDDF{^^ + + + +#x1~~&&A#x2''A aQAaC ( ( 1 1 3 3tCyy"dASAS7S S S S S S SrGc ztddtddttfdtddtddttfdtddtd}td }t ||t t |d z zt d |zd z zzksJt ||t t t|zd z zt d tz|zd z zzksJt|d zzksJtd d d }t|d dt dzd z ksJt|dksJt|dksJt|tddksJt|dtdzdz ksJdS)Na1T)positivea2)negativec&tdSNHr,rfrhsrEztest_Hermite..bwsB33rGc&tdSrkrmrnsrEroztest_Hermite..frprGr9rlr=r?rLr\rn!i i%i)r r5 ValueErrorr,r"rr rrrrr%rr#r)r9rlrfrhs @@rE test_Hermiterx_sc t $ $ $B t $ $ $B :33333444 t $ $ $B t $ $ $B :33333444 t $ $ $Bs ARA ( %a ( ( + +s2s1vvz?.0#ac((Q,.?4@0A0A A A A A % "1 % %a ( (CC!HHqL0A.0#ac!e**q..A1B-C-C C C C C Q442"9    a   A 71::a==Bs2wwJqL ( ( ( ( Q442:::: A;;"     A;;(3s++ + + + + A;;"T"XX+c/ ) ) ) ) ) )rGctj}td|}t|| d|z t d|z zz ksJt |dt ddzz dt dz zksJtd|dzzd|zzdzddt dz zksJt t|d tsJdS) Nr9r=r?r;r\r]FrI)r Halfr-rr rrNr7)rDr9s rEtest_Logarithmicr|xs ACA Q44A2!a%3q1u::-. . . . . A;;"SVVQY,3q661 1 1 1 1 Qq!tVac\A   !a#a&&j. 0 0 0 0 aE***K 8 888888rGcd}tjdz }td||}t||d|z z|z ksJt ||d|z z|dzz ksJt|dzd|zzdzt|dzdt|zzdzcxkrt ddksnJt t|dtsJdS) NrLr;r9r=r?i9' FrI)r rbr.rrrrNr7)rrDr9s rEtest_negative_binomialrs A  Aa##A Q441A;? " " " " A;;!q1u+1, , , , , QTAaCZ!^  !Q$!AaDD& 01 4 K K K K8K8K K K K K K K aE***K 8 888888rGc <td}td}td}td||}t||||z |dz zt| |z zt |dt ||zzzksJt |||t ||zz|t ||zzzz ||t ||zz|t ||zzzz z ksJt|||zksJt|||z ksJt||t|tt|zz|z |z |tt |zzzksJt||t|t|z|z |z |t| zzksJdS)Nmu1mu2rMr9r?) r r/rrrrr#expandrrr rr")rrrMr^s rE test_skellamrs; --C --Cs AS#A 71::a==SW!, SD3J1T#c']]?334 4 4 4 4 A;;    3DsOO(;cS3Y? ) $!#&DsOO(;c$sSy//>Q(Q#R$S S S S S A;;    39 , , , , Q4439     % "1 % %a ( (C C!HH sS 3sA2a4yy=0-2-2 2 2 2 2 ( %a ( ( + +s CFF S3S!WW,0.0. . . . . . .rGcddlm}|d}td|}tt |||dz z ksJtt ||dz|dz dz|dz zz ksJt t |dtsJt||ttt|d  t|zdz|dkfd ksJdS) Nrrr;r9r=r?FrIr\)rT) sympy.core.singletonr r0rrrrNr7r$rrr)r rhor9s rEtest_yule_simonrs &&&&&& !A$$C#sA AaDD>>SC!G_ , , , , HQKK CFsQwlcAg.F$G G G G G aE***K 8 8888 3q66!99 Dq1$5$5#5eAhh#>#BAF"KYWW W W W W W WrGctd}td|}t|t|dz t|z ksJt t |t|t|dz zt|dz dzz t|dzz ksJdS)NrLr9r=r?)r r1rrrr)sr9s rE test_zetars !A S! A Q444!99tAww& & & & & HQKK  Q$qs))d1Q3iil*d1ggqj%9 9 9 9 9 9 9rGc tdtdd}tdd}tdt}t t |dtdd ksJt |dktd d ksJt |dktd d ksJt |dktdd ksJt |dktd d ksJt t |dtdd ksJt t |ddtdzdz ksJt |dkdtdzksJt |dkdtdddtdzdz zztdzdz sJt |dkdtdddtdzdz zztdzdz sJt |dkdtdzdz ksJt t |ddtdzdtdddtdzdz zztdzdz zsJt |tj ktj usJt |tj ktj usJt |dktdtz zksJt t |dtt dzdzzksJdS)Nr^r=rLr_r\er;} @=m rsiiGr?) r+rr*r9rrrrequalsr InfinityrbZero)r^r_r[s rEtest_discrete_probabilityrsO#x1~~&&AQA#qA R1XX;;(2s++ + + + + QU88x2 & & & & QU88xC(( ( ( ( ( Q!V99R(( ( ( ( ( Q!V99S)) ) ) ) ) R1XX;;(3,, , , , , R1XX;;"SWW*Q, & & & & QU88r#b''z ! ! ! ! QU88??2xR001SVV8B;>?BGI J JJJJ Q!V99  Bb 1 1Ac!ffHRK ?@RHJ K KKKK Q!V993r77 1 $ $ $ $ R1XX;;   3r77 R#r**Qs1vvXb[89#b''A!CC E EEEE Q^   % % % % Q^   & & & & QU88q!A#w     R1XX;;!aR!VaK- ' ' ' ' ' 'rGc tddz }tddd|d|z dz zz}t|tjd}t |t t dtddz cxkrdksnJt |dktdd z ksJ|jjj tjksJttfd tdz tj}t |dkdksJt |tksJdS) Nr=r?r9T)integerrg)setcheckr[r;c RtttddS)Nr\T)r)r2rrange)r9srEroz!test_DiscreteRV..s#z!Q 5880DDQQQrG) r r r2Naturalsrr+rrOdomainrr5rwr)rDpdfDr^r9s @rEtest_DiscreteRVrs< !QAsD4000A QUa!e  C1cqz666A Q441YsAaDDF++,, 1 1 1 1 1 1 1 1 1 1 QU88qttAv     8? !* , , , , :QQQQRRR 1ac1:&&A QU88q==== Q442::::::rGc ddlfd}|tdtdddj|t dtdddj|t ddtdd dj|t d ddj|td ddj|td ddjdS) Nrc4 t|}td}td}t|gt||d}t||g||t t |z|zzd fd}dD]|}||} ||} t t| t| z dksJt t| t| z dksJ}dS)Ntr9mpmathcBfdgdS)Nc|S)N)r9frs rEroz^test_precomputed_characteristic_functions..test_cf....saa1ggrGr<) maxdegree)nsum)rrrsupport_lower_limitsupport_upper_limits`rErozLtest_precomputed_characteristic_functions..test_cf..s: $5$5$5$5$5 !486AC$EErG)r?rLr g-q=) rr rr rrr@r r ) distrrrrr9cf1cf2 test_pointn1n2rrs `` @rEtest_cfz:test_precomputed_characteristic_functions..test_cfs8dmm cFF cFFs93D99!<> * *Avz::: GKXa^^ , ,a<<< G S!Xa^^ 4 4aDDD GGCOOQ +++ GIc1  q&*--- GDaLL!VZ(((((rGc Vtd}ttdtj|}|||ddksJtt dtj|}|||ddtdz ksJttddtdd |}|||dtd dksJttd d|}|||ddksJttd dd|}|||dtd  tdztd td ztdzzksJttdd |}t|||dtd dksJttdd|}|||dt dzdt#dzz ksJdS)Nrrrr?rCr=rrLr;r<rDrrKrMr\Z)r r"r+r{diffsubsr-r r.rr*r/rr0rr1rr)r geometric_mgflogarithmic_mgfnegative_binomial_mgf poisson_mgf skellam_mgfyule_simon_mgfzeta_mgfs rE test_moment_generating_functionsrs #AF.yaf/E/EFFqIIM   a % %a + +q 0 0 0 0J0S!&1I1IJJ1MMO    " " ' '1 - -3q66 9 9 9 926a!Q00222355 % %a ( ( - -a 3 3xA F F F F=,WS!__==a@@K   A   # #Aq ) )Q . . . .@,WS!Q-?-?@@CCK   A   # # 1  r77(SVV#Sc"ggA)>%?%?? @ @ @ @C/ #q0A0ABB1EEN N''**//155 6 6(1a.. H H H H7)$sA,,77::H ==   A & &"a%DGG*< < < < < < > 1 1 1 1 1 1rGctdtdd}tdd}t|dk|dkdksJt|dk|dktddksJt|dk|dkdksJtt |d|dkdt dzdz ksJtt |dt |ddksJt|dkt |ddksJt|dkt |ddksJdS) Nr^r?r;r_r=rr)r+rr*rrrrs rEtest_conditionalr$sE#x1~~&&AQA QUAE??a     QUAE??hq!nn , , , , QUAE??a     R1XXqAv  !CGG)A+ - - - - R1XXr!Qxx A % % % % QUBq!HH   " " " " QUBq!HH   " " " " " "rGctdtj}tdtdd}t t ||zdktdksJt t ||zdktdksJt t||zdtddksJdS)NX1X2r=r;z@Sum(Piecewise((1/(4*2**n), n >= -1), (0, True)), (n, -oo, -1))/3zdSum(Piecewise((2**(X2 - n - 2)*(2/3)**(X2 - 1)/6, X2 - n <= 2), (0, True)), (X2, 1, oo), (n, 1, oo)) ) r+r r{rstrrrewriterr)rrs rEtest_product_spacesr0s 4 B 4!Q ( (B qb1~~%%c** + +J L L L L qb1~~%%c** + + = > > > > RR^^  B / / / / / /rGN)_sympy.concrete.summationsrsympy.core.numbersrrrrrr sympy.core.symbolr $sympy.functions.elementary.complexesr r &sympy.functions.elementary.exponentialr #sympy.functions.elementary.integersr(sympy.functions.elementary.miscellaneousr$sympy.functions.elementary.piecewisersympy.functions.special.besselr&sympy.functions.special.beta_functionsr&sympy.functions.special.zeta_functionsrsympy.sets.setsrsympy.simplify.simplifyrsympy.utilities.lambdifyrsympy.core.relationalrrrsympy.logic.boolalgrsympy.sets.fancysetsr sympy.statsrrrrr r!r"r#r$r%r&sympy.stats.drv_typesr'r(r)r*r+r,r-r.r/r0r1r2sympy.testing.pytestr3r4r5r6 sympy.stats.symbolic_probabilityr7(sympy.functions.combinatorial.factorialsr8r9rFrVrYrdrxr|rrrrrrrrrrrrrrGrErs))))))444444444444""""""$$$$$$99999999666666555555999999::::::222222777777777777%%%%%%,,,,,,------((((((((666666""""""&&&&&&//////////////////////////0000000000000000000000000000BAAAAAAAAAAA888888EEEEEE F3KK@@@!!!6444 T T T***2999999...$XXX999(((. )))@===6%%%222 # # #00000rG