o
    GZŽh  ã                   @   sJ  d dl mZ d dlmZ d dlmZ d dlmZ d dl	m
Z
mZmZmZ d dlmZmZmZ d dlmZ dd	„ Zd
d„ Ze
dƒZe deej¡Zg Zg Ze ej¡ e ej¡ e ej ¡ e ejej ¡ e eej ¡ e eej eej  ¡ e ed ej ej ej ¡ e ed e ej dej e  ej  ¡ edƒZ!e ejee!eƒƒejd  ej  ¡ dZ"dZ#e ejej ¡ e eejeƒ¡ e ed e ej ee!eƒƒej  ¡ dZ$dZ%eD ]
Ze eejB ¡ qìdejd  ej  Z&dZ'dZ(dZ)dZ*dd„ Z+edd„ ƒZ,dd„ Z-dd „ Z.d!d"„ Z/d#d$„ Z0d%S )&é    )ÚFunction)ÚIntegral)Úlatex)Úpretty)Ú
CoordSys3DÚDelÚVectorÚexpress)ÚaÚbÚc)ÚXFAILc                 C   s   t | dddS )zASCII pretty-printingF©Zuse_unicodeZ	wrap_line©Úxpretty©Úexpr© r   úO/var/www/auris/lib/python3.10/site-packages/sympy/vector/tests/test_printing.pyr      ó   r   c                 C   s   t | dddS )zUnicode pretty-printingTFr   r   r   r   r   r   Úupretty   r   r   ÚNÚCé   é   Úfuh         âŽ›   2   âŒ         âŽž    
j_N + âŽœx_C  - âŽ® f(b) dbâŽŸ k_N
      âŽ       âŒ¡        âŽ     z‚j_N + /         /       \
      |   2    |        |
      |x_C  -  | f(b) db|
      |        |        |
      \       /         / uy   âŽ› 2    âŽž        âŽ›âŒ         âŽž    
âŽa  + bâŽ  i_N  + âŽœâŽ® f(b) dbâŽŸ k_N
                âŽâŒ¡        âŽ     zy/ 2    \ + /  /       \
\a  + b/ i_N| |        |
           | | f(b) db|
           | |        |
           \/         / u            2
3â‹…y_Câ‹…x_N z         2
3*y_C*x_N ug   âŽ› 2    âŽž                                     
âŽa  + bâŽ  (i_N|k_N)  + (3â‹…y_C - 3â‹…c) (k_N|k_N)zY/ 2    \ (i_N|k_N) + (3*y_C - 3*c) (k_N|k_N)
\a  + b/                                    c                   C   sð   t td ƒdks
J ‚t td ƒdksJ ‚t td ƒdksJ ‚t td ƒdks(J ‚t td	 ƒd
ks2J ‚t td ƒdks<J ‚t tƒdksDJ ‚t td ƒdksNJ ‚t td ƒdksXJ ‚t td ƒdksbJ ‚t td ƒdkslJ ‚t td	 ƒdksvJ ‚d S )Nr   Ú0é   zN.ir   z(-1)*N.ir   z	N.i + N.jé   z&N.j + (C.x**2 - Integral(f(b), b))*N.ké	   z	C.k + N.iz3*C.y*N.x**2z	(N.i|N.k)é   za*(N.i|N.k)é   za*(N.i|N.k) + (-b)*(N.j|N.k)z2(N.j|N.k) + (C.x**2 - Integral(f(b), b))*(N.k|N.k))ÚstrÚvÚsÚdr   r   r   r   Útest_str_printing[   s   r&   c                   C   sÜ   t td ƒdks
J ‚t td ƒdksJ ‚t td ƒdksJ ‚t td ƒtks(J ‚t td ƒd	ks2J ‚t td
 ƒtks<J ‚t tƒtksDJ ‚t td ƒdksNJ ‚t td ƒdksXJ ‚t td ƒtksbJ ‚t td ƒdkslJ ‚d S )Nr   r   r   Úi_Nr!   ú(a) i_N + (-b) j_Nr   r   ú(-1) i_Né   ú(0|0)ú(a) (i_N|k_N) + (-b) (j_N|k_N)é   é
   ú((cos(a)) (i_C|k_N) + (-sin(a)) (j_C|k_N))r   r#   Ú
pretty_v_8Úpretty_v_11r$   Úpretty_sr%   Ú
pretty_d_7r   r   r   r   Útest_pretty_printing_asciik   s   r4   c                   C   sò   t td ƒdks
J ‚t td ƒdksJ ‚t td ƒdksJ ‚t td jƒdks)J ‚t td ƒtks3J ‚t td	 ƒd
ks=J ‚t td ƒtksGJ ‚t tƒtksOJ ‚t td ƒdksYJ ‚t td ƒdkscJ ‚t td ƒtksmJ ‚t td ƒdkswJ ‚d S )Nr   r   r   r'   r!   r(   z((a) i_N, (-b) j_N)r   r   r)   r*   r+   r,   r-   r.   r/   )	r   r#   ÚargsÚupretty_v_8Úupretty_v_11r$   Ú	upretty_sr%   Úupretty_d_7r   r   r   r   Útest_pretty_print_unicode_vz   s   r:   c                   C   sÜ   t td ƒdks
J ‚t td ƒdksJ ‚t td ƒdksJ ‚t td ƒdks(J ‚t td	 ƒd
ks2J ‚t td ƒdks<J ‚t tƒdksDJ ‚t td ƒdksNJ ‚t td ƒdksXJ ‚t td ƒdksbJ ‚t td ƒdkslJ ‚d S )Nr   z\mathbf{\hat{0}}r   z\mathbf{\hat{i}_{N}}r   z- \mathbf{\hat{i}_{N}}r!   zI\left(a\right)\mathbf{\hat{i}_{N}} + \left(- b\right)\mathbf{\hat{j}_{N}}é   zP\left(\mathbf{{x}_{N}} + a^{2}\right)\mathbf{\hat{i}_{N}} + \mathbf{\hat{k}_{N}}r   zk\mathbf{\hat{j}_{N}} + \left(\mathbf{{x}_{C}}^{2} - \int f{\left(b \right)}\, db\right)\mathbf{\hat{k}_{N}}z'3 \mathbf{{y}_{C}} \mathbf{{x}_{N}}^{2}z#(\mathbf{\hat{0}}|\mathbf{\hat{0}})r    zM\left(a\right)\left(\mathbf{\hat{i}_{N}}{\middle|}\mathbf{\hat{k}_{N}}\right)r   z\left(\mathbf{\hat{k}_{C}}{\middle|}\mathbf{\hat{k}_{N}}\right) + \left(\mathbf{\hat{i}_{N}}{\middle|}\mathbf{\hat{k}_{N}}\right)r*   zÀ\left(a^{2} + b\right)\left(\mathbf{\hat{i}_{N}}{\middle|}\mathbf{\hat{k}_{N}}\right) + \left(\int f{\left(b \right)}\, db\right)\left(\mathbf{\hat{k}_{N}}{\middle|}\mathbf{\hat{k}_{N}}\right))r   r#   r$   r%   r   r   r   r   Útest_latex_printingŠ   s   r<   c                  C   sV  ddl m} m}m}m}m} tƒ }tdƒ}|j}|j	}| dƒ}	| ddd}
d| |
d	  |
d
 }}d|
d  ||
d |	 ƒ ||
d | ƒ }|| }d| t
| || ¡ ¡ |	ƒ }| ¡ }d}d}t|ƒ|kslJ ‚t|ƒ|kstJ ‚|dƒ}
d| |
d	  |
d
 }}d|
d  ||
d |	 ƒ ||
d | ƒ }|| }d}t|ƒ|ks©J ‚d S )Nr   )ÚsymbolsÚsinÚcosÚpiÚUnevaluatedExprr   ÚtÚ10T)Zpositiver    iõÿÿÿéûÿÿÿr   éüÿÿÿr!   éýÿÿÿr   u-  âŽ›     âŽ›y_CâŽž    âŽ›  5  âŽžâŽž    
âŽœ2â‹…sinâŽœâ”€â”€â”€âŽŸâ‹…cosâŽ10 â‹…tâŽ âŽŸ i_C
âŽœ     âŽœ  3âŽŸ           âŽŸ    
âŽœ     âŽ10 âŽ            âŽŸ    
âŽœâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€âŽŸ    
âŽœ           4         âŽŸ    
âŽ         10          âŽ     u0  âŽ›   4    âŽ›  5  âŽž    âŽ›y_CâŽž âŽž    
âŽœ-10 â‹…sinâŽ10 â‹…tâŽ â‹…cosâŽœâ”€â”€â”€âŽŸ âŽŸ k_C
âŽœ                   âŽœ  3âŽŸ âŽŸ    
âŽœ                   âŽ10 âŽ  âŽŸ    
âŽœâ”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€â”€âŽŸ    
âŽ           2â‹…Ï€           âŽ     r.   uo   âŽ›    -4    âŽ›    5âŽž    âŽ›      -3âŽžâŽž     
âŽ2â‹…10  â‹…cosâŽtâ‹…10 âŽ â‹…sinâŽy_Câ‹…10  âŽ âŽ  i_C )Zsympyr=   r>   r?   r@   rA   r   r   ÚyÚir   ÚcrossZdoitr   )r=   r>   r?   r@   rA   ZdelopZCC_rG   ZxhatrB   ÚtenZepsÚmuZBxZvecBZvecEZvecB_strZvecE_strr   r   r   Útest_issue_23058£   s,   , 		,rL   c                  C   s~   t dg d¢g d¢d} | j ¡ dksJ ‚| j ¡ dksJ ‚| jjdks%J ‚| jjdks-J ‚| jjd	ks5J ‚| jjd
ks=J ‚d S )NÚA)ÚxrG   Úz)rH   ÚjÚk)Zvector_namesZvariable_nameszA.izA.xZi_AZx_Az\mathbf{{i}_{A}}z\mathbf{\hat{x}_{A}})r   rH   Ú__str__rN   Z_pretty_formZ_latex_form)rM   r   r   r   Útest_custom_namesÕ   s   
ÿrS   N)1Zsympy.core.functionr   Zsympy.integrals.integralsr   Zsympy.printing.latexr   Zsympy.printing.prettyr   r   Zsympy.vectorr   r   r   r	   Z	sympy.abcr
   r   r   Zsympy.testing.pytestr   r   r   Zorient_new_axisrQ   r   r#   r%   ÚappendÚzerorH   rP   rN   rG   r   r6   r0   r7   r1   r$   r8   r2   r9   r3   r&   r4   r:   r<   rL   rS   r   r   r   r   Ú<module>   sX   	 ,(*
2