
    g              	       x   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+d S )!    )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n J |                                 t          k    sJ t          t                    } | t          t                    cxk    rt          t                    k    sn 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 |                                 t          t          t                    z  t          z   k    sJ t          | t                    sJ | j        t          dfk    sJ t          t          t          z            } |                                 | k    sJ t          t          t          z  t          z  t          z            } | j        d         j        t          t          t          t          fk    sJ |                                 t          t          t          t          z  t          z            z  k    sJ t          t          t"          z   t          t"          z
  j        z            } |                                 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             } |                                 t          t          t                    z  t&          t          t"                    z  z   k    sJ t          t                                                    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                                           d          t          ||z             k    sJ t          t          ||z                                                       t+          ddgddgg          k    sJ d S )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      h/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/stats/tests/test_symbolic_multivariate.pytest_multivariate_expectationrE   )   sk   q>>D;q>>9999%6q%9%9999999;;==Aq>>D;q>>9999%6q%9%9999999:!Q9>>>>9>>>>d-.....qsQwD$QqS1W------;;==A/222Q66666d-.....:!Qr"uD;;==D    r"uRx{##D9Q<RR 00000;;==B0Br:::::::AAy())D;;==-ae447H137O7OO!!AC%(()+<QqsU+C+CD D D D D qsQqSy!!D;;==A/222Q7H7K7K5KK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[A[[[[[{27++,,1133vBx"b>R7S7SSSSSSS    c                     t          t          d            t          t                    } | t          t                    cxk    rt	          t                    k    sn J |                                 t          t          t                    k    sJ t          t          j                  } | t          t          j                  cxk    rt	          t          j                  k    sn J |                                 t          t          t                    k    sJ t          t                    } | t          t                    cxk    rt	          t                    k    sn 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 |                                 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            } |                                 t          t          z  t	          t                    z  t          j        z  t          j        z  k    sJ t          t           t"          z            } |                                 | k    sJ t          t$          t           z  t&          z  t"          z            } | j        d         j        t$          t           t&          t"          fk    sJ |                                 | k    sJ t          t          t          z  t          t*          z  z             } |                                 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 d S )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   sS   
:**+++A;;D8A;;3333."3"3333333;;==Jq!,,,,,,AC==D8AC==7777N13$7$7777777;;==Jq!,,,,,,A;;D8A;;3333."3"3333333:!Q9>>>>9>>>>dN+++++AaC==D>!A#&&&&&&;;==AnQ///33333dN+++++:!QAaCE??D;;==AaCq 1 11!#5ac99999BrE??D;;==D    BrE"HRK  D9Q<RR 00000;;==D    AaC!A#ID;;==AaC 5a ; ;;AC?nQ'''+,./q0A0A.A!#.EF F F F F F FrF   c                  h   t          t          d            t          t          d            t          t          j        t
          j                  } | j        dk    sJ |                                 t          dd          k    sJ t          t          t
                    } | t          t          t
                    cxk    rt          t          t
                    k    sn 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
                    } |                                 t          t          t                    k    sJ t          t          t                     } t          | t                    sJ |                                 | k    sJ t          t          t                    } t          | t                    sJ |                                 t#          t                    k    sJ t          t          t           z   t$                    } t          | t                    sJ |                                 t          t          t$                    t          t           t$                    z   k    sJ t          t          t          z  t                     } t          | t                    sJ |                                 t          t          t          t                     z  k    sJ t          t          t&          t           z            } t          | t                    s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 |                                 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 |                                 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 d S )Nc                  @    t          t          t          j                  S rI   )r   r   r   r>   rJ   rF   rD   rK   z3test_multivariate_crosscovariance.<locals>.<lambda>~   s    z!QS11 rF   c                  6    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   }   sD   
:11222
://000 ac13D:;;==Jq!,,,,,,aD:a##BBBB'<Q'B'BBBBBBB;;==Jq!,,,,,,:!Q9>>>>9>>>>d122222acAgq!!D;;==Jq!,,,,,,aDd122222;;==D    aDd122222;;==N1------a!eQDd122222;;==1!Q77:OPQST:U:UUUUUUac1Dd122222;;==A3Aq9999999a1Dd122222;;==1!Q77;;;;;acAgqs1uqy))Dd122222;;==A3Aq999!;;;;;acAaCi!mQSUQSU]Q%677Dd122222;;==A3Aq999!;a@UVWYZ@[@[>[\]>]]
!!Q''
'
)*,-.CAq.I.I,I!,KL L L 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[      s   A A A A A A A A A A A A ' ' ' ' ' ' % % % % % % 0 0 0 0 0 0 ' ' ' ' ' ' ; ; ; ; ; ; 9 9 9 9 9 9 - - - - - -C C C C C C C C C C wu~~1LaLaLaLaLaLa\$1\$1sAq!!sAq!!sAq!!sAq!!sAq!!a##	!Q		V&&Aq
!
!!"QK  
,T ,T ,T^#F #F #FJ/L /L /L /L /LrF   