
    g                     0   d Z ddl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mZmZmZ dd
lmZmZ 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"m#Z#m$Z$m%Z% ddl&m'Z'  ed          \  Z(Z)Z*Z+Z,Z-Z.Z/ ed          \  Z0Z1Z2Z3d Z4d Z5d Z6dS )zTests for cartesian.py    )Ipi)S)symbols)exp)sqrt)
DiracDelta)Interval)qapply	representL2Dagger)
Commutatorhbar)XOpYOpZOpPxOpXYZPxXKetXBraPxKetPxBraPositionKet3DPositionBra3D)DifferentialOperatorzx,y,z,x_1,x_2,x_3,y_1,z_1zpx py px_1 px_2c                  b   t           j        t          t          t          j        t          j                            k    sJ t          t           t                    	                                t          t          z  k    sJ t          t           t          t                    z            t          t          t                    z  k    sJ t          t                                                    t           k    sJ t!          t                                                    t          k    sJ t#          t          t$                              t          t                    z  	                                t'          t          t$          z
            k    sJ t)          t*                    t          t                    z  	                                t-          t           t          z  t*          z  t          z            t/          dt0          z  t          z            z  k    sJ t3          t          t                              t'          t          t4          z
            k    sJ t3          t!          t                              t'          t           t4          z             k    sJ t!          t                    j        t          k    sJ t3          t9                      t                      z            t          t'          t          t:          z
            z  k    sJ t3          t9                      t                      z  t!          d          z            t          t'          t          t<          z
            z  t'          t4          t$          z
            z  k    sJ t3          t!          d          t                      z            t'          t          t$          z
            k    sJ t3          t                      t!                      z            t'          t          t:          z
            t'          t4          t          z
            z  k    sJ t3          t9                      t>                    } | t          t          z  t'          t@          tB          z
            z  tE          t@                    z  k    sJ | t3          t9                      t?                                k    sJ | t3          t9                      tF                    k    sJ | t3          t9                      tG                                k    sJ t3          t9                      tG                      z  tF                    t          t          z  t'          t*          tB          z
            z  tE          t*                    z  k    sJ d S )N   ybasis)$r   hilbert_spacer   r
   r   NegativeInfinityInfinityr   r   doitr   r   r   r   x
dual_classr   r   r"   r	   r   pxr   r   r   r   x_1positionr   x_2x_3r   px_1px_2r   r   )rep_ps    f/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/physics/quantum/tests/test_cartesian.pytest_xr4      s   ?b!*<aj!I!IJJJJJJa!!##qv----!DGG)$q''	))))774''''774''''477OODGG#))++z!a%/@/@@@@@"IId1gg##%%QBqDGDL$qtDy//)* * * *T!WWAG!4!44444T!WWQBH!5!5555577q    SUU466\""a
1s7(;(;&;;;;;SUU466\$s))+,,	*QW

jq1112 2 2 2T#YYtvv%&&*QU*;*;;;;;tvv $QW--
370C0CCD D D D cee4(((EDF:dTk2223G3M3MMMMMMIcee4662222222Icee51111111Icee5773333333SUU577]%000Qz"t)$$$%9"%=%==> > > > > >    c                  "   t           j        t          t          t          j        t          j                            k    sJ t          t           t          t                    z            t          t          t                    z  k    sJ t          t                    
                                t          k    sJ t          t                    
                                t          k    sJ t          t          t                              t          t                    z                                  t!          t          t          z
            k    sJ t#          t                    t          t                    z                                  t%          t&          t          z  t          z  t(          z            t+          dt,          z  t(          z            z  k    sJ t/          t          t                              t!          t          t0          z
            k    sJ t/          t3                      t4                    } | t(           t&          z  t!          t6          t8          z
            z  t;          t6                    z  k    sJ | t/          t3                      t5                                k    sJ | t/          t3                      t<                    k    sJ | t/          t3                      t=                                k    sJ t/          t3                      t=                      z  t<                    t(           t&          z  t!          t          t8          z
            z  t;          t                    z  k    sJ t/          t#          d          t3                      z  t=                      z  t<                    t(           t&          z  t!          t          t>          z
            z  t;          t                    z  k    sJ d S )Nr!   r#   r"   ) r   r%   r   r
   r   r&   r'   r   r   r+   r*   r   r)   r   pyr(   r	   r   r   r   r   r   r   r   r0   r   r   r,   r.   r   r   r"   )rep_xs    r3   test_pr9   4   s   r(1+=qz"J"JKKKKKK"U2YY,2eBii<////99!!U****88  E))))599eBii'--//:b2g3F3FFFFFGGE"II##%%AaCF4Kad4i() ) ) )U2YY:b4i#8#88888dffC(((ETE!GJsSy1112Fs2K2KKKKKKIdffCEE2222222IdffD1111111IdffDFF3333333TVVDFF]$///	a
1s7###$8$;$;;< < < <T#YYtvv%dff,D999	a
1q5!!!"6q"9"99: : : : : :r5   c                     t           j        t          t          t          j        t          j                            k    sJ t          j        t          t          t          j        t          j                            k    sJ t          t          t          t                    } t          t          | z            t          | z  k    sJ t          t           | z            t          | z  k    sJ t          t          | z            t          | z  k    sJ t          t          t           z  | z            t          t          z  | z  k    sJ t          t          t           z  t          z  | z            t          t          z  t          z  | z  k    sJ t          t           t          z  | z            t          t          z  | z  k    sJ t                      | k    sJ t                      t           k    sJ t                      t          k    sJ t          j                    t"          k    sJ t#          j                    t          k    sJ t          t$          t&          t(                    }t+          |          | z                                  t/          t          t$          z
            t/          t          t&          z
            z  t/          t          t(          z
            z  k    sJ | j        t          k    sJ | j        t          k    sJ | j        t          k    sJ |j        t$          k    sJ |j        t&          k    sJ |j        t(          k    sJ d S )N)r   r%   r   r
   r   r&   r'   r   r   r)   r"   zr   r   r   r   r*   r   r,   y_1z_1r   r(   r	   
position_x
position_y
position_z)test_ket	other_kets     r3   
test_3dposrC   J   s   ?b!*<aj!I!IJJJJJJ?b!*<aj!I!IJJJJJJQ1%%H!H*8++++!H*8++++!H*8++++!A#h,1Q3x<////!A#a%.!!QqSU8^3333!A#h,1Q3x<////??h&&&&55A::::55A::::#%%6666#%%6666c3,,I9h&,,..1s7Jq3w///
1s70C0CCD D D D !####!####!####3&&&&3&&&&3&&&&&&r5   N)7__doc__sympy.core.numbersr   r   sympy.core.singletonr   sympy.core.symbolr   &sympy.functions.elementary.exponentialr   (sympy.functions.elementary.miscellaneousr   'sympy.functions.special.delta_functionsr	   sympy.sets.setsr
   sympy.physics.quantumr   r   r   r   r   r   sympy.physics.quantum.cartesianr   r   r   r   r   r   r   r   r   r   r   r   r   r   sympy.physics.quantum.operatorr   r)   r"   r;   r,   r.   r/   r<   r=   r+   r7   r0   r1   r4   r9   rC    r5   r3   <module>rP      s     & & & & & & & & " " " " " " % % % % % % 6 6 6 6 6 6 9 9 9 9 9 9 > > > > > > $ $ $ $ $ $ ? ? ? ? ? ? ? ? ? ? ? ? 2 2 2 2 2 2 2 2                                @ ? ? ? ? ?#*7+F#G#G  1ac3SW.// Bd> > >:: : :,' ' ' ' 'r5   