U
    ¸mœd-P  ã                
   @   sÌ  d dl mZmZ d dlZd dlmZmZmZ d dlm	Z
 d dlmZmZ d dlmZ d dlmZ G dd„ dƒZG d	d
„ d
ƒZG dd„ dƒZdd„ Zdd„ Zej deeddƒeddƒged�dfeddeddƒged�dfeedƒedƒdged�dff¡dd„ ƒZej deddd ged�d!feddddgged�d"ff¡d#d$„ ƒZG d%d&„ d&ƒZej d'e
jd(fe
jd)fe
jd*fe
j d+fe
j!d,fe
j"d-ff¡d.d/„ ƒZ#G d0d1„ d1ƒZ$G d2d3„ d3ƒZ%d4Z&G d5d6„ d6ƒZ'dS )7é    )ÚnanÚinfN)ÚarrayÚarangeÚprintoptions)Úassert_equalÚassert_)ÚFraction)ÚDecimalc                   @   s~  e Zd Zejddd�dd„ ƒZej dddd	gd
fddd	dgdfedƒdff¡dd„ ƒZ	ej dddd	gdfddd	dgdfedƒdff¡dd„ ƒZ
ej dddd	gdfddd	dgdfedƒdff¡dd„ ƒZej dddd	gdfddd	dgdfedƒdff¡dd „ ƒZej dddd	gd!fddd	dgd"fedƒd#ff¡d$d%„ ƒZej dddd	gd&fddd	dgd'fedƒd(ff¡d)d*„ ƒZd+S ),ÚTestStrUnicodeSuperSubscriptsÚclassT©ÚscopeZautousec                 C   s   t  d¡ d S ©NÚunicode©ÚpolyÚset_default_printstyle©Úself© r   ú]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numpy/polynomial/tests/test_printing.pyÚuse_unicode   s    z)TestStrUnicodeSuperSubscripts.use_unicode©ÚinpÚtgté   é   é   õ   1.0 + 2.0Â·x + 3.0Â·xÂ²éÿÿÿÿr   u#   -1.0 + 0.0Â·x + 3.0Â·xÂ² - 1.0Â·xÂ³é   u‡   0.0 + 1.0Â·x + 2.0Â·xÂ² + 3.0Â·xÂ³ + 4.0Â·xâ�´ + 5.0Â·xâ�µ + 6.0Â·xâ�¶ + 7.0Â·xâ�· +
8.0Â·xâ�¸ + 9.0Â·xâ�¹ + 10.0Â·xÂ¹â�° + 11.0Â·xÂ¹Â¹c                 C   s   t t |¡ƒ}t||ƒ d S ©N©Ústrr   Ú
Polynomialr   ©r   r   r   Úresr   r   r   Útest_polynomial_str   s    z1TestStrUnicodeSuperSubscripts.test_polynomial_strõ!   1.0 + 2.0Â·Tâ‚�(x) + 3.0Â·Tâ‚‚(x)u1   -1.0 + 0.0Â·Tâ‚�(x) + 3.0Â·Tâ‚‚(x) - 1.0Â·Tâ‚ƒ(x)u°   0.0 + 1.0Â·Tâ‚�(x) + 2.0Â·Tâ‚‚(x) + 3.0Â·Tâ‚ƒ(x) + 4.0Â·Tâ‚„(x) + 5.0Â·Tâ‚…(x) +
6.0Â·Tâ‚†(x) + 7.0Â·Tâ‚‡(x) + 8.0Â·Tâ‚ˆ(x) + 9.0Â·Tâ‚‰(x) + 10.0Â·Tâ‚�â‚€(x) + 11.0Â·Tâ‚�â‚�(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   ©r$   r   Ú	Chebyshevr   r&   r   r   r   Útest_chebyshev_str   s    z0TestStrUnicodeSuperSubscripts.test_chebyshev_stru!   1.0 + 2.0Â·Pâ‚�(x) + 3.0Â·Pâ‚‚(x)u1   -1.0 + 0.0Â·Pâ‚�(x) + 3.0Â·Pâ‚‚(x) - 1.0Â·Pâ‚ƒ(x)u°   0.0 + 1.0Â·Pâ‚�(x) + 2.0Â·Pâ‚‚(x) + 3.0Â·Pâ‚ƒ(x) + 4.0Â·Pâ‚„(x) + 5.0Â·Pâ‚…(x) +
6.0Â·Pâ‚†(x) + 7.0Â·Pâ‚‡(x) + 8.0Â·Pâ‚ˆ(x) + 9.0Â·Pâ‚‰(x) + 10.0Â·Pâ‚�â‚€(x) + 11.0Â·Pâ‚�â‚�(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   ©r$   r   ÚLegendrer   r&   r   r   r   Útest_legendre_str(   s    z/TestStrUnicodeSuperSubscripts.test_legendre_stru!   1.0 + 2.0Â·Hâ‚�(x) + 3.0Â·Hâ‚‚(x)u1   -1.0 + 0.0Â·Hâ‚�(x) + 3.0Â·Hâ‚‚(x) - 1.0Â·Hâ‚ƒ(x)u°   0.0 + 1.0Â·Hâ‚�(x) + 2.0Â·Hâ‚‚(x) + 3.0Â·Hâ‚ƒ(x) + 4.0Â·Hâ‚„(x) + 5.0Â·Hâ‚…(x) +
6.0Â·Hâ‚†(x) + 7.0Â·Hâ‚‡(x) + 8.0Â·Hâ‚ˆ(x) + 9.0Â·Hâ‚‰(x) + 10.0Â·Hâ‚�â‚€(x) + 11.0Â·Hâ‚�â‚�(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   ©r$   r   ÚHermiter   r&   r   r   r   Útest_hermite_str3   s    z.TestStrUnicodeSuperSubscripts.test_hermite_stru#   1.0 + 2.0Â·Heâ‚�(x) + 3.0Â·Heâ‚‚(x)u4   -1.0 + 0.0Â·Heâ‚�(x) + 3.0Â·Heâ‚‚(x) - 1.0Â·Heâ‚ƒ(x)u»   0.0 + 1.0Â·Heâ‚�(x) + 2.0Â·Heâ‚‚(x) + 3.0Â·Heâ‚ƒ(x) + 4.0Â·Heâ‚„(x) + 5.0Â·Heâ‚…(x) +
6.0Â·Heâ‚†(x) + 7.0Â·Heâ‚‡(x) + 8.0Â·Heâ‚ˆ(x) + 9.0Â·Heâ‚‰(x) + 10.0Â·Heâ‚�â‚€(x) +
11.0Â·Heâ‚�â‚�(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   ©r$   r   ÚHermiteEr   r&   r   r   r   Útest_hermiteE_str>   s    	z/TestStrUnicodeSuperSubscripts.test_hermiteE_stru!   1.0 + 2.0Â·Lâ‚�(x) + 3.0Â·Lâ‚‚(x)u1   -1.0 + 0.0Â·Lâ‚�(x) + 3.0Â·Lâ‚‚(x) - 1.0Â·Lâ‚ƒ(x)u°   0.0 + 1.0Â·Lâ‚�(x) + 2.0Â·Lâ‚‚(x) + 3.0Â·Lâ‚ƒ(x) + 4.0Â·Lâ‚„(x) + 5.0Â·Lâ‚…(x) +
6.0Â·Lâ‚†(x) + 7.0Â·Lâ‚‡(x) + 8.0Â·Lâ‚ˆ(x) + 9.0Â·Lâ‚‰(x) + 10.0Â·Lâ‚�â‚€(x) + 11.0Â·Lâ‚�â‚�(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   ©r$   r   ÚLaguerrer   r&   r   r   r   Útest_laguerre_strJ   s    z/TestStrUnicodeSuperSubscripts.test_laguerre_strN)Ú__name__Ú
__module__Ú__qualname__ÚpytestÚfixturer   ÚmarkÚparametrizer   r(   r,   r/   r2   r5   r8   r   r   r   r   r      sL   

ý

ý

ý

ý

ý

ýr   c                   @   s~  e Zd Zejddd�dd„ ƒZej dddd	gd
fddd	dgdfedƒdff¡dd„ ƒZ	ej dddd	gdfddd	dgdfedƒdff¡dd„ ƒZ
ej dddd	gdfddd	dgdfedƒdff¡dd„ ƒZej dddd	gdfddd	dgdfedƒdff¡dd „ ƒZej dddd	gd!fddd	dgd"fedƒd#ff¡d$d%„ ƒZej dddd	gd&fddd	dgd'fedƒd(ff¡d)d*„ ƒZd+S ),ÚTestStrAsciir   Tr   c                 C   s   t  d¡ d S ©NÚasciir   r   r   r   r   Ú	use_asciiX   s    zTestStrAscii.use_asciir   r   r   r   ú1.0 + 2.0 x + 3.0 x**2r    r   z"-1.0 + 0.0 x + 3.0 x**2 - 1.0 x**3r!   z}0.0 + 1.0 x + 2.0 x**2 + 3.0 x**3 + 4.0 x**4 + 5.0 x**5 + 6.0 x**6 +
7.0 x**7 + 8.0 x**8 + 9.0 x**9 + 10.0 x**10 + 11.0 x**11c                 C   s   t t |¡ƒ}t||ƒ d S r"   r#   r&   r   r   r   r(   \   s    z TestStrAscii.test_polynomial_strú1.0 + 2.0 T_1(x) + 3.0 T_2(x)z+-1.0 + 0.0 T_1(x) + 3.0 T_2(x) - 1.0 T_3(x)z–0.0 + 1.0 T_1(x) + 2.0 T_2(x) + 3.0 T_3(x) + 4.0 T_4(x) + 5.0 T_5(x) +
6.0 T_6(x) + 7.0 T_7(x) + 8.0 T_8(x) + 9.0 T_9(x) + 10.0 T_10(x) +
11.0 T_11(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   r*   r&   r   r   r   r,   g   s    	zTestStrAscii.test_chebyshev_strz1.0 + 2.0 P_1(x) + 3.0 P_2(x)z+-1.0 + 0.0 P_1(x) + 3.0 P_2(x) - 1.0 P_3(x)z–0.0 + 1.0 P_1(x) + 2.0 P_2(x) + 3.0 P_3(x) + 4.0 P_4(x) + 5.0 P_5(x) +
6.0 P_6(x) + 7.0 P_7(x) + 8.0 P_8(x) + 9.0 P_9(x) + 10.0 P_10(x) +
11.0 P_11(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   r-   r&   r   r   r   r/   s   s    	zTestStrAscii.test_legendre_strz1.0 + 2.0 H_1(x) + 3.0 H_2(x)z+-1.0 + 0.0 H_1(x) + 3.0 H_2(x) - 1.0 H_3(x)z–0.0 + 1.0 H_1(x) + 2.0 H_2(x) + 3.0 H_3(x) + 4.0 H_4(x) + 5.0 H_5(x) +
6.0 H_6(x) + 7.0 H_7(x) + 8.0 H_8(x) + 9.0 H_9(x) + 10.0 H_10(x) +
11.0 H_11(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   r0   r&   r   r   r   r2      s    	zTestStrAscii.test_hermite_strz1.0 + 2.0 He_1(x) + 3.0 He_2(x)z.-1.0 + 0.0 He_1(x) + 3.0 He_2(x) - 1.0 He_3(x)z¡0.0 + 1.0 He_1(x) + 2.0 He_2(x) + 3.0 He_3(x) + 4.0 He_4(x) +
5.0 He_5(x) + 6.0 He_6(x) + 7.0 He_7(x) + 8.0 He_8(x) + 9.0 He_9(x) +
10.0 He_10(x) + 11.0 He_11(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   r3   r&   r   r   r   r5   ‹   s    	zTestStrAscii.test_hermiteE_strz1.0 + 2.0 L_1(x) + 3.0 L_2(x)z+-1.0 + 0.0 L_1(x) + 3.0 L_2(x) - 1.0 L_3(x)z–0.0 + 1.0 L_1(x) + 2.0 L_2(x) + 3.0 L_3(x) + 4.0 L_4(x) + 5.0 L_5(x) +
6.0 L_6(x) + 7.0 L_7(x) + 8.0 L_8(x) + 9.0 L_9(x) + 10.0 L_10(x) +
11.0 L_11(x)c                 C   s   t t |¡ƒ}t||ƒ d S r"   r6   r&   r   r   r   r8   —   s    	zTestStrAscii.test_laguerre_strN)r9   r:   r;   r<   r=   rC   r>   r?   r   r(   r,   r/   r2   r5   r8   r   r   r   r   r@   V   sL   

ý

ý

ý

ý

ý

ýr@   c                   @   sX   e Zd Zejddd�dd„ ƒZdd„ Zdd	„ Zd
d„ Zdd„ Z	ej
 dd¡dd„ ƒZdS )ÚTestLinebreakingr   Tr   c                 C   s   t  d¡ d S rA   r   r   r   r   r   rC   ¦   s    zTestLinebreaking.use_asciic                 C   s8   t  dddddg¡}ttt|ƒƒdƒ tt|ƒdƒ d S )NéNa¼ é{   éJ   zJ12345678.0 + 12345678.0 x + 12345678.0 x**2 + 12345678.0 x**3 + 123.0 x**4©r   r%   r   Úlenr$   ©r   Úpr   r   r   Útest_single_line_one_lessª   s
    ÿz*TestLinebreaking.test_single_line_one_lessc                 C   s8   t  dddddg¡}ttt|ƒƒdƒ tt|ƒdƒ d S )NrG   iÒ  éK   zK12345678.0 + 12345678.0 x + 12345678.0 x**2 + 12345678.0 x**3 +
1234.0 x**4rJ   rL   r   r   r   Útest_num_chars_is_linewidth³   s
    ÿz,TestLinebreaking.test_num_chars_is_linewidthc                 C   sD   t  ddddddg¡}ttt|ƒ d¡d ƒdƒ tt|ƒdƒ d S )NrG   r   Ú
r   rI   zZ12345678.0 + 12345678.0 x + 12345678.0 x**2 + 12345678.0 x**3 + 1.0 x**4 +
12345678.0 x**5)r   r%   r   rK   r$   ÚsplitrL   r   r   r   Ú6test_first_linebreak_multiline_one_less_than_linewidth¼   s    ÿÿzGTestLinebreaking.test_first_linebreak_multiline_one_less_than_linewidthc                 C   s(   t  ddddddg¡}tt|ƒdƒ d S )NrG   g=
×Ã)ŒgAr   z[12345678.0 + 12345678.0 x + 12345678.0 x**2 + 12345678.12 x**3 +
1.0 x**4 + 12345678.0 x**5)r   r%   r   r$   rL   r   r   r   Ú+test_first_linebreak_multiline_on_linewidthÇ   s    ÿÿz<TestLinebreaking.test_first_linebreak_multiline_on_linewidth)Úlwr   ))rO   z€0.0 + 10.0 x + 200.0 x**2 + 3000.0 x**3 + 40000.0 x**4 + 500000.0 x**5 +
600000.0 x**6 + 70000.0 x**7 + 8000.0 x**8 + 900.0 x**9)é-   z€0.0 + 10.0 x + 200.0 x**2 + 3000.0 x**3 +
40000.0 x**4 + 500000.0 x**5 +
600000.0 x**6 + 70000.0 x**7 + 8000.0 x**8 +
900.0 x**9)é„   z€0.0 + 10.0 x + 200.0 x**2 + 3000.0 x**3 + 40000.0 x**4 + 500000.0 x**5 + 600000.0 x**6 + 70000.0 x**7 + 8000.0 x**8 + 900.0 x**9c                 C   sj   t  ddddddddd	d
g
¡}t|d��8 tt|ƒ|ƒ t|ƒ d¡D ]}tt|ƒ|k ƒ qFW 5 Q R X d S )Nr   é
   éÈ   i¸  i@œ  i ¡ iÀ'	 ip i@  i„  )Z	linewidthrQ   )r   r%   r   r   r$   rR   r   rK   )r   rU   r   rM   Úliner   r   r   Útest_linewidth_printoptionÑ   s    ÿz+TestLinebreaking.test_linewidth_printoptionN)r9   r:   r;   r<   r=   rC   rN   rP   rS   rT   r>   r?   r[   r   r   r   r   rF   ¤   s   
		
rF   c               	   C   s�   t  dddg¡} t  dddg¡}t  d¡ tt| ƒdƒ tt|ƒdƒ t  d¡ tt| ƒdƒ tt|ƒd	ƒ t t¡� t  d
¡ W 5 Q R X d S )Nr   r   r   rB   rD   rE   r   r   r)   Zinvalid_input)	r   r%   r+   r   r   r$   r<   ÚraisesÚ
ValueError)rM   Úcr   r   r   Útest_set_default_printoptionsæ   s    

r_   c                  C   sx   ddddg} t  | ¡}t  t| td�¡}t  d¡ tt|ƒdƒ tt|ƒdƒ t  d	¡ tt|ƒd
ƒ tt|ƒdƒ dS )z%Test both numpy and built-in complex.y              ð?y      ð?      ð?y       À       @y      @        ©Zdtyper   u*   1j + (1+1j)Â·x - (2-2j)Â·xÂ² + (3+0j)Â·xÂ³u+   1j + (1+1j)Â·x + (-2+2j)Â·xÂ² + (3+0j)Â·xÂ³rB   z)1j + (1+1j) x - (2-2j) x**2 + (3+0j) x**3z*1j + (1+1j) x + (-2+2j) x**2 + (3+0j) x**3N)r   r%   r   Úobjectr   r   r$   )ÚcoefsÚp1Úp2r   r   r   Útest_complex_coefficientsó   s    


re   )rb   r   r   r   r   é   r`   u   1/2 + 3/4Â·xé   é   u   1 + 2Â·x + 5/7Â·xÂ²z1.00z2.2u   1.00 + 2.2Â·x + 3Â·xÂ²c                 C   s&   t  | ¡}t  d¡ tt|ƒ|ƒ d S r   ©r   r%   r   r   r$   ©rb   r   rM   r   r   r   Ú test_numeric_object_coefficients  s    

rk   Úfu   1 + 2Â·x + fÂ·xÂ²u   1 + 2Â·x + [3, 4]Â·xÂ²c                 C   s&   t  | ¡}t  d¡ tt|ƒ|ƒ dS )zK
    Test coef fallback for object arrays of non-numeric coefficients.
    r   Nri   rj   r   r   r   Ú#test_nonnumeric_object_coefficients  s    

rm   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
Ú
TestFormatc                 C   s0   t  d¡ t  ddddg¡}tt|dƒdƒ d S )NrB   r   r   r   r    r   u"   1.0 + 2.0Â·x + 0.0Â·xÂ² - 1.0Â·xÂ³©r   r   r%   r   ÚformatrL   r   r   r   Útest_format_unicode!  s    
zTestFormat.test_format_unicodec                 C   s0   t  d¡ t  ddddg¡}tt|dƒdƒ d S )Nr   r   r   r   r    rB   z!1.0 + 2.0 x + 0.0 x**2 - 1.0 x**3ro   rL   r   r   r   Útest_format_ascii&  s    
 ÿzTestFormat.test_format_asciic                 C   s8   t  d¡ t  dddg¡}tt|ƒdƒ t|› dƒ d S )NrB   r   r   r   rD   ro   rL   r   r   r   Útest_empty_formatstr-  s    
zTestFormat.test_empty_formatstrc              	   C   s6   t  ddddg¡}t t¡� t|dƒ W 5 Q R X d S )Nr   r   r   r    z.2f)r   r%   r<   r\   r]   rp   rL   r   r   r   Útest_bad_formatstr3  s    zTestFormat.test_bad_formatstrN)r9   r:   r;   rq   rr   rs   rt   r   r   r   r   rn      s   rn   )r   r   u   1.0 + 2.0Â·z + 3.0Â·zÂ²u!   1.0 + 2.0Â·Tâ‚�(z) + 3.0Â·Tâ‚‚(z)u!   1.0 + 2.0Â·Hâ‚�(z) + 3.0Â·Hâ‚‚(z)u#   1.0 + 2.0Â·Heâ‚�(z) + 3.0Â·Heâ‚‚(z)u!   1.0 + 2.0Â·Lâ‚�(z) + 3.0Â·Lâ‚‚(z)u!   1.0 + 2.0Â·Pâ‚�(z) + 3.0Â·Pâ‚‚(z)c                 C   s$   | dddgdd�}t |d›|ƒ d S )Nr   r   r   Úz©Úsymbolr   )r   )r   r   rM   r   r   r   Útest_symbol9  s    	rx   c                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )ÚTestReprc                 C   s$   t t ddg¡ƒ}d}t||ƒ d S )Nr   r   zBPolynomial([0., 1.], domain=[-1,  1], window=[-1,  1], symbol='x'))Úreprr   r%   r   ©r   r'   r   r   r   r   r(   G  s    ÿzTestRepr.test_polynomial_strc                 C   s$   t t ddg¡ƒ}d}t||ƒ d S )Nr   r   zAChebyshev([0., 1.], domain=[-1,  1], window=[-1,  1], symbol='x'))rz   r   r+   r   r{   r   r   r   r,   O  s    ÿzTestRepr.test_chebyshev_strc                 C   s$   t t ddg¡ƒ}d}t||ƒ d S )Nr   r   z@Legendre([0., 1.], domain=[-1,  1], window=[-1,  1], symbol='x'))rz   r   r.   r   r{   r   r   r   Útest_legendre_reprW  s    ÿzTestRepr.test_legendre_reprc                 C   s$   t t ddg¡ƒ}d}t||ƒ d S )Nr   r   z?Hermite([0., 1.], domain=[-1,  1], window=[-1,  1], symbol='x'))rz   r   r1   r   r{   r   r   r   Útest_hermite_repr_  s    ÿzTestRepr.test_hermite_reprc                 C   s$   t t ddg¡ƒ}d}t||ƒ d S )Nr   r   z@HermiteE([0., 1.], domain=[-1,  1], window=[-1,  1], symbol='x'))rz   r   r4   r   r{   r   r   r   Útest_hermiteE_reprg  s    ÿzTestRepr.test_hermiteE_reprc                 C   s$   t t ddg¡ƒ}d}t||ƒ d S )Nr   r   z<Laguerre([0., 1.], domain=[0, 1], window=[0, 1], symbol='x'))rz   r   r7   r   r{   r   r   r   Útest_laguerre_repro  s    ÿzTestRepr.test_laguerre_reprN)	r9   r:   r;   r(   r,   r|   r}   r~   r   r   r   r   r   ry   F  s   ry   c                   @   s8   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ ZdS )ÚTestLatexReprz#Test the latex repr used by Jupyterc                 C   s$   ddd„|_ z| ¡ W ¢S |` X d S )NFc                 S   s   t | ƒS r"   )r$   )ÚxÚparensr   r   r   Ú<lambda>  ó    z(TestLatexRepr.as_latex.<locals>.<lambda>)F)Z_repr_latex_scalarÚ_repr_latex_)r   Úobjr   r   r   Úas_latex{  s    zTestLatexRepr.as_latexc                 C   sœ   t  dddg¡}t|  |¡dƒ t jdddgddgd�}t|  |¡dƒ t jdddgd	d
gd�}t|  |¡dƒ t jdddgddgd�}t|  |¡dƒ d S )Nr   r   r   z%$x \mapsto 1.0 + 2.0\,x + 3.0\,x^{2}$éþÿÿÿr   ©ÚdomainzK$x \mapsto 1.0 + 2.0\,\left(1.0 + x\right) + 3.0\,\left(1.0 + x\right)^{2}$ç      à¿ç      à?zE$x \mapsto 1.0 + 2.0\,\left(2.0x\right) + 3.0\,\left(2.0x\right)^{2}$r    zQ$x \mapsto 1.0 + 2.0\,\left(1.0 + 2.0x\right) + 3.0\,\left(1.0 + 2.0x\right)^{2}$©r   r%   r   r‡   rL   r   r   r   Útest_simple_polynomial…  s     
ÿ
ÿ
ÿ
ÿz$TestLatexRepr.test_simple_polynomialc                 C   sL   t  dddg¡}t|  |¡dƒ t jdddgddgd�}t|  |¡dƒ d S )	Nr   r   r   z?$x \mapsto 1.0\,{T}_{0}(x) + 2.0\,{T}_{1}(x) + 3.0\,{T}_{2}(x)$r    r   r‰   zZ$x \mapsto 1.0\,{T}_{0}(1.0 + 2.0x) + 2.0\,{T}_{1}(1.0 + 2.0x) + 3.0\,{T}_{2}(1.0 + 2.0x)$)r   r+   r   r‡   rL   r   r   r   Útest_basis_funcš  s    
ÿ
ÿzTestLatexRepr.test_basis_funcc                 C   s$   t  dddg¡}t|  |¡dƒ d S )Nr   r   r   zB$x \mapsto 1.0\,{He}_{0}(x) + 2.0\,{He}_{1}(x) + 3.0\,{He}_{2}(x)$)r   r4   r   r‡   rL   r   r   r   Útest_multichar_basis_func£  s    
ÿz'TestLatexRepr.test_multichar_basis_funcc                 C   s¦   t jdddgdd�}t|  |¡dƒ t jdddgddgdd	�}t|  |¡d
ƒ t jdddgddgdd	�}t|  |¡dƒ t jdddgddgdd	�}t|  |¡dƒ d S )Nr   r   r   ru   rv   z%$z \mapsto 1.0 + 2.0\,z + 3.0\,z^{2}$rˆ   r   )rŠ   rw   zK$z \mapsto 1.0 + 2.0\,\left(1.0 + z\right) + 3.0\,\left(1.0 + z\right)^{2}$r‹   rŒ   zE$z \mapsto 1.0 + 2.0\,\left(2.0z\right) + 3.0\,\left(2.0z\right)^{2}$r    zQ$z \mapsto 1.0 + 2.0\,\left(1.0 + 2.0z\right) + 3.0\,\left(1.0 + 2.0z\right)^{2}$r�   rL   r   r   r   Útest_symbol_basic¨  s&    
ÿý	ý	ýzTestLatexRepr.test_symbol_basicN)	r9   r:   r;   Ú__doc__r‡   rŽ   r�   r�   r‘   r   r   r   r   r€   x  s   
	r€   )z"1.0 + (1.0e-01) x + (1.0e-02) x**2z"1.2 + (1.2e-01) x + (1.2e-02) x**2z11.23 + 0.12 x + (1.23e-02) x**2 + (1.23e-03) x**3z51.235 + 0.123 x + (1.235e-02) x**2 + (1.235e-03) x**3zG1.2346 + 0.1235 x + 0.0123 x**2 + (1.2346e-03) x**3 + (1.2346e-04) x**4zL1.23457 + 0.12346 x + 0.01235 x**2 + (1.23457e-03) x**3 + (1.23457e-04) x**4za1.234568 + 0.123457 x + 0.012346 x**2 + 0.001235 x**3 + (1.234568e-04) x**4 + (1.234568e-05) x**5zg1.2345679 + 0.1234568 x + 0.0123457 x**2 + 0.0012346 x**3 + (1.2345679e-04) x**4 + (1.2345679e-05) x**5c                   @   sN   e Zd ZdZejddd�dd„ ƒZdd„ Zd	d
„ Zdd„ Z	dd„ Z
dd„ ZdS )ÚTestPrintOptionsz©
    Test the output is properly configured via printoptions.
    The exponential notation is enabled automatically when the values 
    are too small or too large.
    r   Tr   c                 C   s   t  d¡ d S rA   r   r   r   r   r   rC   á  s    zTestPrintOptions.use_asciic              	   C   sH   t  ddddg¡}tt|ƒdƒ tdd�� tt|ƒdƒ W 5 Q R X d S )	NrŒ   ç’$I’$IÂ?çI’$Ir?kAçmÛ¶m§¡AzC0.5 + 0.14285714 x + 14285714.28571429 x**2 + (1.42857143e+08) x**3r   ©Ú	precisionz40.5 + 0.143 x + 14285714.286 x**2 + (1.429e+08) x**3©r   r%   r   r$   r   rL   r   r   r   Útest_strå  s    zTestPrintOptions.test_strc              	   C   sH   t  ddddg¡}t| ¡ dƒ tdd�� t| ¡ dƒ W 5 Q R X d S )	NrŒ   r”   r•   r–   zp$x \mapsto \text{0.5} + \text{0.14285714}\,x + \text{14285714.28571429}\,x^{2} + \text{(1.42857143e+08)}\,x^{3}$r   r—   za$x \mapsto \text{0.5} + \text{0.143}\,x + \text{14285714.286}\,x^{2} + \text{(1.429e+08)}\,x^{3}$)r   r%   r   r…   r   rL   r   r   r   Ú
test_latexî  s    ÿÿzTestPrintOptions.test_latexc              	   C   sh   t  dg¡}tt|ƒdƒ tdd�� tt|ƒdƒ W 5 Q R X tddd�� tt|ƒdƒ W 5 Q R X d S )	NrŒ   z0.5Zfixed)Ú	floatmodez
0.50000000rf   )rœ   r˜   z0.5000r™   rL   r   r   r   Ú
test_fixedú  s    zTestPrintOptions.test_fixedc              
   C   sd   t tƒD ]V\}}t|d��> t dd„ t|d d ƒD ƒ¡}t|ƒ dd¡|ksTt‚W 5 Q R X qd S )Nr—   c                 S   s   g | ]}d d|   ‘qS )gÞƒBÊÀó?rX   r   )Ú.0Úir   r   r   Ú
<listcomp>  s   ÿz7TestPrintOptions.test_switch_to_exp.<locals>.<listcomp>r   r   rQ   ú )	Ú	enumerateÚSWITCH_TO_EXPr   r   r%   Úranger$   ÚreplaceÚAssertionError)r   rŸ   ÚsrM   r   r   r   Útest_switch_to_exp  s    
ÿz#TestPrintOptions.test_switch_to_expc              	   C   sj   t  ttg¡}t|ƒdkst‚| ¡ dks.t‚tddd��& t|ƒdksLt‚| ¡ dks\t‚W 5 Q R X d S )Nznan + inf xz&$x \mapsto \text{nan} + \text{inf}\,x$ZNANZINF)ZnanstrZinfstrzNAN + INF xz&$x \mapsto \text{NAN} + \text{INF}\,x$)r   r%   r   r   r$   r¦   r…   r   rL   r   r   r   Útest_non_finite  s    ÿz TestPrintOptions.test_non_finiteN)r9   r:   r;   r’   r<   r=   rC   rš   r›   r�   r¨   r©   r   r   r   r   r“   Ú  s   
	
r“   )(Úmathr   r   r<   Z
numpy.corer   r   r   Znumpy.polynomialZ
polynomialr   Znumpy.testingr   r   Ú	fractionsr	   Údecimalr
   r   r@   rF   r_   re   r>   r?   ra   rk   rm   rn   r%   r+   r1   r4   r7   r.   rx   ry   r€   r£   r“   r   r   r   r   Ú<module>   sT   JNBÿÿÿù
þ
	ú
2U