U
    ZÊ¹jµ  ã                   @   sH   d dl mZmZmZmZ d dlmZmZ d dlm	Z	 dd„ Z
dd„ ZdS )	é    )ÚsymbolsÚRationalÚFunctionÚdiff)ÚR_nlÚE_nl)Úsimplifyc               
   C   s¾   t dƒ\} }t ddd�}tdƒ}tdƒD ]�}t||ƒ|dƒ d ||d  d|d   | d |d  d  t||| ƒ ||ƒ  }| ||ƒ|t||| d |ƒ ¡}t| ¡ ƒd	ks(t	‚q(d S )
Nzomega rÚlT)ÚintegerÚué   é   é   r   )
r   r   Úranger   r   Úsubsr   r   ZdoitÚAssertionError)ÚomegaÚrr	   r   ÚnZschreqÚresult© r   ú[/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/physics/tests/test_sho.pyÚtest_sho_R_nl   s    *ÿ
ÿÿ"r   c                  C   sD   t dƒ\} }}tt| ||ƒd|  | tddƒ |  ƒdks@t‚d S )Nzn l hwr   é   r   )r   r   r   r   r   )r   r	   Zhwr   r   r   Útest_energy   s    r   N)Z
sympy.corer   r   r   r   Zsympy.physics.shor   r   Zsympy.simplify.simplifyr   r   r   r   r   r   r   Ú<module>   s   