Current File : //lib64/python3.9/site-packages/numpy/polynomial/tests/__pycache__/test_printing.cpython-39.pyc
a
z[yc�= �
@ sb d dl Z d dlmZmZmZ d dlmZ d dlm Z m
Z
d dl m Z d dl
mZ G dd� d�ZG dd � d �ZG d
d � d �Zd d
� Zdd� Ze j�dee dd�e dd�ged�dfedde dd�ged�dfeed�ed�dged�d ff�d d � �Ze j�deg d �ed�d feddddgged�d!ff�d"d#� �ZG d$d%� d%�ZG d&d'� d'�ZG d(d)� d)�ZdS )*� N)�array�arange� printoptions)� assert_equal�assert_)�Fraction)�Decimalc @ sZ e Zd Zejddd�dd� �Zej�dg d�dfg d �d
fed �d ff�d
d� �Z ej�dg d�dfg d �dfed �dff�dd� �Z
ej�dg d�dfg d �dfed �dff�dd� �Z ej�dg d�dfg d �dfed �dff�d d � �Z ej�dg d�d fg d �d fed �d ff�d!d"� �Z
ej�dg d�d#fg d �d$fed �d%ff�d&d'� �Zd(S ))� TestStrUnicodeSuperSubscripts�classT�ZscopeZautousec C s t �d� d S �N�unicode��poly�set_default_printstyle��self� r �J/usr/lib64/python3.9/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 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 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_str2 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_strI s z/TestStrUnicodeSuperSubscripts.test_laguerre_strN)�__name__�
__module__� __qualname__�pytest�fixturer �mark� parametrizer r' r+ r. r1 r4 r7 r r r r r sL
�
�
�
�
�
�r c @ sZ e Zd Zejddd�dd� �Zej�dg d�dfg d �d
fed �d ff�d
d� �Z ej�dg d�dfg d �dfed �dff�dd� �Z
ej�dg d�dfg d �dfed �dff�dd� �Z ej�dg d�dfg d �dfed �dff�d d � �Z ej�dg d�d fg d �d fed �d ff�d!d"� �Z
ej�dg d�d#fg d �d$fed �d%ff�d&d'� �Zd(S ))� TestStrAsciir
Tr c C s t �d� d S �N�asciir r r r r � use_asciiW s zTestStrAscii.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 z�0.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 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+ f 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. r 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! r/ r% r r r r1 ~ 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! r2 r% r r r r4 � 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! r5 r% r r r r7 � s z TestStrAscii.test_laguerre_strN)r8 r9 r: r; r< rB r= r> r r' r+ r. r1 r4 r7 r r r r r? U sL
�
�
�
�
�
�r? c @ sX e Zd Zejddd�dd� �Zdd� Zdd � Zd
d � Zd d
� Z ej
� dd�dd� �Z dS )�TestLinebreakingr
Tr c C s t �d� d S r@ r r r r r rB � s zTestLinebreaking.use_asciic C s2 t �g d��}ttt|��d� tt|�d� d S )N)��[rF rF �� 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� s
�z*TestLinebreaking.test_single_line_one_lessc C s2 t �g d��}ttt|��d� tt|�d� d S )N)rF rF rF rG �
�K zK123456789.0 + 123456789.0 x**1 + 123456789.0 x**2 + 1234.0 x**3 +
10.0 x**4rI rK r r r �test_num_chars_is_linewidth� s
�z,TestLinebreaking.test_num_chars_is_linewidthc C s< t �g d��}ttt|��d�d �d� tt|�d� d S )N)rF rF rF r r rF �
r rH 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 rJ r# �splitrK 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��}tt|�d� d S )N)rF rF rF �{ r rF 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# rK 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**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)�- z�0.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)� z�0.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 sn t �g d��}t|d��B tt|�|� t|��d�D ]}tt|�|k � q6W d � n1 s`0 Y d S )N)
r rN �� i� i@� i � i�' ip i@ i� )Z linewidthrQ )r r$ r r r# rR r rJ )r rV r rL �liner r r �test_linewidth_printoption� s � z+TestLinebreaking.test_linewidth_printoptionN)
r8 r9 r: r; r< rB rM rP rS rU r= r> r[ r r r r rE � s
rE c C s� t �g d��} t �g d��}t �d� tt| �d� tt|�d� t �d� tt| �d� tt|�d� t�t�� t �d� W d � n1 s�0 Y d S ) Nr rA rC rD r
r r( Z
invalid_input) r r$ r* r r r# r; �raises�
ValueError)rL �cr r r � test_set_default_printoptions� s
r_ c C st g d�} t �| �}t �t| td��}t �d� tt|�d� tt|�d� t �d� tt|�d� tt|�d� d S )
z%Test both numpy and built-in complex.)y �?y �? �?y � @y @ �Zdtyper
u, 1j + (1+1j)·x¹ - (2-2j)·x² + (3+0j)·x³u- 1j + (1+1j)·x¹ + (-2+2j)·x² + (3+0j)·x³rA 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# )�coefsZp1Zp2r r r �test_complex_coefficients� s
rc )rb r r r r � r` u 1/2 + 3/4·x¹� � u 1 + 2·x¹ + 5/7·x²z1.00z2.2u 1.00 + 2.2·x¹ + 3·x²c C s& t �| �}t �d� tt|�|� d S r �r r$ r r r# �rb r rL r r r � test_numeric_object_coefficients s
ri )r r �fu 1 + 2·x¹ + f·x²u 1 + 2·x¹ + [3, 4]·x²c C s& t �| �}t �d� tt|�|� dS )zK
Test coef fallback for object arrays of non-numeric coefficients.
r
Nrg rh r r r �#test_nonnumeric_object_coefficients s
rk c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestFormatc C s, t �d� t �g d��}tt|d�d� d S )NrA �r r r r r
u$ 1.0 + 2.0·x¹ + 0.0·x² - 1.0·x³�r r r$ r �formatrK r r r �test_format_unicode s
z TestFormat.test_format_unicodec C s, t �d� t �g d��}tt|d�d� d S )Nr
rm rA z$1.0 + 2.0 x**1 + 0.0 x**2 - 1.0 x**3rn rK r r r �test_format_ascii% s
�z TestFormat.test_format_asciic C s6 t �d� t �g d��}tt|�d� t|� d� d S )NrA r rC rn rK r r r �test_empty_formatstr, s
z TestFormat.test_empty_formatstrc C sF t �g d��}t�t�� t|d� W d � n1 s80 Y d S )Nrm z.2f)r r$ r; r\ r] ro rK r r r �test_bad_formatstr2 s z TestFormat.test_bad_formatstrN)r8 r9 r: rp rq rr rs r r r r rl s rl c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestReprc C s$ t t�ddg��}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 s z TestRepr.test_polynomial_strc C s$ t t�ddg��}d}t||� d S )Nr r z5Chebyshev([0., 1.], domain=[-1, 1], window=[-1, 1]))ru r r* r rv r r r r+ > s zTestRepr.test_chebyshev_strc C s$ t t�ddg��}d}t||� d S )Nr r z4Legendre([0., 1.], domain=[-1, 1], window=[-1, 1]))ru r r- r rv r r r �test_legendre_reprC s zTestRepr.test_legendre_reprc C s$ t t�ddg��}d}t||� d S )Nr r z3Hermite([0., 1.], domain=[-1, 1], window=[-1, 1]))ru r r0 r rv r r r �test_hermite_reprH s zTestRepr.test_hermite_reprc C s$ t t�ddg��}d}t||� d S )Nr r z4HermiteE([0., 1.], domain=[-1, 1], window=[-1, 1]))ru r r3 r rv r r r �test_hermiteE_reprM s zTestRepr.test_hermiteE_reprc C s$ t t�ddg��}d}t||� d S )Nr r z0Laguerre([0., 1.], domain=[0, 1], window=[0, 1]))ru r r6 r rv r r r �test_laguerre_reprR s zTestRepr.test_laguerre_reprN) r8 r9 r: r' r+ rw rx ry rz r r r r rt 8 s rt c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
TestLatexReprz#Test the latex repr used by Jupyterc C s$ dd� |_ z|�� W |` S |` 0 d S )Nc S s t | �S r! )r# )�xr r r �<lambda>_ � z(TestLatexRepr.as_latex.<locals>.<lambda>)Z_repr_latex_scalarZ _repr_latex_)r �objr r r �as_latex[ s
�zTestLatexRepr.as_latexc C s� t �g d��}t| �|�d� t jg d�ddgd�}t| �|�d� t jg d�ddgd�}t| �|�d � t jg d�d
dgd�}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� rK r r r �test_simple_polynomiale s
�
�
�
�z$TestLatexRepr.test_simple_polynomialc C sH t �g d��}t| �|�d� t jg d�ddgd�}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 r* r r� rK 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 r3 r r� rK r r r �test_multichar_basis_func� s
�z'TestLatexRepr.test_multichar_basis_funcN)r8 r9 r: �__doc__r� r� r� r� r r r r r{ X s
r{ ) r; Z
numpy.corer r r Znumpy.polynomialZ
polynomialr Z
numpy.testingr r Z fractionsr �decimalr r r? rE r_ rc r= r> ra ri rk rl rt r{ r r r r �<module> s< JNB
����
�
Mini Shell