U
    ZÊ¹j¶$  ã                   @   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sft‚t| |kƒ ¡ t| |kƒks†t‚t| dkƒ t¡ 	¡ t| dkƒks¬t‚t| |kƒ t¡ 	¡ t| |kƒksÒt‚t
| ƒ ¡ t| ƒksêt‚t
| ƒ t¡ 	¡ t| ƒk�s
t‚t
| d ƒ ¡ t| d ƒk�s,t‚t
||  ƒj||  fk�sJt‚t
||  ƒ ¡ |t
| ƒ k�slt‚t
d|  d|  ||  |  ƒ ¡ dt
| ƒ dt
|ƒ  |t
| | ƒ  k�s¾t‚t
d|  d|  ||  |  ƒjd|  d|  ||  |  fk�st‚t
t| ƒƒt
t| ƒƒ ¡ k�s&t‚t
d| t| ƒ | || d   ||  |  ƒ ¡ d| t
t| ƒ| ƒ |t
| d ƒ  |t
| | ƒ  k�s˜t‚t
| | ƒ ¡ t
| ƒt
|ƒ k�s¾t‚t
| | | |  ƒ ¡ t
| d ƒt
|d ƒ k�sôt‚t
| | | |  ƒ ¡  	¡ dk�st‚t
| | d	d� 	¡ dk�s8t‚t
| t
|ƒ ƒ 	¡ dk�sVt‚t
| t
|ƒ ƒj	dd�dt
t
|ƒƒ k�s„t‚t
| t
|t
d|  ƒ ƒ ƒj	dd�dt
t
|t
d|  ƒ ƒƒ k�sÊt‚t
| t
|t
d|  ƒ ƒ ƒ 	¡ dk�sôt‚t
t
d|  ƒƒ 	¡ dk�st‚t
t
d|  ƒƒj	dd�t
d|  ƒk�s<t‚t
dt
d|  ƒ ƒj	dd�dt
d|  ƒ k�snt‚t
| | d ƒ ¡ dt
| |d  ƒ dt
| d | ƒ  t
| d ƒ t
|d ƒ k�sÈt‚t
| | d ƒ ¡ dt
| |d  ƒ dt
| d | ƒ  t
| d ƒ t
|d ƒ k�s"t‚t
| | d ƒ ¡ dt
| | ƒ t
| d ƒ t
|d ƒ k�sdt‚t|ƒj|fk�szt‚t|ƒ ¡ dk�s�t‚t| ƒ ¡ t| ƒ t¡ 	¡   k�rÂt| ƒk�sÈn t‚t| | ƒj| | fk�sæt‚t| | ƒ ¡ t| ƒk�st‚t| | ƒjt| |ƒfk�s$t‚tt| | ƒƒtk�s>t‚t||  ƒ ¡ |d t| ƒ k�sdt‚t| | ƒ ¡ t| ƒt|ƒ dt| |ƒ  k�s˜t‚t| | | | ƒ ¡ t| ƒt|ƒ t|ƒ t|ƒ dt| |ƒ  dt| |ƒ  dt| |ƒ  dt||ƒ  dt||ƒ  dt||ƒ  k�s*t‚t| d ƒ ¡ t| d ƒk�sLt‚tt| d ƒ�s`t‚t|| d  ƒ ¡ |d t| d ƒ k�sŽt‚tt| ƒƒjt| ƒfk�s¬t‚tt| ƒƒ ¡ tt| ƒƒk�sÎt‚t|t| ƒ ƒ ¡ |d tt| ƒƒ k�süt‚t||ƒj||fk�st‚t||ƒ ¡ dk�s.t‚t| |ƒ ¡ dk�sFt‚t|| ƒ ¡ dk�s^t‚t| |ƒj| |fk�sxt‚tt| |ƒƒtk�s�t‚t||  d |ƒ ¡ |t| |ƒ k�sºt‚t| | ƒj| | fk�sÔt‚t| | ƒ ¡ t| ƒk�sðt‚t||  d || d ƒ ¡ || t| |ƒ k�	s&t‚t| |ƒt|| ƒk�	s@t‚t| | || ƒ ¡ t|| ƒt||ƒ t| |ƒ t||ƒ k�	s„t‚t||  ||  || ||  ƒ ¡ || t|| ƒ || t||ƒ  || t| |ƒ  || t||ƒ  k�	søt‚t|| d  |t|ƒ  || |d  ||  ƒ ¡ || t|| d ƒ || tt|ƒ|ƒ  || t||d  | d ƒ  || t||d  t|ƒƒ  k�
sœt‚t| | d ƒ ¡ t| | d ƒk�
sÂt‚t| t| ƒƒ ¡ tt| ƒ| ƒk�
sèt‚t| d t| ƒ| ƒ ¡ tt| ƒ| | d ƒk�st‚t|| ƒ ¡ dk�s6t‚d S )NÚXé   é   ÚYé   ÚZÚWÚxT©Úrealzy, w, zr   iôÿÿÿ)Úevaluateé   F)Údeepé	   éþÿÿÿ)r   r   r   r   Zevaluate_integralr   ÚAssertionErrorÚrewriter   Údoitr   r   ÚargsÚexpandr   r   r   r   Útyper   r   )r   r   r   r   r    ÚyÚwÚz© r1   új/var/www/html/bwave/python/venv/lib/python3.8/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r3   c            
      C   s.  t dddƒ} t dddƒ}tddƒ}tddƒ}tdƒ\}}}}t|ƒ t¡d	ksRt‚t| ƒ t¡t| d ƒt| ƒd  ks|t‚t| |d
� t¡t| d |ƒt| |ƒd  ks®t‚t| |ƒt| d ƒt| ƒd  ksÔt‚t| | ƒ t¡t| | d ƒt| | ƒd  k�st‚t| | ƒ t¡t| d |d  ƒt| | ƒd  k�sHt‚t|| ƒ t¡| t| ƒ t||  ƒ k�sxt‚t| |ƒ t¡t| | ƒt| ƒt|ƒ  k�sªt‚t| ||d
� t¡t| | |ƒt| |ƒt||ƒ  k�sæt‚tdƒ\}}}tt	| |ƒƒ}tt	||ƒƒ}	t| ƒ t¡t
|| |t tfƒk�s:t‚t|ƒ t¡t||	 |d	tfƒk�sbt‚t| ƒ t¡t
|d | |t tfƒt
|| |t tfƒd  k�sªt‚t|ƒ t¡t|d |	 |d	tfƒt||	 |d	tfƒd  k�sît‚t|| ƒ t¡| t
|tt	| |ƒƒ |t tfƒ t
|| tt	| |ƒƒ |t tfƒ k�sNt‚t| ƒ t¡t| ƒ t
¡k�spt‚t| ƒ t¡t| ƒ t
¡k�s’t‚t|| ƒ t¡d	k�s¬t‚t|| ƒ t
¡d	k�sÆt‚t| |d
� t¡t
|d tt	| |ƒ|ƒ |t tfƒt
|tt	| |ƒ|ƒ |t tfƒd  k�s*t‚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/   r0   ZpxZpzr1   r1   r2   Útest_probability_rewritec   s>    

*2&8<02<*(HDHÿ
""4$ÿr5   c                  C   sŽ  t ddd�} t ddd�}t dƒ}td| |ƒ}t|dd	ƒ}| t¡t|d	 d ƒksVt‚| t¡t|d	 d tt||ƒƒ |t	 t	fƒksŒt‚t
d
ƒ}ttd	ƒ|d	 d  t||  d	  d	|d	   ƒ d	ttƒ |  |t	 t	fƒ}| t¡ |¡søt‚| ¡ | d d| d   d| d	  |d	   d| d	   d|  |d	   d|   d|d   d|d	   d k�snt‚td	dƒ}| ¡ dk�sŠt‚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	   r   Údummy_eqr*   )r6   r7   r    r   ÚMr9   Úexprir1   r1   r2   Útest_symbolic_MomentŒ   sJ    "
ÿ
ÿÿ
ÿÿ
ÿ,
ÿÿÿ
ÿ
ÿÿ
rB   c               
   C   s”  t ddd�} t ddd�}t dƒ}td| |ƒ}t|dƒ}| t¡t|t|ƒ d ƒksXt‚| t¡t|t|tdƒ |t tfƒ d tt	||ƒƒ |t tfƒks¤t‚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¡ |¡�sVt‚| ¡  ¡ d|d  k�stt‚tddƒ}| ¡ dk�s�t‚d S )Nr6   Tr!   r7   r8   r    r   r;   r9   r   é   r$   i5  )r   r   r   r)   r   r(   r   r   r   r   r   r
   r	   r   r?   r*   Úsimplifyr   )r6   r7   r    r   ZCMr9   rA   r1   r1   r2   Útest_symbolic_CentralMomentŸ   sZ    
"
ÿÿÿ
ÿ
ÿÿ
ÿÿ
ÿÿÿÿ
ÿÿþ
þ
rE   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   r3   r5   rB   rE   r1   r1   r1   r2   Ú<module>   s   ,S)