a
    kº”hï  ã                   @   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 ‚qòd S )Né   é   é   é   )r   r   r   r   r   )
Únur   r   r   r   r   r
   r   r   r   )ÚPsiÚn© r   úM/var/www/auris/lib/python3.9/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 ]N}t |d | d ƒD ]6}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   ZHalfr   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J ‚qd S )NÚalphar   )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   


