
    g                         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dS )    )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 ]?}t          t          |t          t          t                    | |         z
            dk    sJ @d S )N            )r   r   r   r   r   )
nur   r   r   r   r	   r   r   r   r   )Psins     [/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/physics/tests/test_qho_1d.pytest_wavefunctionr      s   rEHQNN"S"q!tQ%7%77rEHQNN"T!B$ZZ/!3c2#1*a-6H6HHrEHQNN"a"fq!tma&78a@3sQPQTzST}CUCUUrEHQNN"T"Q$ZZ/1r6AqD=1q53HICQSPSVWYZVZPZ\]P]L^L^^	 C  = =aAu--A6771<<<<<= =    r   c           	          t          | dz             D ]C}t          t          |t          dd          dz  t          t           t          f          dk    sJ Dd S )Nr   r   ranger
   r   r   r   r   is     r   	test_normr#      sb    1q5\\ B Bq!Q**A-B3|<<AAAAAB Br   c                    t          | dz             D ]q}t          |dz   | dz             D ]X}t          t          |t          dd          t          |t          dd          z  t          t           t          f          dk    sJ Yrd S )Nr   r   r   )r   r"   js      r   test_orthogonalityr&       s    1q5\\ H Hq1ua!e$$ 	H 	HAaAq!!%1a"3"33a"b\C CFGH H H H H	HH Hr   c                     t          | dz             D ]:}t          |t                    t          t          z  |t          j        z   z  k    sJ ;d S )Nr   )r    r   r   r   r   Halfr!   s     r   test_energiesr)   (   sP    1q5\\ < <1e}}uAF
 ;;;;;;< <r   
   c           	          t          d          }t          | dz             D ]Y}t          t          | dz             t	          | dz   |          z            t          |t	          | |          z            k    sJ Zd S )Nalphar   )r   r    r   r	   r   )r   r,   r"   s      r   test_coherent_stater-   -   s     7OOE1q5\\ r rQUnQUE&B&BBCCxPUXfghjoXpXpPpGqGqqqqqqr rr   N)r   )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      sb   1 1 1 1 1 1 1 1 1 1 " " " " " " $ $ $ $ $ $ 6 6 6 6 6 6 9 9 9 9 9 9 / / / / / / , , , , , , ! ! ! ! ! ! ! ! ! ! ; ; ; ; ; ; ; ; ; ; 0 0 0 0 0 0Y= = =B B B BH H H H< < < <
r r r r r rr   