U
    9%e                     @   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/Darija-Ai-API/env/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   