o 8Va@sHddlmZmZmZmZddlmZmZddlm Z ddZ ddZ dS) )symbolsRationalFunctiondiff)R_nlE_nl)simplifyc Cstd\}}tddd}td}tdD]H}t|||d d||dd|d|d|ddt|||||}||||t|||d|}t|d ks\JqdS) Nzomega rlT)Zintegerur) rrrangerrZsubsrrZdoit)Zomegarr r nZschreqresultr>/usr/lib/python3/dist-packages/sympy/physics/tests/test_sho.py test_sho_R_nls   * "rcCsDtd\}}}tt|||d||tdd|dks JdS)Nzn l hwr r)rrrr)rr Zhwrrr test_energys6rN) Z sympy.corerrrrZsympy.physics.shorrZsympyrrrrrrrs