o
    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sJ qyd S )N            )r   r   r   r   r   )
nur   r   r   r   r   r
   r   r   r   )Psin r   U/var/www/html/ai/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     


