a
    lº”h¶$  ã                   @   sà   d dl mZ d dlmZ d dlmZmZ d dlmZ d dl	m
Z
mZ d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZmZmZmZmZmZmZmZmZ d dl m!Z!m"Z" dd„ Z#dd„ Z$dd„ Z%dd„ Z&dS )é    )ÚSum)ÚMul)ÚooÚpi)ÚEq)ÚDummyÚsymbols)Úexp)Úsqrt)Úsin)ÚIntegral)Ú	unchanged)	ÚNormalÚPoissonÚvarianceÚ
CovarianceÚVarianceÚProbabilityÚExpectationÚMomentÚCentralMoment)ÚprobabilityÚexpectationc                  C   s:  t dddƒ} t dddƒ}tddƒ}tddƒ}tdd	d
}tdƒ\}}}t| dkƒ ¡ t| dkƒksfJ ‚t| |kƒ ¡ t| |kƒks†J ‚t| dkƒ t¡ ¡ t| dkƒks¬J ‚t| |kƒ t¡ ¡ t| |kƒksÒJ ‚t	| ƒ ¡ t
| ƒksêJ ‚t	| ƒ t¡ ¡ t
| ƒks
J ‚t	| d ƒ ¡ t
| d ƒks,J ‚t	||  ƒj||  fksJJ ‚t	||  ƒ ¡ |t	| ƒ kslJ ‚t	d|  d|  ||  |  ƒ ¡ dt	| ƒ dt	|ƒ  |t	| | ƒ  ks¾J ‚t	d|  d|  ||  |  ƒjd|  d|  ||  |  fksJ ‚t	t| ƒƒt	t| ƒƒ ¡ ks&J ‚t	d| t| ƒ | || d   ||  |  ƒ ¡ d| t	t| ƒ| ƒ |t	| d ƒ  |t	| | ƒ  ks˜J ‚t	| | ƒ ¡ t	| ƒt	|ƒ ks¾J ‚t	| | | |  ƒ ¡ t	| d ƒt	|d ƒ ksôJ ‚t	| | | |  ƒ ¡  ¡ dksJ ‚t	| | d	d ¡ dks8J ‚t	| t	|ƒ ƒ ¡ dksVJ ‚t	| t	|ƒ ƒjdddt	t	|ƒƒ ks„J ‚t	| t	|t	d|  ƒ ƒ ƒjdddt	t	|t	d|  ƒ ƒƒ ksÊJ ‚t	| t	|t	d|  ƒ ƒ ƒ ¡ dksôJ ‚t	t	d|  ƒƒ ¡ dksJ ‚t	t	d|  ƒƒjddt	d|  ƒks<J ‚t	dt	d|  ƒ ƒjdddt	d|  ƒ ksnJ ‚t	| | d ƒ ¡ dt	| |d  ƒ dt	| d | ƒ  t	| d ƒ t	|d ƒ ksÈJ ‚t	| | d ƒ ¡ dt	| |d  ƒ dt	| d | ƒ  t	| d ƒ t	|d ƒ ks"J ‚t	| | d ƒ ¡ dt	| | ƒ t	| d ƒ t	|d ƒ ksdJ ‚t|ƒj|fkszJ ‚t|ƒ ¡ dksJ ‚t| ƒ ¡ t| ƒ t¡ ¡   krÂt| ƒksÈn J ‚t| | ƒj| | fksæJ ‚t| | ƒ ¡ t| ƒksJ ‚t| | ƒjt| |ƒfks$J ‚tt| | ƒƒtks>J ‚t||  ƒ ¡ |d t| ƒ ksdJ ‚t| | ƒ ¡ t| ƒt|ƒ dt| |ƒ  ks˜J ‚t| | | | ƒ ¡ t| ƒt|ƒ t|ƒ t|ƒ dt| |ƒ  dt| |ƒ  dt| |ƒ  dt||ƒ  dt||ƒ  dt||ƒ  ks*J ‚t| d ƒ ¡ t| d ƒksLJ ‚tt| d ƒs`J ‚t|| d  ƒ ¡ |d t| d ƒ ksŽJ ‚tt| ƒƒjt| ƒfks¬J ‚tt| ƒƒ ¡ tt| ƒƒksÎJ ‚t|t| ƒ ƒ ¡ |d tt| ƒƒ ksüJ ‚t||ƒj||fksJ ‚t||ƒ ¡ dks.J ‚t| |ƒ ¡ dksFJ ‚t|| ƒ ¡ dks^J ‚t| |ƒj| |fksxJ ‚tt| |ƒƒtksJ ‚t||  d |ƒ ¡ |t| |ƒ ksºJ ‚t| | ƒj| | fksÔJ ‚t| | ƒ ¡ t| ƒksðJ ‚t||  d || d ƒ ¡ || t| |ƒ k	s&J ‚t| |ƒt|| ƒk	s@J ‚t| | || ƒ ¡ t|| ƒt||ƒ t| |ƒ t||ƒ k	s„J ‚t||  ||  || ||  ƒ ¡ || t|| ƒ || t||ƒ  || t| |ƒ  || t||ƒ  k	søJ ‚t|| d  |t|ƒ  || |d  ||  ƒ ¡ || t|| d ƒ || tt|ƒ|ƒ  || t||d  | d ƒ  || t||d  t|ƒƒ  k
sœJ ‚t| | d ƒ ¡ t| | d ƒk
sÂJ ‚t| t| ƒƒ ¡ tt| ƒ| ƒk
sèJ ‚t| d t| ƒ| ƒ ¡ tt| ƒ| | d ƒksJ ‚t|| ƒ ¡ dks6J ‚d S )NÚXé   é   ÚYé   ÚZÚWÚxT©Úrealzy, w, zr   iôÿÿÿ)Úevaluateé   F)Údeepé	   éþÿÿÿ)r   r   r   r   Zevaluate_integralr   Úrewriter   Údoitr   r   ÚargsÚexpandr   r   r   r   Útyper   r   )r   r   r   r   r    ÚyÚwÚz© r0   úY/var/www/auris/lib/python3.9/site-packages/sympy/stats/tests/test_symbolic_probability.pyÚtest_literal_probability   sÚ    

  && ""RF"Hÿÿ&6&.$ÿ**2$ÿ
ÿ
ÿ$ÿ
ÿ
ÿ 
ÿ
ÿ8 &44ÿÿÿþþþ".".*6DFÿÿ^ÿÿ&&6r2   c            
      C   s.  t dddƒ} t dddƒ}tddƒ}tddƒ}tdƒ\}}}}t|ƒ t¡d	ksRJ ‚t| ƒ t¡t| d ƒt| ƒd  ks|J ‚t| |d
 t¡t| d |ƒt| |ƒd  ks®J ‚t| |ƒt| d ƒt| ƒd  ksÔJ ‚t| | ƒ t¡t| | d ƒt| | ƒd  ksJ ‚t| | ƒ t¡t| d |d  ƒt| | ƒd  ksHJ ‚t|| ƒ t¡| t| ƒ t||  ƒ ksxJ ‚t| |ƒ t¡t| | ƒt| ƒt|ƒ  ksªJ ‚t| ||d
 t¡t| | |ƒt| |ƒt||ƒ  ksæJ ‚tdƒ\}}}tt| |ƒƒ}tt||ƒƒ}	t| ƒ t¡t	|| |t
 t
fƒks:J ‚t|ƒ t¡t||	 |d	t
fƒksbJ ‚t| ƒ t¡t	|d | |t
 t
fƒt	|| |t
 t
fƒd  ksªJ ‚t|ƒ t¡t|d |	 |d	t
fƒt||	 |d	t
fƒd  ksîJ ‚t|| ƒ t¡| t	|tt| |ƒƒ |t
 t
fƒ t	|| tt| |ƒƒ |t
 t
fƒ ksNJ ‚t| ƒ t¡t| ƒ t	¡kspJ ‚t| ƒ t¡t| ƒ t	¡ks’J ‚t|| ƒ t¡d	ks¬J ‚t|| ƒ t	¡d	ksÆJ ‚t| |d
 t¡t	|d tt| |ƒ|ƒ |t
 t
fƒt	|tt| |ƒ|ƒ |t
 t
fƒd  ks*J ‚d S )Nr   r   r   r   r   r   r   z
x, y, w, zr   )Ú	conditionzW, x, z)r   r   r   r   r(   r   r   r   r   r   r   r   )
r   r   r   r   r    r-   r.   r/   ZpxZpzr0   r0   r1   Útest_probability_rewritec   s>    

*2&8<02<*(HDHÿ
""4$ÿr4   c                  C   sŽ  t ddd} t ddd}t dƒ}td| |ƒ}t|dd	ƒ}| t¡t|d	 d ƒksVJ ‚| t¡t|d	 d tt||ƒƒ |t tfƒksŒJ ‚t	d
ƒ}tt
d	ƒ|d	 d  t||  d	  d	|d	   ƒ d	t
tƒ |  |t tfƒ}| t¡ |¡søJ ‚| ¡ | d d| d   d| d	  |d	   d| d	   d|  |d	   d|   d|d   d|d	   d ksnJ ‚td	dƒ}| ¡ dksŠJ ‚d S )NÚmuTr!   Úsigma©Zpositiver    r   r   r   Úké   r   é   é   é    é   r$   )r   r   r   r(   r   r   r   r   r   r   r
   r	   r   Údummy_eqr)   )r5   r6   r    r   ÚMr8   Úexprir0   r0   r1   Útest_symbolic_MomentŒ   sJ    "
ÿ
ÿÿ
ÿÿ
ÿ,
ÿÿÿ
ÿ
ÿÿ
rA   c               
   C   s”  t ddd} t ddd}t dƒ}td| |ƒ}t|dƒ}| t¡t|t|ƒ d ƒksXJ ‚| t¡t|t|tdƒ |t tfƒ d tt||ƒƒ |t tfƒks¤J ‚t	d	ƒ}tt
d
ƒ|tt
d
ƒ| t||  d
  d
|d
   ƒ d
t
tƒ |  |t tfƒ d  t||  d
  d
|d
   ƒ d
t
tƒ |  |t tfƒ}| t¡ |¡sVJ ‚| ¡  ¡ d|d  kstJ ‚tddƒ}| ¡ dksJ ‚d S )Nr5   Tr!   r6   r7   r    r   r:   r8   r   é   r$   i5  )r   r   r   r(   r   r   r   r   r   r   r
   r	   r   r>   r)   Zsimplifyr   )r5   r6   r    r   ZCMr8   r@   r0   r0   r1   Útest_symbolic_CentralMomentŸ   sZ    
"
ÿÿÿ
ÿ
ÿÿ
ÿÿ
ÿÿÿÿ
ÿÿþ
þ
rC   N)'Zsympy.concrete.summationsr   Zsympy.core.mulr   Zsympy.core.numbersr   r   Zsympy.core.relationalr   Zsympy.core.symbolr   r   Z&sympy.functions.elementary.exponentialr	   Z(sympy.functions.elementary.miscellaneousr
   Z(sympy.functions.elementary.trigonometricr   Zsympy.integrals.integralsr   Zsympy.core.exprr   Zsympy.statsr   r   r   r   r   r   r   r   r   Zsympy.stats.rvr   r   r2   r4   rA   rC   r0   r0   r0   r1   Ú<module>   s   ,S)