a
    Ÿ¬<bµ  ã                   @   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(J ‚q(d S )
Nzomega rÚlT)ÚintegerÚué   é   é   r   )	r   r   Úranger   r   Úsubsr   r   Zdoit)ÚomegaÚrr	   r   ÚnZschreqÚresult© r   úl/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/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@J ‚d S )Nzn l hwr   é   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   