ó
”¼\c           @   s_   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   Z	 d S(   i’’’’(   t   symbols(   t   dynamicsymbols(   t   ReferenceFramet   Pointt   Particle(   t   LagrangesMethodt
   Lagrangianc          C   sĻ  t  d  }  \ } } t  d d  } \ } } t d  \ } } } t d  }	 t d  }
 |
 j d | |	 j  } | j |	 | |	 j  |
 j d | |	 j  } | j |	 | |	 j  t d | |  } | | d d | _ t d	 | |  } | | | d d | _ | | | |	 j f | | | |	 j f g } t	 |	 | |  } t
 | | | f d
 | d |	 } | j   | j d } | j |  | k s£t  | j d } | j |  | k sĖt  d  S(   Ns   q1, q2i   s   m, k, nut   Nt   Ot   P1t   pP1i   t   pP2t	   forcelistt   framei    (   R   R    R   R   t	   locatenewt   xt   set_velR   t   potential_energyR   R   t   form_lagranges_equationst   eomt   difft   AssertionError(   t   qst   q1t   q2t   qdst   q1dt   q2dt   mt   kt   nuR   R   R	   t   P2R
   R   R   t   lagt   l_methodt   eq1t   eq2(    (    sK   lib/python2.7/site-packages/sympy/physics/mechanics/tests/test_lagrange2.pyt   test_lagrange_2forces	   s,    !
N(
   t   sympy.core.backendR    t   sympy.physics.mechanicsR   R   R   R   R   R   R$   (    (    (    sK   lib/python2.7/site-packages/sympy/physics/mechanics/tests/test_lagrange2.pyt   <module>   s   