
    sg                     D    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y)    )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   }|j                   ||      |t        ||| dz  |      z        }t        |j                               d	k(  rJ  y )
Nzomega rlT)integeru         r   )	r   r   ranger   r   subsr   r	   doit)omegarr   r   nschreqresults          O/var/www/html/venv/lib/python3.12/site-packages/sympy/physics/tests/test_sho.pytest_sho_R_nlr      s    y!HE1T"AA 1X ,1q!$$Q&1a!e9qAv*>Qhq!tmAo+&(,Q5(9+:;<Q4*@ @QqT1T!Qa%;#;<&!+++	,    c                      t        d      \  } }}t        t        | ||      d| z  |z   t        dd      z   |z  z
        dk(  sJ y )Nzn l hwr      r   )r   r	   r   r   )r   r   hws      r   test_energyr      sJ    x HAq"DArNacAgA&>%BBCq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 ( ,
,Ir   