U
    	Ç-eï  ã                   @   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òt
‚qòd S )Né   é   é   é   )r   r   r   r   r   )Únur   r   r   r   r   r
   r   r   r   ÚAssertionError)ÚPsiÚn© r   ú`/var/www/html/Darija-Ai-Train/env/lib/python3.8/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t‚qd S )Nr   r   ©Úranger	   r   r   r   r   ©r   Úir   r   r   Ú	test_norm   s    r!   c              	   C   s`   t | d ƒD ]N}t |d | d ƒD ]6}tt|tddƒt|tddƒ tt tfƒdks"t‚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t‚qd S )Nr   )r   r   r   r   r   ZHalfr   r   r   r   r   Útest_energies(   s    r$   é
   c                 C   sR   t dƒ}t| d ƒD ]8}tt| d ƒt| d |ƒ ƒt|t| |ƒ ƒkst‚qd S )NÚalphar   )r   r   r
   r   r   r   )r   r&   r    r   r   r   Útest_coherent_state-   s    r'   N)r   )r   )r   )r%   ) Zsympy.core.numbersr   r   r   Zsympy.core.singletonr   Zsympy.core.symbolr   Z&sympy.functions.elementary.exponentialr   Z(sympy.functions.elementary.miscellaneousr   Zsympy.integrals.integralsr	   Zsympy.simplify.simplifyr
   Z	sympy.abcr   r   r   Zsympy.physics.qho_1dr   r   r   Zsympy.physics.quantum.constantsr   r   r   r!   r#   r$   r'   r   r   r   r   Ú<module>   s   


