o
    jgï  ã                   @   sÐ   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 e Zdd„ Zddd„Zddd„Zddd„Zddd„ZdS )é    )ÚRationalÚooÚpi)ÚS)ÚSymbol)Úexp)Úsqrt)Ú	integrate)Úsimplify)ÚomegaÚmÚx)Úpsi_nÚE_nÚcoherent_state)Úhbarc                  C   s  t t tddƒ tt  td  d ƒ t t tddƒ tdt  ƒ t tt  td  d ƒ t t tddƒ dt  td  d  tdƒ tt  td  d ƒ t t tddƒ tt d ƒ dt  td  dt   tt  td  d ƒ dœ} | D ]}tt|ttt	ƒ| |  ƒdksŒJ ‚qyd S )Né   é   é   é   )r   r   r   r   r   )
Únur   r   r   r   r   r
   r   r   r   )ÚPsiÚn© r   úW/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/physics/tests/test_qho_1d.pyÚtest_wavefunction   s   &6BJü$ÿr   r   c                 C   s>   t | d ƒD ]}tt|tddƒd tt tfƒdksJ ‚qd S )Nr   r   ©Úranger	   r   r   r   ©r   Úir   r   r   Ú	test_norm   s   *ÿr    c              	   C   s`   t | d ƒD ]'}t |d | d ƒD ]}tt|tddƒt|tddƒ tt tfƒdks,J ‚qqd S )Nr   r   r   )r   r   Újr   r   r   Útest_orthogonality    s   $ÿ
ÿÿÿr"   c                 C   s6   t | d ƒD ]}t|tƒtt |tj  ksJ ‚qd S )Nr   )r   r   r   r   r   ÚHalfr   r   r   r   Útest_energies(   s   "ÿr$   é
   c                 C   sR   t dƒ}t| d ƒD ]}tt| d ƒt| d |ƒ ƒt|t| |ƒ ƒks&J ‚q
d S )NÚalphar   )r   r   r
   r   r   )r   r&   r   r   r   r   Útest_coherent_state-   s   6ÿ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   r   r   Ú<module>   s     


