
    ЦiL                     b   S r SSK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  SSK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!J"r"J#r#J$r$J%r%J&r&  \!" S5      r'\"" S5      r(\#" S5      r)\#" S5      r*\#" S5      r+\#" S5      r,\$" S5      r-\$" S5      r.\%" S5      r/\&" S5      r0\" S5      r1\" S5      r2\" S5      r3\" S5      r4\" SSS/0S 9r5\" \'\0SS!S"9r6\" \(\0SS!S"9r7\" \0\0SS!S"9r8\" \/\0SS!S"9r9\" \,\0SS!S"9r:\" \.\0SS!S"9r;S# r<S$ r=S% r>S& r?S' r@g())zTests for sho1d.py    )IInteger)S)Symbol)sqrt)Dagger)hbar)
Commutator)qapply)InnerProduct)XPx)KroneckerDelta)ComplexSpace)	represent)import_module)skip)	RaisingOp
LoweringOpSHOKetSHOBraHamiltonianNumberOpak      bHNomegam   numpyscipyfromlistsparse)import_kwargssympybasisndimformatc                  B   [        [        5      [        :X  d   e[        [        [        5      R	                  5       [        S5      :X  d   e[        [        [        5      R	                  5       [        S5      [        -  :X  d   e[        [        [        -  5      [        [        R                  S-   5      [        [        R                  S-   5      -  R                  5       :X  d   e[        [        [        -  5      [        [        R                  S-   5      [        [        R                  S-   5      -  R                  5       :X  d   e[        [        [        -  5      [        [        R                  S-   5      [        [        R                  S-   5      -  R                  5       :X  d   e[        R                  S5      R	                  5       [        S5      [        [        S5      [         -  ["        -  [$        -  5      -  [        S5      [&        -  [(        -  ["        [$        -  [*        -  -   -  :X  d   e[        R,                  [/        [0        R2                  5      :X  d   e[5        [6        S-
  5       H#  n [8        U S-   U 4   [        U S-   5      :X  a  M#   e   [:        (       d  [=        S5        [?        [        [        SSS9n[5        [6        S-
  5       H'  n XS-   U 4   [A        [        U S-   5      5      :X  a  M'   e   [:        (       d  [=        S5        [B        (       d  [=        S	5        [?        [        [        SS
SS9n[5        [6        S-
  5       H'  n X S-   U 4   [A        [        U S-   5      5      :X  a  M'   e   URD                  S:X  d   eURD                  S:X  d   eg )Nr   xp   znumpy not installed.r#   r$   r*   zscipy not installed.zscipy.sparselil)r+   r,   r-   spmatrixfloat64)#r   adr   r
   doitr   r    r   r   r   nr   expandkzkfrewriter	   r"   r!   r   r   r   hilbert_spacer   r   Infinityranger,   ad_rep_sympynpr   r   floatr%   dtype)iad_rep_numpyad_rep_scipys      e/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/sympy/physics/quantum/tests/test_sho1d.pytest_RaisingOprG   /   s   ":??b!!!#wr{222b!!!#wr{2~555"Q$<DqM&q/9AACCCC"R%=T"$$(^F244!8,<<DDFFFF"R%=T"$$(^F244!8,<<DDFFFF::d  "	DD*501	1GBKM"4Dqwqy4PQR R R|AJJ77774!8_AE!G$QU333  2#$RqqAL4!8_E!G$d1q5k(::::  2#$5#$RqqRWXL4!8_E!G$d1q5k(::::  ******    c                  D   [        [        5      [        :X  d   e[        [        [        5      R	                  5       [        S5      :X  d   e[        [        [        5      R	                  5       [        :X  d   e[        [        [        -  5      [        [        R                  5      [        [        R                  [        S5      -
  5      -  R                  5       :X  d   e[        [        [        -  5      [        S5      :X  d   e[        [        [        -  5      [        [        R                  5      [        [        R                  [        S5      -
  5      -  R                  5       :X  d   e[        R                  S5      R	                  5       [        S5      [        [        S5      [         -  ["        -  [$        -  5      -  [&        [(        -  ["        [$        -  [*        -  -   -  :X  d   e[-        [.        S-
  5       H"  n [0        X S-   4   [        U S-   5      :X  a  M"   e   g )Nr   r   r0   r1   )r   r   r5   r
   r6   r   r    r   r   r   r7   r   r8   r9   r:   r;   r	   r"   r!   r   r   r   r>   r,   a_reprC   s    rF   test_LoweringOprL   O   sw   !9??a!!#wqz111a  "a'''!A#;49VACC
N%;;CCEEEE!B$<71:%%%!B$<DJvbdd71:o'>>FFHHHH99T?!	DD*501	1AbD1U7194DEF F F4!8_Q1uW~a!e,,, rH   c                  N   [        [        [        5      R                  5       [        :X  d   e[        [        [        5      R                  5       [        S5      [        -  :X  d   e[        [        [        5      R                  5       [        S5      :X  d   e[        [        [        -  5      [        R                  [        -  R                  5       :X  d   e[        R                  S5      R                  5       [        [        -  :X  d   e[        R                  S5      R                  5       [        S5      [        S5      [        -  [        -  [        -  -  [        S-  [        [        -  [         -  S-  -   -  [        S5      [        S5      -  -
  :X  d   e[        R                  S5      R                  5       [        [        [        -  -  [        S5      [        S5      -  -
  :X  d   e[#        [$        5       H  n [&        X 4   U :X  a  M   e   [&        [(        [*        -  :X  d   eg )Nr/   r   r   r0   r   r1   r   )r
   r    r5   r6   r   r   r   r   r   r7   r8   r;   r"   r	   r!   r   r   r>   r,   N_repr?   rJ   rK   s    rF   test_NumberOprO   [   s   a!!#r)))a  "gbk!m333a  "gaj000!A#;133q5..****99S> BqD(((99T?!gaj'!*Q,t2CE2I&J
A5Q& ")!*WQZ"7&8 8 8 899S> AtEzNWQZ
5J$JJJJ4[QSzQ L&&&&rH   c                     [        [        [        5      R                  5       [	        S5      :X  d   e[        [        [        -  5      [        [        -  [        R                  [	        S5      [	        S5      -  -   -  [        -  R                  5       :X  d   e[        R                  S5      R                  5       [        [        -  [        [        -  [	        S5      [	        S5      -  -   -  :X  d   e[        R                  S5      R                  5       [	        S5      [	        S5      [        -  -  [        S-  [        [        -  [         -  S-  -   -  :X  d   e[        R                  S5      R                  5       [        [        -  [        [	        S5      [	        S5      -  -   -  :X  d   e[#        [$        5       H9  n [&        X 4   [        [        -  U [	        S5      [	        S5      -  -   -  :X  a  M9   e   g )Nr   r   r1   r   r0   r    )r
   r   r    r6   r   r   r   r	   r!   r7   r8   r;   r5   r   r"   r   r   r>   r,   H_reprK   s    rF   test_HamiltonianrR   h   sm   a  "gaj000!A#;DJgaj.C(CDaGOOQQQQ99S> DJ1wqz'!*7L0L$MMMM99T?!	WQZ\	"RUagai!^%;<= = =99S> DJGAJwqz4I0I$JJJJ4[QSzT%ZWQZ
-B)BCCCC rH   c                  &   [        S5      R                  5       [        :X  d   e[        S5      R                  5       [         :X  d   e[        [        [
        5      R                  5       [        [
        R                  [        R                  5      :X  d   e[
        R                  [        [        R                  5      :X  d   e[        [        R                  S4   [        S5      :X  d   e[         S["        R                  4   [        S5      :X  d   eg )Nr   r   r   r   )r   
dual_classr   r   r   r   r6   r   r7   r<   r   r   r=   k3_repk3r   b3_repb3 rH   rF   test_SHOKetrZ   r   s    #;!!#v---#;!!#v---!!!#~acc133'??????l1::6666"$$'?gaj(((!RTT'?gaj(((rH   N)A__doc__sympy.core.numbersr   r   sympy.core.singletonr   sympy.core.symbolr   (sympy.functions.elementary.miscellaneousr   sympy.physics.quantumr   sympy.physics.quantum.constantsr	   r
   sympy.physics.quantum.qapplyr   "sympy.physics.quantum.innerproductr   sympy.physics.quantum.cartesianr   r   (sympy.functions.special.tensor_functionsr   sympy.physics.quantum.hilbertr   sympy.physics.quantum.representr   sympy.externalr   sympy.testing.pytestr   sympy.physics.quantum.sho1dr   r   r   r   r   r   r5   r   r   r9   r:   rV   r   rX   r   r    r!   r"   r,   r@   r%   r?   rJ   rN   rQ   rU   rW   rG   rL   rO   rR   rZ   rY   rH   rF   <module>rk      sO    + " $ 9 ( 0 , / ; 1 C 6 5 ( %? ? s^sO
3KAYAYAY
3KAYSMw
3Kqz7gj8*-EF11W=!11W5!11W5!11W5	2QQw	7	2QQw	7+@
-'D)rH   