ó
9­\c           @   s  d  d l  m Z m Z m Z d  d l  m Z d  d l m Z m Z m	 Z	 d  d l
 m Z m Z m Z d  d l m Z d  d l m Z d „  Z d „  Z e d	 ƒ Z e j d
 e e j ƒ Z g  Z g  Z e j e j ƒ e j e j ƒ e j e j ƒ e j e j e j ƒ e j e e j ƒ e j e e j e e j ƒ e j e d e j e j e j ƒ e j e d e e j d e j e e j ƒ e d ƒ Z  e j e j e e  e ƒ ƒ e j d e j ƒ e d ƒ Z! e d ƒ Z" e j e j e j ƒ e j e	 e j e ƒ ƒ e j e d e e j e e  e ƒ ƒ e j ƒ e d ƒ Z# e d ƒ Z$ x" e D] Z e j e e j Bƒ qqWd e j d e j Z% e d ƒ Z& e d ƒ Z' e d ƒ Z( e d ƒ Z) d „  Z* e d „  ƒ Z+ d „  Z, d „  Z- d „  Z. d S(   iÿÿÿÿ(   t   Integralt   latext   Function(   t   pretty(   t
   CoordSys3Dt   Vectort   express(   t   at   bt   c(   t   u_decode(   t   XFAILc         C   s   t  |  d t d t ƒS(   s   ASCII pretty-printingt   use_unicodet	   wrap_line(   t   xprettyt   False(   t   expr(    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyR   
   s    c         C   s   t  |  d t d t ƒS(   s   Unicode pretty-printingR   R   (   R   t   TrueR   (   R   (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   upretty   s    t   Nt   Ci   i   t   fsh         âŽ›   2   âŒ         âŽž    
N_j + âŽœC_x  - âŽ® f(b) dbâŽŸ N_k
      âŽ       âŒ¡        âŽ     s‚   N_j + /         /       \
      |   2    |        |
      |C_x  -  | f(b) db|
      |        |        |
      \       /         / sy   âŽ› 2    âŽž        âŽ›âŒ         âŽž    
âŽa  + bâŽ  N_i  + âŽœâŽ® f(b) dbâŽŸ N_k
                âŽâŒ¡        âŽ     sy   / 2    \ + /  /       \
\a  + b/ N_i| |        |
           | | f(b) db|
           | |        |
           \/         / s            2
3â‹…C_yâ‹…N_x s            2
3*C_y*N_x sg   âŽ› 2    âŽž                                     
âŽa  + bâŽ  (N_i|N_k)  + (3â‹…C_y - 3â‹…c) (N_k|N_k)sY   / 2    \ (N_i|N_k) + (3*C_y - 3*c) (N_k|N_k)
\a  + b/                                    c           C   sT  t  t d ƒ d k s t ‚ t  t d ƒ d k s8 t ‚ t  t d ƒ d k sT t ‚ t  t d ƒ d k sp t ‚ t  t d	 ƒ d
 k sŒ t ‚ t  t d ƒ d k s¨ t ‚ t  t ƒ d k sÀ t ‚ t  t d ƒ d k sÜ t ‚ t  t d ƒ d k sø t ‚ t  t d ƒ d k st ‚ t  t d ƒ d k s0t ‚ t  t d	 ƒ d d k sPt ‚ d  S(   Ni    t   0i   s   N.ii   s   (-1)*N.ii   s	   N.i + N.ji   s&   N.j + (C.x**2 - Integral(f(b), b))*N.ki	   s	   C.k + N.is   3*C.y*N.x**2s	   (N.i|N.k)i   s   a*(N.i|N.k)i   s   a*(N.i|N.k) + (-b)*(N.j|N.k)s   (N.j|N.k) + (C.x**2 - s   Integral(f(b), b))*(N.k|N.k)(   t   strt   vt   AssertionErrort   st   d(    (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   test_str_printingb   s    c           C   s4  t  t d ƒ d k s t ‚ t  t d ƒ d k s8 t ‚ t  t d ƒ d k sT t ‚ t  t d ƒ t k sp t ‚ t  t d ƒ d	 k sŒ t ‚ t  t d
 ƒ t k s¨ t ‚ t  t ƒ t k sÀ t ‚ t  t d ƒ d k sÜ t ‚ t  t d ƒ d k sø t ‚ t  t d ƒ t k st ‚ t  t d ƒ d k s0t ‚ d  S(   Ni    u   0i   u   N_ii   u   (a) N_i + (-b) N_ji   i   u   (-1) N_ii   u   (0|0)u   (a) (N_i|N_k) + (-b) (N_j|N_k)i   i
   u(   (cos(a)) (C_i|N_k) + (-sin(a)) (C_j|N_k)(	   R   R   R   t
   pretty_v_8t   pretty_v_11R   t   pretty_sR   t
   pretty_d_7(    (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   test_pretty_printing_asciir   s    c           C   sS  t  t d ƒ d k s t ‚ t  t d ƒ d k s8 t ‚ t  t d ƒ d k sT t ‚ t  t d j ƒ d k ss t ‚ t  t d ƒ t k s t ‚ t  t d	 ƒ d
 k s« t ‚ t  t d ƒ t k sÇ t ‚ t  t ƒ t k sß t ‚ t  t d ƒ d k sû t ‚ t  t d ƒ d k st ‚ t  t d ƒ t	 k s3t ‚ t  t d ƒ d k sOt ‚ d  S(   Ni    u   0i   u   N_ii   u   (a) N_i + (-b) N_ju   ((a) N_i, (-b) N_j)i   i   u   (-1) N_ii   u   (0|0)u   (a) (N_i|N_k) + (-b) (N_j|N_k)i   i
   u(   (cos(a)) (C_i|N_k) + (-sin(a)) (C_j|N_k)(
   R   R   R   t   argst   upretty_v_8t   upretty_v_11R   t	   upretty_sR   t   upretty_d_7(    (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   test_pretty_print_unicode   s    c           C   s\  t  t d ƒ d k s t ‚ t  t d ƒ d k s8 t ‚ t  t d ƒ d k sT t ‚ t  t d ƒ d d	 k st t ‚ t  t d
 ƒ d d k s” t ‚ t  t d ƒ d d d k s¸ t ‚ t  t ƒ d k sÐ t ‚ t  t d ƒ d k sì t ‚ t  t d ƒ d d k st ‚ t  t d ƒ d d d k s0t ‚ t  t d ƒ d d d d k sXt ‚ d  S(   Ni    s   \mathbf{\hat{0}}i   s   \mathbf{\hat{i}_{N}}i   s   - \mathbf{\hat{i}_{N}}i   s   (a)\mathbf{\hat{i}_{N}} + s   (- b)\mathbf{\hat{j}_{N}}i   s*   (\mathbf{{x}_{N}} + a^{2})\mathbf{\hat{i}_s   {N}} + \mathbf{\hat{k}_{N}}i   s$   \mathbf{\hat{j}_{N}} + (\mathbf{{x}_s$   {C}}^{2} - \int f{\left(b \right)}\,s    db)\mathbf{\hat{k}_{N}}s'   3 \mathbf{{y}_{C}} \mathbf{{x}_{N}}^{2}s#   (\mathbf{\hat{0}}|\mathbf{\hat{0}})i   s"   (a)(\mathbf{\hat{i}_{N}}{|}\mathbfs   {\hat{k}_{N}})i	   s!   (\mathbf{\hat{k}_{C}}{|}\mathbf{\s&   hat{k}_{N}}) + (\mathbf{\hat{i}_{N}}{|s   }\mathbf{\hat{k}_{N}})i   s$   (a^{2} + b)(\mathbf{\hat{i}_{N}}{|}\s%   mathbf{\hat{k}_{N}}) + (\int f{\left(s$   b \right)}\, db)(\mathbf{\hat{k}_{N}s   }{|}\mathbf{\hat{k}_{N}})(   R   R   R   R   R   (    (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   test_latex_printing‘   s(    c          C   sÄ   t  d d d d d g d d d d	 g ƒ}  |  j j ƒ  d
 k sE t ‚ |  j j ƒ  d k s` t ‚ |  j j d k sx t ‚ |  j j d k s t ‚ |  j j d k s¨ t ‚ |  j j d k sÀ t ‚ d  S(   Nt   At   vector_namest   xt   yt   zt   variable_namest   it   jt   ks   A.is   A.xt   A_it   A_xs   \mathbf{{i}_{A}}s   \mathbf{\hat{x}_{A}}(   R   R/   t   __str__R   R+   t   _pretty_formt   _latex_form(   R)   (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   test_custom_names©   s    N(/   t   sympyR    R   R   R   R   t   sympy.vectorR   R   R   t	   sympy.abcR   R   R	   t   sympy.core.compatibilityR
   t   ut   sympy.utilities.pytestR   R   R   t   orient_new_axisR1   R   R   R   t   appendt   zeroR/   R0   R+   R,   R   R#   R   R$   R   R   R%   R   R&   R    R   R!   R'   R(   R7   (    (    (    s?   lib/python2.7/site-packages/sympy/vector/tests/test_printing.pyt   <module>   s`   			&22		3									