
    g                     F    d dl mZmZmZmZ d dlmZmZ d dlm	Z	 d Z
d ZdS )    )symbolsRationalFunctiondiff)R_nlE_nl)simplifyc                      t          d          \  } }t          dd          }t          d          }t          d          D ]}t           ||          |d           dz  ||dz   z  d|dz  z  z  | dz  |dz  z  dz  z   t	          |||           z
   ||          z  z   }|                     ||          |t          ||| dz  |          z            }t          |                                          d	k    sJ d S )
Nzomega rlT)integeru         r   )	r   r   ranger   r   subsr   r	   doit)omegarr   r   nschreqresults          X/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/physics/tests/test_sho.pytest_sho_R_nlr      s"   y!!HE1T"""AA 1XX , ,1q!$$$Q&1a!e9qAv*>Qhq!tmAo+&(,Q5(9(9+:;<1Q44*@ @QQqTT1T!Qa%;%;#;<<&&!+++++	, ,    c                      t          d          \  } }}t          t          | ||          d| z  |z   t          dd          z   |z  z
            dk    sJ d S )Nzn l hwr      r   )r   r	   r   r   )r   r   hws      r   test_energyr      s]    x  HAq"DArNNacAgA&>%BBCCqHHHHHHr   N)
sympy.corer   r   r   r   sympy.physics.shor   r   sympy.simplify.simplifyr	   r   r    r   r   <module>r$      s    8 8 8 8 8 8 8 8 8 8 8 8 ( ( ( ( ( ( ( ( , , , , , ,
, 
, 
,I I I I Ir   