K i2ddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZdd lm Z ddl!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLddlMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWddlXmYZYmZZZm[Z[m\Z\ddl]m^Z^ddlm_Z_ddl`maZaddlmbZbddlcmdZddZedZfdZgdZhdZidZjd Zkd!Zld"Zmd#Znd$Zod%Zpd&Zqd'Zrd(Zse[d)Ztd*Zud+Zvd,Zwd-Zxd.Zyd/Zzd0Z{d1Z|d2Z}d3Z~d4Zd5Zd6Zd7Zd8Zd9Zd:Zd;Zd<Zd=Zy>)?)Sum)Basic)Tuple)Lambda)Rationalnanoopi)Eq)S)Symbolsymbols)FallingFactorialbinomial)explog)cossin) DiracDelta) integrate)AndOr)Matrix)Interval)Indexed)"DieNormal ExponentialFiniteRVPEHvariancedensitygiven independent dependentwherepspaceGaussianUnitaryEnsemblerandom_symbolssample Geometricfactorial_momentBinomialHypergeometricDiscreteUniformPoissoncharacteristic_functionmoment_generating_functionBernoulliProcessVariance Expectation Probability Covariance covariancecmomentmomentmedian) IndependentProductPSpacers_swapDensityNamedArgsMixin RandomSymbol sample_iterPSpace is_randomRandomIndexedSymbolRandomMatrixSymbol)raisesskipXFAILwarns_deprecated_sympy) import_module)comp)BernoulliDistribution)Dummy) Piecewisectdtd}}tddd}t|dzdkjt ddk(sJt|dzdkj t ddj |jk(sJtt||kD|dkDj tt|jd t|jd k(sJtt|d kjdk(sJdt|d kjvsJtdddtddd}}tt|dzdk|dk\jt ddk(sJt|t|dzdk|dk\}|jjjt ddk(sJ|jjj td|jk|jdzdkt |jk|jtkk(sJtt 5t||d z}dddy#1swYyxYw) NXYZr)rrr(setr as_boolean as_relationalsymbolrr lenr%r)domainr rH TypeError)rRrSrTXXs _/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sympy/stats/tests/test_rv.py test_wherere!s* s8SXqAsAqA A   8B? 22 2 A  & & (HRO,I,I!((,S SS S QUAE" # . . 0C 188QAHHa5* ** * uQU| A %% % a!e    #q! fS!Q/qA QTQYQ' ( , ,A >> > q#adaia( )B 99    8Aq> 11 1 99   & & ( AM188Q;!+bS188^QXX]K LL L   1a!e_s I33I<cPtdddtddd}}ttd|zdz|hk(sJttd|z|z||hk(sJttd|z|jz|hk(sJttdtk(sJy)NrRrrUrSrV)rr\r+r_rRrSs rdtest_random_symbolsrh8s #q! fS!Q/qA ~acAg& 'A3 .. . ~acAg& 'Aq6 11 1 ~acAHHn- .1# 55 5 ~a !SU ** *c ,ddlm}tddd}tdgd}t dd}t d }t |t|dz dz }t |td |z|zd z td|z|zd z zt||zd z z}t |tdt||zzdz }t|j|sJt|j|sJt|j|sJy) Nr)IrRrUrSrUrVrTrV_trmr[) sympy.core.numbersrkrr1r2rrrr3dummy_eq)rkrRrSrTtr QRs rdtest_characteristic_functionrtAs$s1QAW%AQA Aq#q!teAg,Aq#ac!e*Q,QqSUA-AaC :;Aq#a#ac(lQ&'(A #1 % . .q 11 1 "1 % . .q 11 1 "1 % . .q 11 1ric tddd}tdgd}tdd}td}t |t |dzdz }t |t d |zd z t d|zd z zt |d z z}t |t dt |zdz }t |j|sJt |j|sJt |j|sJy) NrRrrUrSrlrTrVrnrmr[)rr1r2rrrr4rp)rRrSrTrqr rrrss rdtest_moment_generating_functionrvRss1QAW%AQA Aq#ad1f+Aq3qs8A:AaC *SVAX57Aq#a#a&j1n%&A &a ( 1 1! 44 4 %a ( 1 1! 44 4 %a ( 1 1! 44 4ricBtddd}tdgd}tdd}td}|s t d |dzd z}t |}|dzd |zzd z}t |}|d zd z}t |} d } | |sJ| |sJ| | sJy)NrRrrUrSrlrTrVscipy#Scipy is not installed. Abort testsr[rZrXct|dr@t|ds t|dr(t|jr|j|uryy)N__iter__next__next__TF)hasattrcallabler{)objs rd is_iteratorz%test_sample_iter..is_iteratorss@ C $ S& ! C $ S\\ " LLNc !ri)rr1r2rLrIrC) rRrSrTrxexpriteratorexpr2 iterator2expr3 iterator3rs rdtest_sample_iterrass1QAY'AQA ' "E  23 a4!8D4 H qD1Q3JNEE"I qD1HEE"I  x  y !! ! y !! !ricttdddtddd}}tdttdttfdt ||jk(sJt d|zdz|jk(sJt d|z|zt |j|jk(sJy) NrRrrUrSxctdS)Nr)rirdztest_pspace..s ve}ric tdkSNrUrrsrdrztest_pspace..sva!e}rirV)rr rH ValueErrorr)r>)rRrSrs @rd test_pspacers #q! fS!Q/qAs A :,- :,- !9  !A#'?ahh && & !A#'?6qxxJ JJ Jrictddd}tdd}tddd}tddd}d|z|z}|jt||f||fd|z|zk(sJy)NrrrUyrVr[)rrsubsr?)rRrSrcYYrs rd test_rs_swaprsssAqACA Q B Q B Q37D 99WaVb"X. /1R4"9 << .sqQxri)rrHrrgs@@rd test_overlaprs*sAqAsAqA :'(rictddd}tddd}|j}|j}t||zt||k(sJt||zt||k(sJyNrRrrUrS)rr)r>)rRrSpxpys rdtest_IndependentProductPSpacerslsAqAsAqA B B !a%=4R< << < !a%=4R< << .saA!4rig?c38K|]}|tddvyw)rUrmNrange.0is rd ztest_Sample..sCAqE!QKCc38K|]}|tddvyw)rXrmNrrs rdrztest_Sample..sHAqE!QKHrr[numpyz#Numpy is not installed. Abort tests)size)rrr rLrIr, isinstancefloatr is_numberr!r#rHrbrrallr$int32int64float64ndarrayrK)rRrxrrSrs @@rd test_Samplers C AsAqAsD!A ' "E  23 !9* ** * fQUmU ++ + QUQYA" - 7 77 7 QUr " , ,, , QTAX" % / // / a!eaZB ' 1 11 1 AEb ) 3 33 3 945 SVq[R (C // / SVq[#a&1* 4 ;; ; Cr)BC CC C HAaCB)GH HH H ' "E  23 fQi%++u{{!; << < fQi // / fQQ' 77 7  ! qQ   s 9IIctd}|s tdtddd}tdd}t t d||zz |dt f|d kDd jsJy) NrxryrSrrUrTrrVr[r)rLrIrr r!rr r)rxrSrs rdtest_samplingErsc ' "E  23sAqAsD!A S1a4!Q $a!e : D DD Drictddd}tddd}t|d}t||dkD}||cxk(r|k(sJJy)NrRrrUrSTrV)rr%)rRrSABs rd test_givenrsMsAqAsAqA aA aQA ;Q;;;rictdd}tddtj}t dddd}t |ddk(sJt |dtjk(sJt |dt ddk(sJtd\}}}}tdd|}t dd dd}t |||dztd|zd|zd|z ztd|zzd|z dztd |zzk(sJt ||d td |zd z d td|zd z ztd|d z zk(sJy) NrRrVrSrTrXrUr[zx y z lrrrm) r2r/r rr0r.rrr)rRrSrTrrrls rdtest_factorial_momentrs}QAa AsAq!$A Aq !Q && & Aq !QVV ++ + Aq !Xa^ 33 3#JAq!QaAsB1%A Aq !QT*: 1+&!QU ,Q22&367!eaZ Q "6#&# ## # Aq !Q'71'=%= & 1% %b (&)+;Aq+A"+D&E EE EricNtdtd}}t|d|zsJt|d|zrJtdddtddd}}t||sJt|d|zsJt t ||t ||zd\}}t||sJy)NrRrSrVrrUr[)rr&r'rr%rr rRrSrcrs rdtest_dependencers s8SXqA q!A#  AaC  #q! fS!Q/qA q!   Q!  5A;1q5! -FB R  rictdtd}}t|||zsJtt||||zdkD\}}t||sJy)NrRrSrZ)rr'r%rrs rdtest_dependent_finiter sW s8SXqA QA   5A;A *FB R  rictdddtddd}}tdd}tdd}t||z t||z|}t |||t t fdk(sJy) NrRrrUrSrT)realr)rr r$r rr )rRrSrrdenss rdtest_normalityr*sp #q! fS!Q/qAsAsA 1q5"QUA, 'D T!Wq2#rl +q 00 0ricntdd}t|}|jt|k(sJy)NrRrY)rr@doitr$)rRds rd test_Densityr3s. C A A 668wqz !! !ricGddtt}|tdtdjdk(sJjdk(sJt t fdGddttt t fdy) NceZdZdZy) test_NamedArgsMixin..Foo)foobarN)__name__ __module__ __qualname__ _argnamesrrirdFoor9s rirrUrVcjSr)baz)asrdrz%test_NamedArgsMixin..As 155ric eZdZy) test_NamedArgsMixin..BarN)rrrrrirdBarrCs rircNtdtdjSNrUrV)r r)rsrdrz%test_NamedArgsMixin..Fs3qtQqT?#6#6ri)rrAr rrrHAttributeError)rrrs @@rdtest_NamedArgsMixinr8sk!e^! AaD!A$A 55A:: 55A:: >=) e^  >67ricntdddk(sJtddtdk(sJy)Nr[rVr)r$rrrirdtest_density_constantrHs6 71:a=A   71:a=JqM )) )rictddk(sJtdddk(sJtdddk(sJtd}t|dk(sJt|ddk(sJt|ddk(sJy)Nr[rrXrrrU)r#r;r rs rdtest_cmoment_constantrLs A;!   1a=A   1a=A  s A A;!   1b>Q   1a=A  rictdddk(sJtdddk(sJtdddk(sJtd}t|d|dzk(sJy)Nr[rrUrV r)r<r rs rdtest_moment_constantrUs_ !Q<1   !Q<1   !Q<1  s A !Q<1a4  ricZtddk(sJtd}t||k(sJy)Nr[r)r=r rs rdtest_median_constantr\s+ !9>>s A !9>>ric:tddd}|jsJy)NrrrU)ris_realrs rd test_realr assAqA 999rictddttkdk(sJttkDdk(sJtdktkDdk(sJttktkDdk(sJttkdkDdk(sJtdkdk(dk(sJttdttfdy)NrRr[rUrrVctdSrrrrirdrz"test_issue_10052..ns qtric"tdkdSrrrsrdrz"test_issue_10052..osqQ{ri)rr r rHrrs@rdtest_issue_10052rfsCA QV9>> QV9>> QUAF q  QVQV  !! ! QVQU q  QUAF q   :|$ :*+riclddd}td|}t|dk(sJt|dk\dk(sJy)Nr)rrUrRrVr)rr!r )r$rRs rdtest_issue_11934rqs;nGgA Q43;; a!e9>>rictdd}t||k\dk(sJt||kDdk(sJt||dzkDdk(sJy)NrRrXrUr)rr rs rdtest_issue_8129rwsNCA Q!V9>> QU8q== Q1W:??richtddd}tddd}t|dkD|}t|dk|}t||zdkD|}|t||zdkD|k(sJ|ttjtj tj k(sJ|tjk(sJyr)rr rNr rZeroOne)rRrSUVWs rdtest_issue_12237r}ssAqAsAqA !a% A !a% A !a%!)QA !a%!)Q   %affaffaee< << < ;;ricxtddd}tddd}td\}}tdd}tdd }t |rJt ||zrJt ||zrJt t |dz|dzgrJt |sJt |dz|zsJt ||dzzsJt |d kDsJt |d dksJt |sJt ||z|dzsJt t ||dg||ggsJt t |d sJy) NrRrrUrSza, brrVrg?rZr[rX)rrr*r5rErr )rRrSrbrrs rdtest_is_randomrsQsAqAsAqA 6?DAqQ'Ac"A| Q Q Aq!t -.. . Q<< QTAX   QAX   QU   QqTAX   Q<< QUQqT\ "" " Va1YA/0 11 1 R1X  ricttd}t|}td}tddd}tddd}tt dd}t |t k(sJt |t k(sJt |t k(sJt|t|k(sJt|dkDt|dkDk(sJt|t|k(sJt|t|k(sJt||t||k(sJt||t||k(sJy) NrrSrTrVrUrRIr[)rrBrGrFrr)rDr!r7r r8r#r6r:r9)rrRrSrTrrs rdtest_issue_12283r s  AQASA31%A31%A WT1- .B !9  ": !! ! !9  Q4;q> !! ! QU8{1q5) )) ) A;(1+ %% % B<8B< '' ' a z!Q/ // / a z!Q/ // /ricNtdd}tddd}tt|dt ddz k(sJtt|ddk(sJtt |dkD|dkdk(sJtt |dkD|dkDt ddz k(sJy)NrRrYrSrrUrVr[)rrr r r rrrgs rdtest_issue_6810r"s C AsAqA R1X;!A$q&  R1X;!   RAq1u  ! ## # SQA  1Q46 )) )ric ntd\}}td||}tdd}tt ||z d|z | |zzzt ||zd|z | |zzzt ||zzt ||z|dk\||kzftdf|d|f}|jt|sJy)Nzn prkTrrUr) rr/rOrrPrrrrpr")nprr$eqs rdtest_issue_20286r(s 5>DAqaA cT"A YAq1uQ//AqD!a%A269J4J8TUWX>4Y0ZZ[cdegh[iilmqrlrwx|}w}k~CFHLBMNQRTUWXPY ZB ;;qt  riN)sympy.concrete.summationsrsympy.core.basicrsympy.core.containersrsympy.core.functionrrorrr r sympy.core.relationalr sympy.core.singletonr sympy.core.symbolr r(sympy.functions.combinatorial.factorialsrr&sympy.functions.elementary.exponentialrr(sympy.functions.elementary.trigonometricrr'sympy.functions.special.delta_functionsrsympy.integrals.integralsrsympy.logic.boolalgrrsympy.matrices.densersympy.sets.setsrsympy.tensor.indexedr sympy.statsrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=sympy.stats.rvr>r?r@rArBrCrDrErFrGsympy.testing.pytestrHrIrJrKsympy.externalrLrMsympy.stats.frv_typesrNrO$sympy.functions.elementary.piecewiserPrerhrtrvrrrrrrrrrrrrrrrrrrrrrrrr rrrrrr r"r(rrirdr?sX)"'&66$"/Q=?>/)'$( ___LL(#7#:.+2" 5"BK=   )=-" HEEE$ 1" 8 *   ,  (0"*ri