U
    L?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st
qd S )N            )r   r   r   r   r   )nur   r   r   r   r   r
   r   r   r   AssertionError)Psin r   Q/var/www/html/venv/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   


