
    sg              	       
   d dl mZmZ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 d dlmZ d d	lmZmZmZ  ed
      \  ZZ edee      Z edee      Z edee      Z edee      Z eded      Z eded      Z eddd      Z eddd      Z  eded      Z! eded      Z" eded      Z# eded      Z$ edee      Z% eddd      Z& edd d      Z' ede eddd      z  ge'd gg      Z(d Z)d Z*d Z+y )!    )ExpectationNormalVariance
Covariance)raises)symbols)
ShapeError)Matrix)MatrixSymbol)
ZeroMatrix)RandomMatrixSymbol)ExpectationMatrixVarianceMatrixCrossCovarianceMatrixzj,kABCDa   bA2   B2XYZWRX2normalnormal2c            	         t        t              } | t        t              cxk(  rt        t              k(  sJ  J | j                         t        k(  sJ t        t              } | t        t              cxk(  rt        t              k(  sJ  J | j
                  t        dfk(  sJ | j                  t        k(  sJ | j                  dk(  sJ t        | t              sJ t        t        t        z  t        z         } | t        t        t        z  t        z         k(  sJ | j                         t        t        t              z  t        z   k(  sJ t        | t              sJ | j
                  t        dfk(  sJ t        t        t        z        } | j                         | k(  sJ t        t        t        z  t        z  t        z        } | j                   d   j                   t        t        t        t        fk(  sJ | j                         t        t        t        t        z  t        z        z  k(  sJ t        t        t"        z   t        t"        z
  j$                  z        } | j                         t        t        t        j$                  z        t        t        t"        j$                  z        z
  t        t"        t        j$                  z        z   t        t"        t"        j$                  z        z
  k(  sJ t        t        t        z  t&        t"        z  z         } | j                         t        t        t              z  t&        t        t"              z  z   k(  sJ t        t              j)                         t+        ddt,        z  gddgg      k(  sJ t+        t/        ddd      t/        ddd      gt/        dd	d      t/        d
dd      gg      }t+        t/        ddd      t/        ddd      gt/        ddd      t/        ddd      gg      }t        t        ||z               j)                  d      t        ||z         k(  sJ t        t        ||z               j)                         t+        ddgddgg      k(  sJ y )Nr   r   r   N11   N12   N21   N22   M11M12M21   M22   F)deep         )r   r   r   expandr   shapekrowscols
isinstancer   r   m1r    r   r   argsr   Tr   doitr
   jr   )exprx1x2s      _/var/www/html/venv/lib/python3.12/site-packages/sympy/stats/tests/test_symbolic_multivariate.pytest_multivariate_expectationrE   )   s   q>D;q>9%6q%999999;;=Aq>D;q>9%6q%999999::!Q99>>99>>d-...qsQwD$QqS1W----;;=A/22Q6666d-...::!Qr"uD;;=D   r"uRx{#D99Q<RR 0000;;=B0Br:::::AAyy()D;;=-ae47H1337OO!!ACC%()+<QqssU+CD D D D qsQqSy!D;;=A/22Q7H7K5KKKKKr?!Va1X1v,>%????	E2q6%Q/0E2q6%Q/0 
B 
E1a&1-.E1a&1-. 
B
 {27+,11u1=ARSUXZSZA[[[[{27+,113vBx"b>R7SSSS    c                     t        t        d        t        t              } | t        t              cxk(  rt	        t              k(  sJ  J | j                         t        t        t              k(  sJ t        t        j                        } | t        t        j                        cxk(  rt	        t        j                        k(  sJ  J | j                         t        t        t              k(  sJ t        t              } | t        t              cxk(  rt	        t              k(  sJ  J | j                  t        t        fk(  sJ | j                  t        k(  sJ | j                  t        k(  sJ t        | t              sJ t        t        t        z        } | t	        t        t        z        k(  sJ | j                         t        t	        t              z  t        j                  z  k(  sJ t        | t              sJ | j                  t        t        fk(  sJ t        t        t        z  t        z        } | j                         t        t        z  t	        t              z  t        j                  z  t        j                  z  k(  sJ t        t         t"        z        } | j                         | k(  sJ t        t$        t         z  t&        z  t"        z        } | j(                  d   j(                  t$        t         t&        t"        fk(  sJ | j                         | k(  sJ t        t        t        z  t        t*        z  z         } | j                         dt        z  t-        t        t*              z  t        j                  z  t        t	        t              z  t        j                  z  z   t        t	        t*              z  t        j                  z  z   k(  sJ y )Nc                       t        t              S N)r   r    rF   rD   <lambda>z,test_multivariate_variance.<locals>.<lambda>Y   s
    x{ rF   r   r   )r   r	   r   r   r   r6   r   r8   r>   r   r7   r9   r:   r;   r   r   r<   r    r   r   r=   r   r   rA   s    rD   test_multivariate_variancerM   X   s   
:*+A;D8A;3."333333;;=Jq!,,,,ACC=D8ACC=7N133$777777;;=Jq!,,,,A;D8A;3."333333::!Q99>>99>>dN+++AaC=D>!A#&&&&;;=AnQ//3333dN+++::!QAaCE?D;;=AaCq 11!##5acc9999BrE?D;;=D   BrE"HRK D99Q<RR 0000;;=D   AaC!A#ID;;=AaC 5a ;;ACC?nQ''+,./q0A.A!##.EF F F FrF   c                  	   t        t        d        t        t        d        t        t        j                  t
        j                        } | j                  dk(  sJ | j                         t        dd      k(  sJ t        t        t
              } | t        t        t
              cxk(  rt        t        t
              k(  sJ  J | j                         t        t        t              k(  sJ | j                  t        t        fk(  sJ | j                  t        k(  sJ | j                  t        k(  sJ t        | t              sJ t        t        t        z  t        z   t
              } | j                         t        t        t              k(  sJ t        t        t               } t        | t              sJ | j                         | k(  sJ t        t        t              } t        | t              sJ | j                         t#        t              k(  sJ t        t        t         z   t$              } t        | t              sJ | j                         t        t        t$              t        t         t$              z   k(  sJ t        t        t        z  t               } t        | t              sJ | j                         t        t        t        t               z  k(  sJ t        t        t&        t         z        } t        | t              sJ | j                         t        t        t               t&        j                  z  k(  sJ t        t        t        z  t        z   t&        j                  t         z  t
        z         } t        | t              sJ | j                         t        t        t        t               z  t&        z  k(  sJ t        t        t        z  t&        t         z  z   t        z   t(        j                  t$        z  t*        j                  t,        z  z   t
        z         } t        | t              sJ | j                         t        t        t        t,              z  t*        z  t        t        t        t$              z  t(        z  z   t&        t        t         t,              z  t*        z  z   t&        t        t         t$              z  t(        z  z   k(  sJ y )Nc                  >    t        t        t        j                        S rI   )r   r   r   r>   rJ   rF   rD   rK   z3test_multivariate_crosscovariance.<locals>.<lambda>~   s    z!QSS1 rF   c                  *    t        t        t              S rI   )r   r   r   rJ   rF   rD   rK   z3test_multivariate_crosscovariance.<locals>.<lambda>   s    z!Q/ rF   )r   r   r   )r   r	   r   r   r>   r   r7   r6   r   r   r8   r9   r:   r;   r   r   r   r   r   r   r   r   r   rL   s    rD   !test_multivariate_crosscovariancerQ   }   sv   
:12
:/0 acc133D::;;=Jq!,,,,aD:a#B'<Q'BBBBBB;;=Jq!,,,,::!Q99>>99>>d1222acAgq!D;;=Jq!,,,,aDd1222;;=D   aDd1222;;=N1----a!eQDd1222;;=1!Q7:OPQST:UUUUUac1Dd1222;;=A3Aq99999a1Dd1222;;=1!Q7;;;;acAgqss1uqy)Dd1222;;=A3Aq99!;;;;acAaCi!mQSSUQSSU]Q%67Dd1222;;=A3Aq99!;a@UVWYZ@[>[\]>]]
!!Q'
'
)*,-.CAq.I,I!,KL L L LrF   N),sympy.statsr   r   r   r   sympy.testing.pytestr   sympy.core.symbolr   sympy.matrices.exceptionsr	   sympy.matrices.denser
   "sympy.matrices.expressions.matexprr   "sympy.matrices.expressions.specialr   sympy.stats.rvr   -sympy.stats.symbolic_multivariate_probabilityr   r   r   r@   r8   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r<   rE   rM   rQ   rJ   rF   rD   <module>r[      sM   A A ' % 0 ' ; 9 -C C u~1aaaaaa$1$1sAq!sAq!sAq!sAq!sAq!a#	!Q	&Aq
!!"QK 
,T^#FJ/LrF   