o
    à8Va#  ã                   @   s    d dl mZmZmZmZmZmZmZmZm	Z	m
Z
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 )é    )ÚsymbolsÚMulÚsinÚIntegralÚooÚEqÚSumÚsqrtÚexpÚpiÚDummy)Ú	unchanged)	ÚNormalÚPoissonÚvarianceÚ
CovarianceÚVarianceÚProbabilityÚExpectationÚMomentÚCentralMoment)ÚprobabilityÚexpectationc                  C   s0  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s3J ‚t| |kƒ ¡ t| |kƒksCJ ‚t| dkƒ t¡ ¡ t| dkƒksVJ ‚t| |kƒ t¡ ¡ t| |kƒksiJ ‚t	| ƒ ¡ t
| ƒksuJ ‚t	| ƒ t¡ ¡ t
| ƒks„J ‚t	| d ƒ ¡ t
| d ƒks”J ‚t	||  ƒj||  fks¢J ‚t	||  ƒ ¡ |t	| ƒ ks²J ‚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sFJ ‚t	| | ƒ ¡ t	| ƒt	|ƒ ksYJ ‚t	| | | |  ƒ ¡ t	| d ƒt	|d ƒ kstJ ‚t	| | | |  ƒ ¡  ¡ dks‡J ‚t	| | d	d ¡ dks–J ‚t	| t	|ƒ ƒ ¡ dks¥J ‚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s1J ‚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s¬J ‚t|ƒj|fks·J ‚t|ƒ ¡ dksÂJ ‚t| ƒ ¡ t| ƒ t¡ ¡   krÜt| ƒksßJ ‚ 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s-J ‚t| | ƒ ¡ t| ƒt|ƒ dt| |ƒ  ksGJ ‚t| | | | ƒ ¡ t| ƒt|ƒ t|ƒ t|ƒ dt| |ƒ  dt| |ƒ  dt| |ƒ  dt||ƒ  dt||ƒ  dt||ƒ  ksJ ‚t| d ƒ ¡ t| d ƒks¡J ‚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sJ ‚t|| ƒ ¡ dks*J ‚t| |ƒj| |fks7J ‚tt| |ƒƒtksCJ ‚t||  d |ƒ ¡ |t| |ƒ ksXJ ‚t| | ƒj| | fkseJ ‚t| | ƒ ¡ t| ƒkssJ ‚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sIJ ‚t| | d ƒ ¡ t| | d ƒks\J ‚t| t| ƒƒ ¡ tt| ƒ| ƒksoJ ‚t| d t| ƒ| ƒ ¡ tt| ƒ| | d ƒksŠJ ‚t|| ƒ ¡ dks–J ‚d S )NÚXé   é   ÚYé   ÚZÚWÚxT©Úrealzy, w, zr   iôÿÿÿ)Zevaluateé   F)Z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© r.   úM/usr/lib/python3/dist-packages/sympy/stats/tests/test_symbolic_probability.pyÚtest_literal_probability   sÚ   

  &&  PD"Hÿÿ&6&.$ÿ**2$ÿ
ÿ
ÿ$ÿ
ÿ
ÿ 
ÿ
ÿ: &44ÿÿÿþþþ".".*6DFÿÿ^ÿÿ&&6r0   c            
      C   s$  t dddƒ} t dddƒ}tddƒ}tddƒ}tdƒ\}}}}t|ƒ t¡d	ks)J ‚t| ƒ t¡t| d ƒt| ƒd  ks>J ‚t| |d
 t¡t| d |ƒt| |ƒd  ksWJ ‚t| |ƒt| d ƒt| ƒd  ksjJ ‚t| | ƒ t¡t| | d ƒt| | ƒd  ks…J ‚t| | ƒ t¡t| d |d  ƒt| | ƒd  ks¢J ‚t|| ƒ t¡| t| ƒ t||  ƒ ks¹J ‚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s,J ‚t| ƒ t¡t	|d | |t
 t
fƒt	|| |t
 t
fƒd  ksPJ ‚t|ƒ t¡t|d |	 |d	t
fƒt||	 |d	t
fƒd  ksrJ ‚t|| ƒ t¡| t	|tt| |ƒƒ |t
 t
fƒ t	|| tt| |ƒƒ |t
 t
fƒ ks¢J ‚t| ƒ t¡t| ƒ t	¡ks³J ‚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   )Z	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pzr.   r.   r/   Útest_probability_rewrite[   s>   

*2&6:.0:*(HDH
ÿ""4$ÿr1   c                  C   sŒ  t ddd} t dddd}t dƒ}td| |ƒ}t|dd	ƒ}| t¡t|d	 d ƒks,J ‚| t¡t|d	 d tt||ƒƒ |t tfƒksGJ ‚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s·J ‚td	dƒ}| ¡ dksÄJ ‚d S )NÚmuTr!   Úsigma©r"   Zpositiver    r   r   r   Úké   r   é   é   é    é   r#   )r   r   r   r&   r   r   r   r   r   r   r	   r
   r   Údummy_eqr'   )r2   r3   r    r   ÚMr5   Úexprir.   r.   r/   Útest_symbolic_Moment„   sJ   "

ÿÿÿ
ÿÿ
ÿ,
ÿÿÿ
ÿ
ÿ
ÿ
r>   c               
   C   s  t ddd} t dddd}t dƒ}td| |ƒ}t|dƒ}| t¡t|t|ƒ d ƒks-J ‚| t¡t|t|tdƒ |t tfƒ d tt||ƒƒ |t tfƒksSJ ‚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¡ |¡s«J ‚| ¡  ¡ d|d  ks¹J ‚tddƒ}| ¡ dksÆJ ‚d S )Nr2   Tr!   r3   r4   r    r   r7   r5   r   é   r#   i5  )r   r   r   r&   r   r   r   r   r   r   r	   r
   r   r;   r'   Zsimplifyr   )r2   r3   r    r   ZCMr5   r=   r.   r.   r/   Útest_symbolic_CentralMoment—   sZ   
"
ÿÿÿ

ÿÿÿ
ÿÿ
ÿÿÿÿ
ÿÿþ
þ
r@   N)Zsympyr   r   r   r   r   r   r   r	   r
   r   r   Zsympy.core.exprr   Zsympy.statsr   r   r   r   r   r   r   r   r   Zsympy.stats.rvr   r   r0   r1   r>   r@   r.   r.   r.   r/   Ú<module>   s   4 ,S)