Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test_loadtxt.cpython-39.opt-1.pyc
a
z[yc � � @ s* d Z ddlZddlZddlZddlmZmZ ddlmZ ddl Z
ddl m Z ddl
mZmZmZ dd� Zej�d g d
��d d � �Zejd
d� �Zej�dg d��dd� �Zdd� Zdd� Zdd� Zdd� Zdd � Z ej�d d �d d � �Z ej�d d �d!d"� �Z ej�d#d$d%g�d&d'� �Z ej�d(g d)��d*d+� �Z ej�d,d-�d.d/� �Z!d0d1� Z"ej�d2e#e$f�d3d4� �Z%ejj&e�o�ej'j(d5kd6d7�ej�d2e
�)d8�e
�)d9�f�d:d;� ��Z*d<d=� Z+d>d?� Z,d@dA� Z-dBdC� Z.ej�dDg dE��ej�dFg dE��dGdH� ��Z/dIdJ� Z0dKdL� Z1dMdN� Z2dOdP� Z3ej�d2dQ�dRdS� �Z4ej�dTdUe
�)dV�fdWe
�)dX�fg�dYdZ� �Z5d[d\� Z6d]d^� Z7ejj&e�o�ej'j(d5kd6d7�ej�d2e
j8d_ �ej�9d`�dadb� ���Z:ejj&e�o�ej'j(d5kd6d7�ej�d2e
j8d_ �ej�9d`�dcdd� ���Z;ej�d2e
j<e
j=f�ej�dedf�dgdh� ��Z>didj� Z?dkdl� Z@dmdn� ZAejj&e dod7�dpdq� �ZBejj&e�onej'j(d5kd6d7�drds� �ZCej�dtde#gduf�dvdw� �ZDejj&e�o�ej'j(d5kd6d7�dxdy� �ZEej�dzd{�d|d}� �ZFd~d� ZGej�d�d��d�d�� �ZHd�d�� ZIejj&e�oej'j(d5kd6d7�d�d�� �ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOej�d�d��ej�d�d��ej�d�dd�g�d�d�� ���ZPej�dd��d�d�� �ZQej�d2d�gd�d�d�d�d�d�d�d�d�d�d�eRd��fd�d�eRd��fd�g�ej�d�d�d�g�d�d�� ��ZSej�d2e
j8d_ d� d� �d�d�� �ZTej�d2d��d�d�� �ZUejj&e�o@ej'j(d5kd6d7�ej�d2d��ej�d�g d���d�d�� ���ZVejj&e�o�ej'j(d5kd6d7�ej�d2e
j8d_ d� d� �d�d�� ��ZWejj&e�o�ej'j(d5kd6d7�ej�d2e
j8d_ d� d� �d�d ��ZXej�d�d�d�gd�d�gg�d�dDŽ �ZYej�d�d�d�gd�d�gd�d�gd�d�gg�d�d̈́ �ZZej�d�g dϢ�d�dф �Z[d�dӄ Z\d�dՄ Z]G d�dׄ d׃Z^d�dل Z_d�dۄ Z`d�d݄ Zad�d߄ Zbej�d,d-�d�d� �Zcej�d�d�d�d� �Zdej�d�d�d�g�ej�d�d�d�d� ��Zed�d� Zfejj&e�oej'j(d5kd6d7�d�d� �Zgd�d� ZhdS )�z�
Tests specific to `np.loadtxt` added during the move of loadtxt to be backed
by C code.
These tests complement those found in `test_io.py`.
� N)�NamedTemporaryFile�mkstemp)�StringIO)� assert_equal)�assert_array_equal� HAS_REFCOUNT�IS_PYPYc C sB t d�} t�g d�g d�g d�g d�g�}ttj| dd�|� dS ) z0Test that both 'e' and 'E' are parsed correctly.zE1.0e-1,2.0E1,3.0
4.0e-2,5.0E-1,6.0
7.0e-3,8.0E1,9.0
0.0e-4,1.0E-1,2.0)皙�����?g 4@� @)g{�G�z�?� �?� )gy�&1�|?g T@� )r r � �,�� delimiterN)r �np�arrayr �loadtxt)�data�expected� r �B/usr/lib64/python3.9/site-packages/numpy/lib/tests/test_loadtxt.py�test_scientific_notation s ��r �comment)z..z//z@-zthis is a comment:c C sD d}t |�d| ��}tj|d| d�}t|ddgddgd d
gg� d S ) Nz,# IGNORE
1.5, 2.5# ABC
3.0,4.0# XXX
5.5,6.0
�#r �r �comments� �?� @r
� @� @� @)r �replacer r r )r Zcontent�txt�ar r r �test_comment_multiple_chars" s r& c C sH t d�} t�dtjfdtjfddtjfg�}tjg d�|d�}| ||fS )z|
Fixture providing hetergeneous input data with a structured dtype, along
with the associated structured array.
zo1000;2.4;alpha;-34
2000;3.1;beta;29
3500;9.9;gamma;120
4090;8.1;delta;0
5001;4.4;epsilon;-99
6543;7.8;omega;-1
�f0�f1��f2ZS7�f3�)i� g333333@Zalphai���)i� g������@Zbeta� )i�
g������#@Zgamma�x )i� g333333 @Zdeltar )i� g������@�epsiloni����)i� g333333 @Zomega�����dtype)r r r2 �uint16�float64�int8r )r r2 r r r r �mixed_types_structured* s �
�� r6 �skiprows)r � r � c C s2 |\}}}t j||d| d�}t||| d � � d S )N�;)r2 r r7 �r r r )r7 r6 r r2 r r% r r r �1test_structured_dtype_and_skiprows_no_empty_linesK s
r< c C s` | \}}}t j||ddd�\}}}}t||d � t||d � t||d � t||d � d S )Nr: T)r2 r �unpackr' r( r* r+ r; )r6 r r2 r r% �b�c�dr r r �test_unpack_structuredS s
rA c C sB t �ddg�} td�}t jddg| d�}tt j|d| d�|� d S ) N)r% �u1r )r> rB r z0,1,2,3
6,7,8,9
��r r8 �r r9 ))r � )� r
r1 r �r r2 �r r2 r r r r �r2 r r r r r � test_structured_dtype_with_shape] s rK c C s< t �dg�} td�}t jdg| d�}tt j|| d�|� d S )N)r% rB )r r z0 1 2 3
)rC r1 rI rJ r r r �&test_structured_dtype_with_multi_shaped s rL c C sr t �dtfdtfg�} t �dtfd| dfg�}td�}t jddd gfd
d d gfg|d
�}tt j||dd�|� d S )N�x�y�codeZpoints)r z100,1,2,3,4
200,5,6,7,8
�d )� �?� @)r
r �� )� @r" )g @g @r1 r �r2 r )r r2 �float�intr r r r )Zpoint�dtr r r r r � test_nested_structured_subarrayk s
��rY c C sB t jddd�} td�}t jddg| d�}tt j|d| d �|� d S )
Nzi1, i4, i1, i4, i1, i4T�Zalignz1,2,3,4,5,6
7,8,9,10,11,12
)r8 r r9 � � r )rF rG r
�
� � r1 r rH rI )rX r r r r r � test_structured_dtype_offsetsz s r` �param)r7 �max_rowsc C sF t jtdd��$ tjdi | di�� W d � n1 s80 Y dS )z;skiprows and max_rows should raise for negative parameters.z argument must be nonnegative��match�foo.bar���N)re ��pytest�raises�
ValueErrorr r �ra r r r �"test_exception_negative_row_limits� s rl c C sF t jtdd��$ tjdi | di�� W d � n1 s80 Y d S )Nzargument must be an integerrc re rQ )re �rh ri � TypeErrorr r rk r r r �$test_exception_noninteger_row_limits� s ro z data, shape)z
1 2 3 4 5
)r8 r\ )z
1
2
3
4
5
)r\ r8 c C s� t �g d��}|�|�}tt jt| �td�|� tt jt| �tdd�|� tt jt| �tdd�|� tt jt| �tdd�|� d S )N)r8 r r9 r[ r\ r1 r )r2 �ndminr8 r )r r Zreshaper r r rW )r �shapeZarrZarr2dr r r � test_ndmin_single_row_or_col� s
rr �badval)r0 r9 Nzplate of shrimpc C s@ t jtdd�� tjd| d� W d � n1 s20 Y d S )Nz Illegal value of ndmin keywordrc re )rp rg )rs r r r �test_bad_ndmin� s rt �ws)� � u � u c C sf t d| � d| � d| � d| � d| � d� �}t�g d�g d�g d �g d
�g�}ttj|td d d �|� d S )
Nz1 2z30
z
4 5 60z
z
7 8 z 90
# comment
3 2 1�r8 r � �r[ r\ �< �rF rG �Z �r9 r r8 r �r2 r r �r r r r r rW )ru r$ r r r r � test_blank_lines_spaces_delimit� s ���� "�r� c C sF t d�} t�g d�g d�g d�g d�g�}ttj| tddd�|� d S ) Nz&1,2,30
4,5,60
7,8,90
# comment
3,2,1ry r{ r} r r r r� r� )r$ r r r r �!test_blank_lines_normal_delimiter� s
"�r� r2 c C sJ t d�}tj|| ddd�}t|j| � t|tjddgddgg| d �� d S )
Nz1.5,2.5
3.0,4.0
5.5,6.0r r )r2 r rb z1.5z2.5z3.0z4.0r1 )r r r r r2 r �r2 r$ �resr r r �test_maxrows_no_blank_lines� s r� )rF r9 rG z PyPy bug in error formatting)�reasonZf8�i2c C sV t d�}d| � d�}tjt|d�� tj|| dd� W d � n1 sH0 Y d S )Nz
1,2
3,XXX
5,6z"could not convert string 'XXX' to z at row 1, column 2rc r rU �r rh ri rj r r )r2 r$ �msgr r r �!test_exception_message_bad_values� s r� c C sV t d�} ddd� i}t�ddgdtjgdd gg�}tj| tjd
|d d �}t||� d S ) Nz1.5,2.5
3.0,XXX
5.5,6.0r0 c S s | dkrt jS t| �S �NZXXX�r �nanrV ��sr r r �<lambda>� � z2test_converters_negative_indices.<locals>.<lambda>r r r
r! r" r )r2 r �
converters�encoding)r r r r� r r4 r �r$ �convr r� r r r � test_converters_negative_indices� s �r� c C s� t d�} ddd� i}t�ddgdtjgdd gg�}tj| tjd
|d dgd d �}t||� tjt d
�d
d dgddd� id�}t|d dgd dgg� d S )Nz$1.5,2.5,3.5
3.0,4.0,XXX
5.5,6.0,7.5
r0 c S s | dkrt jS t| �S r� r� r� r r r r� � r� z?test_converters_negative_indices_with_usecols.<locals>.<lambda>r g @r
r! g @r r )r2 r r� �usecolsr� z0,1,2
0,1,2,3,4c S s dS )Nr0 r �rM r r r r� � r� )r r� r� )r r r r� r r4 r r r� r r r �-test_converters_negative_indices_with_usecols� s �
�r� c C s� t d�} t�ddgddgddgg�}tj| tdddgd�}t||� t d�} tjtdd��& tj| tdddgd� W d � n1 s�0 Y d S ) N�$0,0,XXX
0,XXX,0,XXX
0,XXX,XXX,0,XXX
r r ����r2 r r� z0,0,XXX
0
0,XXX,XXX,0,XXX
z/invalid column index -2 at row 2 with 1 columnsrc ) r r r r rV r rh ri rj �r$ r r� r r r �test_ragged_usecols� s
�r� c C s$ t d�} tj| t�g �dg d�}d S )Nr� r r� )r r r r2 �r$ r� r r r �test_empty_usecols s r� �c1)r% u のu 🫕�c2c C sn t d| � d|� d��}tj|t�d�dd�}tjd| � d��d�d|� d��d�gt�d�d �}t||� d S )
Nza,z ,c,1.0
e,z,2.0,g�U12r rU z,c,1.0ze,r1 )r r r r2 r �splitr )r� r� r$ r� r r r r � test_large_unicode_characters s "�r� c C sX t d�} ddd� i}tj| t�d�|dd d�}t�dd gd
d gd d
gg�}t||� d S )Nu cat,dog
αβγ,δεζ
abc,def
r c S s | � � S �N��upperr� r r r r� r� z-test_unicode_with_converter.<locals>.<lambda>r� r )r2 r� r r� ZCATZdogu ΑΒΓu δεζ�ABC�def)r r r r2 r r �r$ r� r� r r r r �test_unicode_with_converter s � r� c C sh t d�} t�dtjfdtjfdg�}dd� dd� d�}tj| |d |d
�}tjg d �|d �}t||� d S )
Nz$1.5,2.5,Abc
3.0,4.0,dEf
5.5,6.0,ghI
�m�r)rO �U8c S s t dt| � �S )Nr] )rW rV r� r r r r� * r� z6test_converter_with_structured_dtype.<locals>.<lambda>c S s | � � S r� r� r� r r r r� * r� )r r0 r )r2 r r� ))� r r� )rz r �DEF)�7 r" ZGHIr1 )r r r2 Zint32�float32r r r )r$ rX r� r� r r r r �$test_converter_with_structured_dtype' s �r� c C sJ t d�} tj}tj| t�d�|dd�}t�ddgddgg�}t||� d S )
aL
With the default 'bytes' encoding, tokens are encoded prior to being
passed to the converter. This means that the output of the converter may
be bytes instead of unicode as expected by `read_rows`.
This test checks that outputs from the above scenario are properly decoded
prior to parsing by `read_rows`.
zabc,def
rst,xyzZU3r )r2 r� r r� r� ZRSTZXYZN)r �bytesr� r r r2 r r r� r r r �!test_converter_with_unicode_dtype2 s �r� c C sN dd } | d d� d } t | d �}tj|dtd�}t|t�dd gd
�� d S ) Nz 1.5, 2.5,�P� r0 �
r r rH r r )r r� )r r r rV r Ztile)�rowr$ r� r r r �test_read_huge_rowC s
r� ZedfgFDGc C s& d}t �| �} t j|g| d�d }d S )Na� 0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000.123456789r1 r )r r2 r )r2 �field�valuer r r �test_huge_floatK s
r� )� given_dtype�expected_dtype�S�S5�UZU5c C sR t d�}tj|| dd�}tjddgddgdd gg|d
�}t||� t|j|� d S ) z�
The given dtype is just 'S' or 'U' with no length. In these cases, the
length of the resulting dtype is determined by the longest string found
in the file.
zAAA,5-1
BBBBB,0-3
C,4-9
r rU ZAAAz5-1ZBBBBBz0-3�Cz4-9r1 N)r r r r r r2 )r� r� r$ r� r r r r �test_string_no_length_givenT s
�
r� c C sB g d�} t d�| ��}t�|�}t�dd� | D ��}t||� dS )z�
Some tests that the conversion to float64 works as accurately as the
Python built-in `float` function. In a naive version of the float parser,
these strings resulted in values that were off by an ULP or two.
)z0.9999999999999999z9876543210.123456z5.43215432154321e+300z0.901z0.333r� c S s g | ] }t |��qS r )rV )�.0r� r r r �
<listcomp>y r� z)test_float_conversion.<locals>.<listcomp>N)r �joinr r r r )�stringsr$ r� r r r r �test_float_conversionj s
r� c C sP t d�} tj| tdd�}t|ddgddgg� t|�tj�ddgddgg� d S )Nz 1, 0
10, -1r rU TFr8 r )r r r �boolr �viewZuint8r� r r r � test_bool} s r� Z
AllIntegerz0error:.*integer via a float.*:DeprecationWarningc C s� t �| �} | jdkrPt�t�� t jdg| d� W d � qP1 sD0 Y n dD ]D}t�t��& t j|� d�g| d� W d � qT1 s�0 Y qTd S )N�uz-1
r1 )z++z+-z--z-+z2
)r r2 �kindrh ri rj r )r2 �signr r r �test_integer_signs� s
0 r� c C sF t d�}t�t�� tj|| dd� W d � n1 s80 Y d S )Nz1.0, 2.1, 3.7
4, 5, 6r rU r� )r2 r$ r r r �%test_implicit_cast_float_to_int_fails� s r� � with_parens)FTc C sX d}|s |� dd�� dd�}tjt|�| dd�}tjg d�g d�g| d �}t||� d S )
Nz*(1.0-2.5j),3.75,(7+-5.0j)
(4),(-19e2j),(0)�(� �)r rU )y �? �g @y @ �)r y � ���r r1 )r# r r r r r )r2 r� r� r� r r r r �test_complex_parsing� s �r� c C sJ dd� } t j| � tdd�}t �g d�g d�g d�g d�g�}t||� d S ) Nc s s0 t d�D ]"} | � dd| � d| d � �V qd S )Nr[ r r ��range��ir r r �gen� s z%test_read_from_generator.<locals>.genr rU )r r r )r8 r r8 )r r[ r[ )r9 r r
)r r rW r r �r� r� r r r r �test_read_from_generator� s "r� c C s: dd� } t j| � ddd�}t jg d�dd�}t||� d S )Nc s s& t d�D ]} | � d| d � �V qd S )Nr9 rv r[ r� r� r r r r� � s z/test_read_from_generator_multitype.<locals>.genzi, drv rU ))r g )r8 g �?)r r r1 )r r r r r� r r r �"test_read_from_generator_multitype� s r� c C sL dd� } t jtdd��" tj| � ddd� W d � n1 s>0 Y d S )Nc s s dD ]
} | V qd S )N)z1,2s 3, 5i�1 r )�entryr r r r� � s z)test_read_from_bad_generator.<locals>.gen�&non-string returned while reading datarc zi, ir rU rm )r� r r r � test_read_from_bad_generator� s
�r� zPython lacks refcountsc sj t � �d� � �fdd�} tdd �}tjtdd��$ tj|t d| id� W d � n1 s\0 Y d S ) Nr c s � dkrt d��� d7 � �S )Ni� zfailed half-way through!r8 )rj r� �Z already_read�sentinelr r r� � s z/test_object_cleanup_on_read_error.<locals>.convzx
i' zat row 5000, column 1rc )r2 r� )�objectr rh ri rj r r )r� r$ r r� r �!test_object_cleanup_on_read_error� s 2r� c C sD t d�} t�t�� tj| dd� W d � n1 s60 Y dS )z9Test exception when a character cannot be encoded as 'S'.u –r� r1 Nr� )r r r r �#test_character_not_bytes_compatible� s r� r� r� c C sH d}t jt|d��" tjtd�| d� W d � n1 s:0 Y d S )Nz\converters must be a dictionary mapping columns to converter functions or a single callable.rc �1 2
3 4�r� �rh ri rn r r r )r� r� r r r �test_invalid_converter� s �r� c C s� t jtdd��& tjtd�dtid� W d � n1 s:0 Y t jtdd��( tjtd�dtidd� W d � n1 s�0 Y d S )Nzkeys of the converters dictrc r� r% r� r )r� r� )rh ri rn r r r rW r r r r �+test_converters_dict_raises_non_integer_key� s 4r� � bad_col_ind)r9 rf c C sL t d�}tjtdd��" tj|| tid� W d � n1 s>0 Y d S )Nr� z converter specified for columnrc r� )r rh ri rj r r rW )r� r r r r �'test_converters_dict_raises_non_col_key� s r� c C sH t jtdd��& tjtd�ddid� W d � n1 s:0 Y d S )Nz4values of the converters dictionary must be callablerc r� r r8 r� r� r r r r �,test_converters_dict_raises_val_not_callable s �r� �q)�"�'�`c
C sr t | � d| � d| � d| � d| � d| � d� �}t�ddtjfg�}tjg d �|d
�}tj||d | d �}t||� d S )
N�alpha, xz, 2.5
�beta, yz, 4.5
�gamma, zz, 5.0
)r' r� r( ))r� r )r� g @)r� rT r1 r �r2 r � quotechar)r r r2 r4 r r r )r� r$ r2 r r� r r r �test_quoted_field s &��r� c C s� t d�} t�ddtjfdtjfg�}tjtdd�� tj| |dd� W d � n1 sX0 Y | �d
� tj d g|d �}tj| |dd
d�}t
||� d S )z1Support for quoted fields is disabled by default.z"lat,long", 45, 30
)r' ZU24r( r* z the number of columns changedrc r rU Nr )zlat,longg �F@g >@r1 r� r� ) r r r2 r4 rh ri rj r �seekr r )r$ r2 r r� r r r �test_quote_support_default s .
r� c C sN t d�} d}tjt|d�� tj| ddd� W d � n1 s@0 Y d S )N�1,2
3,4z,.*must be a single unicode character or Nonerc r z''�r r� �r rh ri rn r r )r$ r� r r r � test_quotechar_multichar_error) s r� c C s� t d�} d}tjt|d��" tj| dddd� W d � n1 sB0 Y tjt|d��& tj| ddd gdd� W d � n1 s�0 Y tj| dd
d d�}t|d d
gddgg� d S )Nr� zWwhen multiple comments or a multi-character comment is given, quotes are not supported.rc r Z123r� )r r r� r �%)r r� r8 r r9 r[ )r rh ri rj r r r )r$ r� r� r r r �'test_comment_multichar_error_with_quote2 s �04r c C s^ t d�} t�dtjfdtjfddtjfg�}tjg d�|d�}tj| |dd d
�}t||� d S ) Nz{1000;2.4;'alpha';-34
2000;3.1;'beta';29
3500;9.9;'gamma';120
4090;8.1;'delta';0
5001;4.4;'epsilon';-99
6543;7.8;'omega';-1
r' r( r) r+ r, r1 r: r� r� ) r r r2 r3 r4 r5 r r r )r r2 r r� r r r �!test_structured_dtype_with_quotesB s �
�� r c C s: t d�} tjg d�dd�}tj| dddd�}t||� d S )Nz 1
"4"
""��1�4r� �U1r1 r r� �r r2 r� �r r r r r r� r r r � test_quoted_field_is_not_empty_ s r c C s: t d�} tjg d�dd�}tj| dddd�}t||� d S )Nz1
"4"
"r r r1 r r� r r r� r r r �(test_quoted_field_is_not_empty_nonstricte s r
c C s6 t d�} tjddd�}tj| dddd�}t||� d S )Nz "Hello, my name is ""Monty""!"zHello, my name is "Monty"!ZU40r1 r r� r� r r� r r r �"test_consecutive_quotechar_escapedm s r r )r� z
z# 1 2 3
# 4 5 6
rp )r r8 r r� �r8 r r9 c C s� |durd}n|dkr d}nd}t | �}tjtdd�� tj|||d�}W d � n1 s\0 Y td d
��b}|�| � |�d � tjtdd�� tj|||d�}W d � n1 s�0 Y W d � n1 s�0 Y dS ) zDCheck that a UserWarning is emitted when no data is read from input.N)r r9 r rD )r �input contained no datarc )rp r� �w)�moder ) r rh �warns� UserWarningr r r �writer� )r rp r� Zexpected_shaper$ r� �fhr r r �test_warn_on_no_datat s .
.r rE c C sL d}t |�}tjtdd�� tj|| d� W d � n1 s>0 Y d S )Nz 1 2 3
4 5 6r
rc )r7 )r rh r r r r )r7 r r$ r r r �test_warn_on_skipped_data� s r r� )r� r8 )Zu2r8 )Zi4� )Zu4r )Zi8�
)�u8r )Zfloat16g#�{ ?)r� g�� . �7)� complex64y�� . �7���`F$:)r4 g��
complex128ywfUD3":�)Z
longdoubler Z clongdoubleyp`P@0 �B �B)ZU2u �swapTFc C sR t �| �} d|� d�g}|r$| �� } t jdd| fgdd�}t j||dd dd �}d S )
Nzx,r� )r% ZS1r> FrZ r r8 )r2 r r� rb )r r2 Z newbyteorderr )r2 r� r r Zfull_dtr� r r r � test_byteswapping_and_unaligned� s
�r ZefdFD�?c C s8 t d�}tj|| ddd�}t|t�ddg��| �� d S )Nu 3 ," 2
"r r� r� r9 r )r r r r r Zastyper� r r r �!test_unicode_whitespace_stripping� s r ZFDc C sB d}||� dd�g}tj|| dd�}t|t�g d�gd �� d S )Nz0 1 , 2+3j , ( 4+5j ), ( 6+-7j ) , 8j , ( 9j )
rv u r rU )r8 y @ @y @ @y @ �y @y "@r )r# r r r r )r2 �liner r� r r r �)test_unicode_whitespace_stripping_complex� s r r� )z1 +2jz1+ 2jz1+2 jz1+-+3z(1jz(1z(1+2jz1+2j)c C sD t �t��& tj|d g| dd� W d � n1 s60 Y d S )Nr� r rU rg )r2 r� r r r �test_bad_complex� s r! ZefgdFDGc C sX | � � dkrt�d� t�t��$ tjdg| ddd� W d � n1 sJ0 Y d S )N�gz0longdouble/clongdouble assignment may misbehave.z1 r r� r� )�lowerrh �xfailri rj r r r1 r r r �test_nul_character_error� s
r% c C sb | dkrt �d� | dv r$t �d� t �t�� tjdg| d� W d � n1 sT0 Y d S )N�ez5half assignment currently uses Python float converterZeGz2clongdouble assignment is buggy (uses `complex`?).z1_1
r1 )rh �skipr$ ri rj r r r1 r r r �test_no_thousands_support� s
r( z1,2
z2
,3
z2
,3
c C sD d}t jt|d�� tj| dd� W d � n1 s60 Y d S )Nz7Found an unquoted embedded newline within a single linerc r r rg )r r� r r r � test_bad_newline_in_iterator� s r) z2,3
z'2
',3
z'2
',3
z'2
',3
c C s* t j| ddd�}t|ddgddgg� d S )Nr r� r� rQ rR r
r; )r r� r r r � test_good_newline_in_iterator� s r* �newline)r� �
�
c sT g d�}� fdd�|D �}t j|tddd�}t|dd� � �gd d
� � d�gg� d S ) N)z1,"2
"
z3,"4
z1"
c s g | ]}|� d � ��qS )r� )r# )r� r� �r+ r r r� r� z2test_universal_newlines_quoted.<locals>.<listcomp>r r� r� r �2�3r )r r r� r )r+ r r� r r. r � test_universal_newlines_quoted s r1 c C s@ t jddgdd�} t| g d�g d�g� t jddgd td
�} d S ) Nz1 2 3
z4 5 6� r r )r[ r\ r z1 ,2 ,3
z4 ,5 ,6r rH )r r r r� )r� r r r �test_null_character s
�r3 c C sR G dd� d�} t jtdd��" tj| � tdd� W d � n1 sD0 Y d S )Nc @ s e Zd Zdd� Zdd� ZdS )z:test_iterator_fails_getting_next_line.<locals>.BadSequencec S s dS )NrP r ��selfr r r �__len__# s zBtest_iterator_fails_getting_next_line.<locals>.BadSequence.__len__c S s" |dkrt d��|� d|d � �S )N�2 �Bad things happened!z, r8 )� RuntimeError)r5 �itemr r r � __getitem__& s zFtest_iterator_fails_getting_next_line.<locals>.BadSequence.__getitem__N)�__name__�
__module__� __qualname__r6 r; r r r r � BadSequence" s r? r8 rc r rU )rh ri r9 r r rW )r? r r r �%test_iterator_fails_getting_next_line! s r@ c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze j
� d
g d��dd� �Z dS )�TestCReaderUnitTestsc C sN t jtdd��, tjjjt� t�d�dd� W d � n1 s@0 Y d S )Nz.*readrc r� T�r2 �filelike) rh ri �AttributeErrorr �core�_multiarray_umath�_load_from_fileliker� r2 r4 r r r �test_not_an_filelike2 s �z)TestCReaderUnitTests.test_not_an_filelikec C s\ G dd� d�}t jtdd��, tjjj|� t�d�dd� W d � n1 sN0 Y d S )Nc @ s e Zd ZdZdd� ZdS )zBTestCReaderUnitTests.test_filelike_read_fails.<locals>.BadFileLiker c S s$ | j d7 _ | j dkr td��dS )Nr8 � � Bad bad bad!z1,2,3
)�counterr9 �r5 �sizer r r �read> s
zGTestCReaderUnitTests.test_filelike_read_fails.<locals>.BadFileLike.readN�r< r= r> rK rN r r r r � BadFileLike; s rP rJ rc r� TrB )rh ri r9 r rE rF rG r2 �r5 rP r r r �test_filelike_read_fails7 s
�z-TestCReaderUnitTests.test_filelike_read_failsc C s\ G dd� d�}t jtdd��, tjjj|� t�d�dd� W d � n1 sN0 Y d S )Nc @ s e Zd ZdZdd� ZdS )z@TestCReaderUnitTests.test_filelike_bad_read.<locals>.BadFileLiker c S s dS )Ni� r rL r r r rN P s zETestCReaderUnitTests.test_filelike_bad_read.<locals>.BadFileLike.readNrO r r r r rP M s rP r� rc r� TrB )rh ri rn r rE rF rG r2 rQ r r r �test_filelike_bad_readH s ��z+TestCReaderUnitTests.test_filelike_bad_readc C sN t jtdd��, tjjjt� t�d�dd� W d � n1 s@0 Y d S )Nz/error reading from object, expected an iterablerc r� FrB � rh ri rn r rE rF rG r� r2 r4 r r r �test_not_an_iterX s ��z%TestCReaderUnitTests.test_not_an_iterc C sH t jtdd��& tjjjt� ddd� W d � n1 s:0 Y d S )Nzinternal error: dtype mustrc r� FrB )rh ri rn r rE rF rG r� r4 r r r �
test_bad_type^ s �z"TestCReaderUnitTests.test_bad_typec C sP t jtdd��. tjjjt� t�d�ddd� W d � n1 sB0 Y d S )Nzencoding must be a unicoderc r� F�{ )r2 rC r� rT r4 r r r �test_bad_encodingc s �z&TestCReaderUnitTests.test_bad_encodingr+ )r, r� r- c C s\ t d�d|�dd�}tjjj|t�d�dddd d
�}t|d d �d f d d
|� �ddg� d S )Nz0
1
"2
"
3
4 #
r� r� r. ZU10Tr� r r8 )r2 rC �quoter Z skiplinesr r r/ r0 z4 )r r# r rE rF rG r2 r )r5 r+ r r� r r r � test_manual_universal_newlinesh s � �z3TestCReaderUnitTests.test_manual_universal_newlinesN)
r< r= r> rH rR rS rU rV rX rh �mark� parametrizerZ r r r r rA / s rA c C sF t jtdd��$ tjtd�ddd� W d � n1 s80 Y d S )N�".*control characters.*incompatiblerc �1, 2, 3r r r� r r r r �'test_delimiter_comment_collision_raisesx s r_ c C sF t jtdd��$ tjtd�ddd� W d � n1 s80 Y d S )Nr] rc r^ r r� r� r r r r �)test_delimiter_quotechar_collision_raises} s r` c C sF t jtdd��$ tjtd�ddd� W d � n1 s80 Y d S )Nr] rc �1 2 3r )r r� r� r r r r �'test_comment_quotechar_collision_raises� s rb c C sJ t jtdd��( tjtd�dddgd� W d � n1 s<0 Y d S )Nz0Comment characters.*cannot include the delimiterrc r^ r r r r� r r r r �5test_delimiter_and_multiple_comments_collision_raises� s �rc c C s� t jtdd��< tjtd| � d| � d| � d| � d� �| d� W d � n1 sP0 Y t jtdd��< tjtd| � d| � d| � d| � d� �| d � W d � n1 s�0 Y d S )
Nr] rc r r/ z3
4�5z6
�r �r� r� )ru r r r �,test_collision_with_default_delimiter_raises� s Jrg �nl)r� r, c C s� t d| � d| � d| � | � d| � d| � d| � | � ��}d}tjt|d�� tj|| d � W d � n1 sj0 Y tjt|d�� tj|| d
� W d � n1 s�0 Y tjt|d�� tj|| d � W d � n1 s�0 Y d S ) Nr r/ r0 r rd �6z&control character.*cannot be a newlinerc r re rf r� )rh r$ r� r r r �%test_control_character_newline_raises� s 4,,rj )� generic_data�
long_datum�unitless_dtyper� )z2012-03z
2013-01-15ZM8zM8[D])�
spam-a-lot�tis_but_a_scratchr� �U17�nrows)r] r� �`� c C s� | gd |g }t j||d�}td�|��}t j||d�}t||� t� \} }
t�| � t |
d�� } | �
d�|�� W d � n1 s�0 Y t j|
|d�}t� |
� t||� dS )z�Check that the correct unit (e.g. month, day, second) is discovered from
the data when a user specifies a unitless datetime.r� r1 r� r N� r r r r� r r r �os�close�openr �remove) rk rl rm r� rq r r r$ r% �fd�fnamer r r r � test_parametric_unit_discovery� s
.
rz c C s� dgd dg } t jdgd dg dd�}dd� }td �| ��}t j|d
|d d �}t||� t� \}}t�|� t |d �� }|�
d �| �� W d � n1 s�0 Y t j|d
|d d �}t� |� t||� d S )
Nrn rr ZXXXtis_but_a_scratchro rp r1 c S s
| � d�S r� )�stripr� r r r r� � r� z>test_str_dtype_unit_discovery_with_converter.<locals>.<lambda>r� r� )r2 r� r� r rs )r r r� r$ r% rx ry r r r r �,test_str_dtype_unit_discovery_with_converter� s �
.
r| c C s t jtdd��" tjtd�dd� W d � n1 s60 Y t jtdd��" tjtd�dd� W d � n1 sv0 Y t jtdd��" tjtd�dd� W d � n1 s�0 Y t jtdd��& tjtd�d dgd� W d � n1 s�0 Y d S )
Nz#Text reading control character mustrc ra r� r rf z"comments cannot be an empty stringre r )rh ri rn r r r rj r r r r � test_control_character_empty� s 000r} c C s& t jtd�ddd�} t| g d�� dS )z<Byte control characters (comments, delimiter) are supported.z
#header
1,2,3� #� ,)r r r N)r r r r )r% r r r � test_control_characters_as_bytes� s r� )i�__doc__�sysrt rh Ztempfiler r �ior Znumpyr Znumpy.ma.testutilsr Z
numpy.testingr r r r r[ r\ r& Zfixturer6 r< rA rK rL rY r` rl ro rr rt r� r� rV r� r� Zskipif�implementation�versionr2 r� r� r� r� r� r� r� r� r� r� r� r� r� r� Z typecodes�filterwarningsr� r� r r r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r r r �reprr r r r! r% r( r) r* r1 r3 r@ rA r_ r` rb rc rg rj rz r| r} r� r r r r �<module> s�
��
�
�
��
�
�
�
�
�
� �
� ��� �� �
�
I�
�� �
Mini Shell