|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/numpy/polynomial/tests/__pycache__/ |
Upload File : |
o
6��a�= �
@ 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 @ �Z 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��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 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 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)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 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)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 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)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 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)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 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)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 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 Zd Zejddd�dd� �Zdd� Zdd � Zd
d � Zd d
� Z ej
� dd�dd� �Z dS )�TestLinebreakingr Tr c C r rH r r r r r rJ � r zTestLinebreaking.use_asciic C �2 t �g d��}ttt|��d� tt|�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��}ttt|��d�d �d� tt|�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��}tt|�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 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 sp t �g d��}t|d��" tt|�|� t|��d�D ]
}tt|�|k � qW 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� tt| �d� tt|�d� t �d� tt| �d� tt|�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| 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 @ ��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²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) �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� tt|�|� dS )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 Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestFormatc C �, t �d� t �g d��}tt|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��}tt|�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 Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestReprc C �$ 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 � 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� zTestRepr.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� zTestRepr.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� zTestRepr.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� zTestRepr.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� zTestRepr.test_laguerre_reprN) r@ rA rB r- r2 r� r� r� r� r r r r r� 8 s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
TestLatexReprz#Test the latex repr used by Jupyterc C s dd� |_ 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
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� rT 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 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> JNB
����
�