
    sg                         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y)    )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 ]&  }t        t        |t              | |   z
        dk(  r&J  y )N         )r   r   r   r   r   )r   r   r   r   r   r
   r   )Psins     Q/var/www/html/venv/lib/python3.12/site-packages/sympy/physics/tests/test_pring.pytest_wavefunctionr      s    d1r6lNd1r6lNc!a%j(d1r6lNc!a%!)n,d1r6lNc!a%!)n,	C  :Q*SV34999:    c                     t        | dz         D ]A  }t        t        |t              t        | t              z  t        ddt        z  f      dk(  rAJ  y Nr   r   r   ranger	   r   r   r   r   is     r   	test_normr      s]    1q5\ KAqb!!44q!QVnFIJK 	K KKr   c           	          t        | dz         D ]W  }t        |dz   | dz         D ]@  }t        t        |t              t        |t              z  t        ddt        z  f      dk(  r@J  Y y r   r   )r   r   js      r   test_orthogonalityr!      sz    1q5\ NqsAaC 	NAQ"\!Q%77!QBILMN N N	NNr   c                     t        | dz         D ]I  }t        t        |t        t              |dz  t
        dz  z  dt        z  t        dz  z  z  z
        dk(  rIJ  y )Nr   r   r   )r   r
   r   r   r   r   r   s     r   test_energyr#   %   sf    1Q3Z H1aO1tQw1q51a4<@ACFG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-      s7    4 & 6 9 / ,   0:KNHr   