
    g                         d dl mZmZ d dl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mZmZ d dlmZ d	 ZddZddZddZdS )    )wavefunctionenergy)Ipi)exp)sqrt)	integrate)simplify)mxr)hbarc                     dt          dt          z            z  dt          dt          z            z  t          t          t          z            z  dt          dt          z            z  t          dt          z  t          z            z  dt          dt          z            z  t          dt          z  t          z            z  d} | D ]3}t          t          |t                    | |         z
            dk    sJ 4d S )N         )r   r   r   r   r   )r   r   r   r   r   r
   r   )Psins     Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/physics/tests/test_pring.pytest_wavefunctionr      s    d1r6llNd1r6llNc!a%jj(d1r6llNc!a%!)nn,d1r6llNc!a%!)nn,	 C  : :Q**SV34499999: :    r   c                     t          | dz             D ]R}t          t          |t                    t          | t                    z  t          ddt          z  f          dk    sJ Sd S Nr   r   r   ranger	   r   r   r   r   is     r   	test_normr      s    1q5\\ K KAqb!!4!44q!QVnF FIJK K K K KK Kr   c           	          t          | dz             D ]j}t          |dz   | dz             D ]Q}t          t          |t                    t          |t                    z  t          ddt          z  f          dk    sJ Rkd S r   r   )r   r   js      r   test_orthogonalityr!      s    1q5\\ N NqsAaC 	N 	NAQ""\!Q%7%77!QBI ILMN N N N N	NN Nr   c                     t          | dz             D ]W}t          t          |t          t                    |dz  t
          dz  z  dt          z  t          dz  z  z  z
            dk    sJ Xd S )Nr   r   r   )r   r
   r   r   r   r   r   s     r   test_energyr#   %   s    1Q3ZZ H H1aOO1tQw1q51a4<@AC CFGH H H H HH Hr   N)r   )sympy.physics.pringr   r   sympy.core.numbersr   r   &sympy.functions.elementary.exponentialr   (sympy.functions.elementary.miscellaneousr   sympy.integrals.integralsr	   sympy.simplify.simplifyr
   	sympy.abcr   r   r   sympy.physics.quantum.constantsr   r   r   r!   r#    r   r   <module>r-      s   4 4 4 4 4 4 4 4 & & & & & & & & 6 6 6 6 6 6 9 9 9 9 9 9 / / / / / / , , , , , ,           0 0 0 0 0 0: : :K K K KN N N NH H H H H Hr   