U
    ZÊ¹jx  ã                   @   sH   d dl mZ d dlmZ d dlmZmZmZ d dlmZmZ dd„ Z	dS )é    )Úsymbols)Údynamicsymbols)ÚReferenceFrameÚPointÚParticle)ÚLagrangesMethodÚ
Lagrangianc                  C   sP  t dƒ\} }t ddƒ\}}tdƒ\}}}tdƒ}tdƒ}| d| |j ¡}	|	 |||j ¡ | d||j ¡}
|
 |||j ¡ td|	|ƒ}|| d  d |_td	|
|ƒ}|| | d  d |_|	| | |j f|
| | |j fg}t	|||ƒ}t
|| |f||d
�}| ¡  |jd }| |¡|k�s.t‚|jd }| |¡|k�sLt‚d S )Nzq1, q2é   zm, k, nuÚNÚOÚP1ÚpP1é   ÚpP2)Ú	forcelistÚframer   )r   r   r   r   Z	locatenewÚxZset_velr   Zpotential_energyr   r   Zform_lagranges_equationsZeomÚdiffÚAssertionError)Úq1Úq2Zq1dZq2dÚmÚkÚnur
   r   r   ZP2r   r   r   ZlagZl_methodZeq1Zeq2© r   úk/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/physics/mechanics/tests/test_lagrange2.pyÚtest_lagrange_2forces	   s.    ÿ

r   N)
Zsympyr   Zsympy.physics.mechanicsr   r   r   r   r   r   r   r   r   r   r   Ú<module>   s   