o
    6a=                  
   @   sb  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g dedd feddddggedd!ffd"d# ZG d$d% d%ZG d&d' d'ZG d(d) d)ZdS )*    N)arrayarangeprintoptions)assert_equalassert_)Fraction)Decimalc                   @   Z  e Zd Zejddddd Zejdg ddfg d	d
feddffdd Z	ejdg ddfg d	dfeddffdd Z
ejdg ddfg d	dfeddffdd Zejdg ddfg d	dfeddffdd Zejdg ddfg d	dfedd ffd!d" Zejdg dd#fg d	d$fedd%ffd&d' Zd(S ))TestStrUnicodeSuperSubscriptsclassTscopeautousec                 C      t d d S Nunicodepolyset_default_printstyleself r   F/usr/lib/python3/dist-packages/numpy/polynomial/tests/test_printing.pyuse_unicode      z)TestStrUnicodeSuperSubscripts.use_unicodeinptgt            1.0 + 2.0·x¹ + 3.0·x²r   r!   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      t t|}t|| d S Nstrr   
Polynomialr   r   r   r   resr   r   r   test_polynomial_str      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   r&   r'   r)   r   	Chebyshevr   r+   r   r   r   test_chebyshev_str   r.   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   r&   r'   r)   r   Legendrer   r+   r   r   r   test_legendre_str'   r.   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   r&   r'   r)   r   Hermiter   r+   r   r   r   test_hermite_str2   r.   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   r&   r'   r)   r   HermiteEr   r+   r   r   r   test_hermiteE_str=      	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   r&   r'   r)   r   Laguerrer   r+   r   r   r   test_laguerre_strI   r.   z/TestStrUnicodeSuperSubscripts.test_laguerre_strN)__name__
__module____qualname__pytestfixturer   markparametrizer   r-   r2   r5   r8   r;   r?   r   r   r   r   r
      sN    























r
   c                   @   r	   ))TestStrAsciir   Tr   c                 C   r   Nasciir   r   r   r   r   	use_asciiW   r   zTestStrAscii.use_asciir   r   1.0 + 2.0 x**1 + 3.0 x**2r#   z%-1.0 + 0.0 x**1 + 3.0 x**2 - 1.0 x**3r%   z0.0 + 1.0 x**1 + 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   r&   r'   r(   r+   r   r   r   r-   [   r.   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)z0.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   r&   r'   r0   r+   r   r   r   r2   f   r<   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)z0.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   r&   r'   r3   r+   r   r   r   r5   r   r<   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)z0.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   r&   r'   r6   r+   r   r   r   r8   ~   r<   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)z0.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   r&   r'   r9   r+   r   r   r   r;      r<   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)z0.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   r&   r'   r=   r+   r   r   r   r?      r<   zTestStrAscii.test_laguerre_strN)r@   rA   rB   rC   rD   rJ   rE   rF   r   r-   r2   r5   r8   r;   r?   r   r   r   r   rG   U   sN    























rG   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   r   rH   r   r   r   r   r   rJ      r   zTestLinebreaking.use_asciic                 C   2   t g d}ttt|d tt|d d S )N)[rO   rO     r   J   zJ123456789.0 + 123456789.0 x**1 + 123456789.0 x**2 + 1234.0 x**3 + 1.0 x**4r   r*   r   lenr)   r   pr   r   r   test_single_line_one_less   
   z*TestLinebreaking.test_single_line_one_lessc                 C   rN   )N)rO   rO   rO   rP   
   K   zK123456789.0 + 123456789.0 x**1 + 123456789.0 x**2 + 1234.0 x**3 +
10.0 x**4rR   rT   r   r   r   test_num_chars_is_linewidth   rW   z,TestLinebreaking.test_num_chars_is_linewidthc                 C   s<   t g d}ttt|dd d tt|d d S )N)rO   rO   rO   r%   r   rO   
r   rQ   z[123456789.0 + 123456789.0 x**1 + 123456789.0 x**2 + 12.0 x**3 + 1.0 x**4 +
123456789.0 x**5)r   r*   r   rS   r)   splitrT   r   r   r   6test_first_linebreak_multiline_one_less_than_linewidth   s   zGTestLinebreaking.test_first_linebreak_multiline_one_less_than_linewidthc                 C   s    t g d}tt|d d S )N)rO   rO   rO   {   r   rO   z\123456789.0 + 123456789.0 x**1 + 123456789.0 x**2 + 123.0 x**3 +
1.0 x**4 + 123456789.0 x**5)r   r*   r   r)   rT   r   r   r   +test_first_linebreak_multiline_on_linewidth   s   z<TestLinebreaking.test_first_linebreak_multiline_on_linewidth)lwr   ))rY   z0.0 + 10.0 x**1 + 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)-   z0.0 + 10.0 x**1 + 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)   z0.0 + 10.0 x**1 + 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   sp   t g d}t|d" tt|| t|dD ]
}tt||k  qW d    d S 1 s1w   Y  d S )N)
r   rX      i  i@  i  i'	 ip i@  i  )	linewidthr[   )r   r*   r   r   r)   r\   r   rS   )r   r`   r   rU   liner   r   r   test_linewidth_printoption   s   "z+TestLinebreaking.test_linewidth_printoptionN)r@   rA   rB   rC   rD   rJ   rV   rZ   r]   r_   rE   rF   rf   r   r   r   r   rM      s    
		
rM   c                  C   s   t g d} t g d}t d tt| d tt|d t d tt| d tt|d tt t d W d    d S 1 sJw   Y  d S )	Nr   rI   rK   rL   r   r"   r/   invalid_input)	r   r*   r1   r   r   r)   rC   raises
ValueError)rU   cr   r   r   test_set_default_printoptions   s   

"rk   c                  C   st   g d} 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      @        dtyper   u,   1j + (1+1j)·x¹ - (2-2j)·x² + (3+0j)·x³u-   1j + (1+1j)·x¹ + (-2+2j)·x² + (3+0j)·x³rI   z,1j + (1+1j) x**1 - (2-2j) x**2 + (3+0j) x**3z-1j + (1+1j) x**1 + (-2+2j) x**2 + (3+0j) x**3N)r   r*   r   objectr   r   r)   )coefsp1p2r   r   r   test_complex_coefficients   s   


rr   )ro   r   r   r    r!      rl   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)   ro   r   rU   r   r   r    test_numeric_object_coefficients  s   

rx   )r   r    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   Nrv   rw   r   r   r   #test_nonnumeric_object_coefficients  s   

rz   c                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )

TestFormatc                 C   ,   t d t g d}tt|dd d S )NrI   r   r    r   r$   r   u$   1.0 + 2.0·x¹ + 0.0·x² - 1.0·x³r   r   r*   r   formatrT   r   r   r   test_format_unicode   s   
zTestFormat.test_format_unicodec                 C   r|   )Nr   r}   rI   z$1.0 + 2.0 x**1 + 0.0 x**2 - 1.0 x**3r~   rT   r   r   r   test_format_ascii%  s
   

zTestFormat.test_format_asciic                 C   s6   t d t g d}tt|d t| d d S )NrI   r   rK   r~   rT   r   r   r   test_empty_formatstr,  s   
zTestFormat.test_empty_formatstrc                 C   sH   t g d}tt t|d W d    d S 1 sw   Y  d S )Nr}   z.2f)r   r*   rC   rh   ri   r   rT   r   r   r   test_bad_formatstr2  s   "zTestFormat.test_bad_formatstrN)r@   rA   rB   r   r   r   r   r   r   r   r   r{     s
    r{   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   $   t tddg}d}t|| d S )Nr   r   z6Polynomial([0., 1.], domain=[-1,  1], window=[-1,  1]))reprr   r*   r   r   r,   r   r   r   r   r-   9     zTestRepr.test_polynomial_strc                 C   r   )Nr   r   z5Chebyshev([0., 1.], domain=[-1,  1], window=[-1,  1]))r   r   r1   r   r   r   r   r   r2   >  r   zTestRepr.test_chebyshev_strc                 C   r   )Nr   r   z4Legendre([0., 1.], domain=[-1,  1], window=[-1,  1]))r   r   r4   r   r   r   r   r   test_legendre_reprC  r   zTestRepr.test_legendre_reprc                 C   r   )Nr   r   z3Hermite([0., 1.], domain=[-1,  1], window=[-1,  1]))r   r   r7   r   r   r   r   r   test_hermite_reprH  r   zTestRepr.test_hermite_reprc                 C   r   )Nr   r   z4HermiteE([0., 1.], domain=[-1,  1], window=[-1,  1]))r   r   r:   r   r   r   r   r   test_hermiteE_reprM  r   zTestRepr.test_hermiteE_reprc                 C   r   )Nr   r   z0Laguerre([0., 1.], domain=[0, 1], window=[0, 1]))r   r   r>   r   r   r   r   r   test_laguerre_reprR  r   zTestRepr.test_laguerre_reprN)	r@   rA   rB   r-   r2   r   r   r   r   r   r   r   r   r   8  s    r   c                   @   s0   e Zd Z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 |_ z| W |` S |` w )Nc                 S   s   t | S r'   )r)   )xr   r   r   <lambda>_  s    z(TestLatexRepr.as_latex.<locals>.<lambda>)_repr_latex_scalar_repr_latex_)r   objr   r   r   as_latex[  s   
zTestLatexRepr.as_latexc                 C   s   t g d}t| |d t jg dddgd}t| |d t jg dddgd}t| |d	 t jg dd
dgd}t| |d d S )Nr   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}$g      g      ?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   rT   r   r   r   test_simple_polynomiale  s    



z$TestLatexRepr.test_simple_polynomialc                 C   sH   t g d}t| |d t jg dddgd}t| |d d S )Nr   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   r1   r   r   rT   r   r   r   test_basis_funcz  s   

zTestLatexRepr.test_basis_funcc                 C   s"   t g d}t| |d d S )Nr   zB$x \mapsto 1.0\,{He}_{0}(x) + 2.0\,{He}_{1}(x) + 3.0\,{He}_{2}(x)$)r   r:   r   r   rT   r   r   r   test_multichar_basis_func  s   
z'TestLatexRepr.test_multichar_basis_funcN)r@   rA   rB   __doc__r   r   r   r   r   r   r   r   r   X  s    
	r   )rC   
numpy.corer   r   r   numpy.polynomial
polynomialr   numpy.testingr   r   	fractionsr   decimalr   r
   rG   rM   rk   rr   rE   rF   rn   rx   rz   r{   r   r   r   r   r   r   <module>   s>    JNB

	 