
    sg                         d dl 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mZmZ d d	lmZmZmZ d d
lmZ eez  ez  Zd ZddZddZddZddZy)    )Rationaloopi)S)Symbol)exp)sqrt)	integrate)simplify)omegamx)psi_nE_ncoherent_state)hbarc            	         t         t        z  t        dd      z  t        t          t        dz  z  dz        z  t         t        z  t        dd      z  t        dt         z        z  t        z  t        t          t        dz  z  dz        z  t         t        z  t        dd      z  dt         z  t        dz  z  dz
  z  t        d      z  t        t          t        dz  z  dz        z  t         t        z  t        dd      z  t        t         dz        z  dt         z  t        dz  z  dt        z  z
  z  t        t          t        dz  z  dz        z  d} | D ]0  }t        t        |t        t        t              | |   z
        dk(  r0J  y )N            )r   r   r   r   r   )
nur   r   r   r   r	   r   r   r   r   )Psins     R/var/www/html/venv/lib/python3.12/site-packages/sympy/physics/tests/test_qho_1d.pytest_wavefunctionr      sK   rEHQN"S"q!tQ%77rEHQN"T!B$Z/!3c2#1*a-6HHrEHQN"a"fq!tma&78a@3sQPQTzST}CUUrEHQN"T"Q$Z/1r6AqD=1q53HICQSPSVWYZVZPZ\]P]L^^	C  =aAu-A671<<<=    c           	          t        | dz         D ]6  }t        t        |t        dd      dz  t        t         t        f      dk(  r6J  y )Nr   r   ranger
   r   r   r   r   is     r   	test_normr#      sH    1q5\ Bq!Q*A-B3|<AAABr   c                     t        | dz         D ]]  }t        |dz   | dz         D ]F  }t        t        |t        dd      t        |t        dd      z  t        t         t        f      dk(  rFJ  _ y )Nr   r   r   )r   r"   js      r   test_orthogonalityr&       s    1q5\ Hq1ua!e$ 	HAaAq!%1a"33a"b\CFGH H H	HHr   c                     t        | dz         D ]6  }t        |t              t        t        z  |t        j
                  z   z  k(  r6J  y )Nr   )r    r   r   r   r   Halfr!   s     r   test_energiesr)   (   s>    1q5\ <1e}uAFF
 ;;;;<r   c           	          t        d      }t        | dz         D ]D  }t        t        | dz         t	        | dz   |      z        t        |t	        | |      z        k(  rDJ  y )Nalphar   )r   r    r   r	   r   )r   r+   r"   s      r   test_coherent_stater,   -   sf     7OE1q5\ rQUnQUE&BBCxPUXfghjoXpPpGqqqqrr   N)r   )
   ) sympy.core.numbersr   r   r   sympy.core.singletonr   sympy.core.symbolr   &sympy.functions.elementary.exponentialr   (sympy.functions.elementary.miscellaneousr	   sympy.integrals.integralsr
   sympy.simplify.simplifyr   	sympy.abcr   r   r   sympy.physics.qho_1dr   r   r   sympy.physics.quantum.constantsr   r   r   r#   r&   r)   r,    r   r   <module>r9      sV    1 1 " $ 6 9 / , ! ! ; ; 0Y=BH<
rr   