
    \h              	       
   S SK JrJrJrJr  S SKJr  S SKJr  S SK	J
r
  S SKJr  S SKJr  S SKJr  S SKJr  S S	KJrJrJr  \" S
5      u  rr\" S\\5      r\" S\\5      r\" S\\5      r\" S\\5      r\" S\S5      r\" S\S5      r\" SSS5      r\" SSS5      r \" S\S5      r!\" S\S5      r"\" S\S5      r#\" S\S5      r$\" S\\5      r%\" SSS5      r&\" SS S5      r'\" S\\" SSS5      -  /\'S //5      r(S r)S r*S r+g )!    )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            	      	   [        [        5      n U [        [        5      s=:X  a  [        [        5      :X  d   e   eU R                  5       [        :X  d   e[        [        5      n U [        [        5      s=:X  a  [        [        5      :X  d   e   eU R
                  [        S4:X  d   eU R                  [        :X  d   eU R                  S:X  d   e[        U [        5      (       d   e[        [        [        -  [        -   5      n U [        [        [        -  [        -   5      :X  d   eU R                  5       [        [        [        5      -  [        -   :X  d   e[        U [        5      (       d   eU R
                  [        S4:X  d   e[        [        [        -  5      n U R                  5       U :X  d   e[        [        [        -  [        -  [        -  5      n U R                   S   R                   [        [        [        [        4:X  d   eU R                  5       [        [        [        [        -  [        -  5      -  :X  d   e[        [        ["        -   [        ["        -
  R$                  -  5      n U R                  5       [        [        [        R$                  -  5      [        [        ["        R$                  -  5      -
  [        ["        [        R$                  -  5      -   [        ["        ["        R$                  -  5      -
  :X  d   e[        [        [        -  [&        ["        -  -   5      n U R                  5       [        [        [        5      -  [&        [        ["        5      -  -   :X  d   e[        [        5      R)                  5       [+        SS[,        -  /SS//5      :X  d   e[+        [/        SSS5      [/        SSS5      /[/        SS	S5      [/        S
SS5      //5      n[+        [/        SSS5      [/        SSS5      /[/        SSS5      [/        SSS5      //5      n[        [        X-   5      5      R)                  SS9[        X-   5      :X  d   e[        [        X-   5      5      R)                  5       [+        SS/SS//5      :X  d   eg )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      d/var/www/auris/envauris/lib/python3.13/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
 {27+,11u1=ARSUSZA[[[[{27+,113vBx"b>R7SSSS    c                     [        [        S 5        [        [        5      n U [        [        5      s=:X  a  [	        [        5      :X  d   e   eU R                  5       [        [        [        5      :X  d   e[        [        R                  5      n U [        [        R                  5      s=:X  a  [	        [        R                  5      :X  d   e   eU R                  5       [        [        [        5      :X  d   e[        [        5      n U [        [        5      s=:X  a  [	        [        5      :X  d   e   eU R                  [        [        4:X  d   eU R                  [        :X  d   eU R                  [        :X  d   e[        U [        5      (       d   e[        [        [        -  5      n U [	        [        [        -  5      :X  d   eU R                  5       [        [	        [        5      -  [        R                  -  :X  d   e[        U [        5      (       d   eU R                  [        [        4:X  d   e[        [        [        -  [        -  5      n U R                  5       [        [        -  [	        [        5      -  [        R                  -  [        R                  -  :X  d   e[        [         ["        -  5      n U R                  5       U :X  d   e[        [$        [         -  [&        -  ["        -  5      n U R(                  S   R(                  [$        [         [&        ["        4:X  d   eU R                  5       U :X  d   e[        [        [        -  [        [*        -  -   5      n U R                  5       S[        -  [-        [        [*        5      -  [        R                  -  [        [	        [        5      -  [        R                  -  -   [        [	        [*        5      -  [        R                  -  -   :X  d   eg )Nc                       [        [        5      $ N)r   r    rF   rD   <lambda>,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_variancerN   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                   
   [        [        S 5        [        [        S 5        [        [        R                  [
        R                  5      n U R                  S:X  d   eU R                  5       [        SS5      :X  d   e[        [        [
        5      n U [        [        [
        5      s=:X  a  [        [        [
        5      :X  d   e   eU R                  5       [        [        [        5      :X  d   eU R                  [        [        4:X  d   eU R                  [        :X  d   eU R                  [        :X  d   e[        U [        5      (       d   e[        [        [        -  [        -   [
        5      n U R                  5       [        [        [        5      :X  d   e[        [        [         5      n [        U [        5      (       d   eU R                  5       U :X  d   e[        [        [        5      n [        U [        5      (       d   eU R                  5       [#        [        5      :X  d   e[        [        [         -   [$        5      n [        U [        5      (       d   eU R                  5       [        [        [$        5      [        [         [$        5      -   :X  d   e[        [        [        -  [         5      n [        U [        5      (       d   eU R                  5       [        [        [        [         5      -  :X  d   e[        [        [&        [         -  5      n [        U [        5      (       d   eU R                  5       [        [        [         5      [&        R                  -  :X  d   e[        [        [        -  [        -   [&        R                  [         -  [
        -   5      n [        U [        5      (       d   eU R                  5       [        [        [        [         5      -  [&        -  :X  d   e[        [        [        -  [&        [         -  -   [        -   [(        R                  [$        -  [*        R                  [,        -  -   [
        -   5      n [        U [        5      (       d   eU R                  5       [        [        [        [,        5      -  [*        -  [        [        [        [$        5      -  [(        -  -   [&        [        [         [,        5      -  [*        -  -   [&        [        [         [$        5      -  [(        -  -   :X  d   eg )Nc                  >    [        [        [        R                  5      $ rI   )r   r   r   r>   rJ   rF   rD   rK   3test_multivariate_crosscovariance.<locals>.<lambda>~   s    z!QSS1rF   c                  *    [        [        [        5      $ rI   )r   r   r   rJ   rF   rD   rK   rQ      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   rM   s    rD   !test_multivariate_crosscovariancerS   }   s   
: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2acAgq!D;;=Jq!,,,,aDd12222;;=D   aDd12222;;=N1----a!eQDd12222;;=1!Q7:OPQST:UUUUUac1Dd12222;;=A3Aq99999a1Dd12222;;=1!Q7;;;;acAgqss1uqy)Dd12222;;=A3Aq99!;;;;acAaCi!mQSSUQSSU]Q%67Dd12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   rN   rS   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   