File: //usr/local/lib/python3.10/dist-packages/numpy/lib/tests/__pycache__/test__iotools.cpython-310.pyc
o
���g�5 � @ s� d dl Z d dlmZ d dlZd dlmZmZmZm Z d dl
m Z m Z m
Z
mZmZmZ G dd� d�ZG dd� d�Zd d
� ZG d d � d �ZG d
d� d�ZdS )� N)�date)�assert_� assert_equal�assert_allclose�
assert_raises)� LineSplitter�
NameValidator�StringConverter�has_nested_fields�
easy_dtype�
flatten_dtypec @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�TestLineSplitterz Tests the LineSplitter class.c C s: d}t � |�}t|g d�� t d�|�}t|g d�� dS )z Test LineSplitter w/o delimiter� 1 2 3 4 5 # test)�1�2�3�4�5� N�r r ��self�strg�test� r �H/usr/local/lib/python3.10/dist-packages/numpy/lib/tests/test__iotools.py�test_no_delimiter s
z"TestLineSplitter.test_no_delimiterc C s< d}t d�|�}t|g d�� t d�|�}t|ddg� dS )zTest space delimiterr � �r r r r r r z z1 2 3 4r Nr r r r r �test_space_delimiter s
z%TestLineSplitter.test_space_delimiterc C s@ d}t d�|�}t|g d�� d}t d�|�}t|g d�� dS )zTest tab delimiterz 1 2 3 4 5 6� )r r r r �5 6z 1 2 3 4 5 6)z1 2�3 4r! Nr r r r r �test_tab_delimiter! s z#TestLineSplitter.test_tab_delimiterc C sb d}t d�|�}t|g d�� d}t d�|�}t|g d�� d}t ddd�|�}t|g d�� d S )
z Test LineSplitter on delimiterz
1,2,3,4,,5�,r z 1,2,3,4,,5 # tests 1,2,3,4,,5 % test� ,� %)� delimiter�commentsNr r r r r �test_other_delimiter* s z%TestLineSplitter.test_other_delimiterc C sZ d}t d�|�}t|g d�� d}t d�|�}t|dg� d}t d�|�}t|dg� dS ) z'Test LineSplitter w/ fixed-width fieldsz 1 2 3 4 5 # test� )r r r r r r r � 1 3 4 5 6# test� z1 3 4 5 6� Nr r r r r �test_constant_fixed_width9 s z*TestLineSplitter.test_constant_fixed_widthc C s@ d}t d�|�}t|g d�� d}t d�|�}t|g d�� d S )Nr+ )r* � r/ r* )r r z4 5�6)r/ r/ � )r r" r! r r r r r �test_variable_fixed_widthG s z*TestLineSplitter.test_variable_fixed_widthN)
�__name__�
__module__� __qualname__�__doc__r r r# r) r. r2 r r r r r
s r
c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestNameValidatorc C s� g d�}t � �|�}t|g d�� t dd��|�}t|g d�� t dd��|�}t|g d�� t dd��|�}t|g d�� ttt dd� d S )
zTest case sensitivity)�A�a�b�cF)�case_sensitive)r8 �A_1�B�C�upper�lower)r9 �a_1r: r; �foobarN)r �validater r �
ValueError)r �namesr r r r �test_case_sensitivityU s z'TestNameValidator.test_case_sensitivityc C s2 g d�}t g d�d�}|�|�}t|g d�� dS )zTest excludelist)�dates�dataz
Other Data�mask)rH rI rJ )� excludelist)�dates_�data_�
Other_Data�mask_N)r rD r )r rF � validatorr r r r �test_excludelistd s
z"TestNameValidator.test_excludelistc C sb d}t � }t||�g d�� d}t||�g d�� d}t||�g d�� d}t||�g d�� dS ) zTest validate missing names�r9 r: r; )r r: r; )�f0r: r; )r9 r: r )r9 r: rS )r rS r )�f1rS �f2N�r r �r �namelistrP r r r �test_missing_namesk s z$TestNameValidator.test_missing_namesc C s8 d}t � }t||dd�d� t||ddd�g d�� d S )
zTest validate nb namesrR � ��nbfields)r9 � zg%i)r\ �
defaultfmt)r9 r: r; �g0�g1NrV rW r r r �test_validate_nb_namesw s �z(TestNameValidator.test_validate_nb_namesc C s4 d}t � }t||�du � t||dd�g d�� dS )zTest validate no namesNr* r[ )rS rT rU )r r r rW r r r �test_validate_wo_names s z(TestNameValidator.test_validate_wo_namesN)r3 r4 r5 rG rQ rY ra rb r r r r r7 S s r7 c C s t t�| d�d d� � S )Nz%Y-%m-%dr* )r �time�strptime)�sr r r �_bytes_to_date� s rf c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�TestStringConverterzTest StringConverterc C s& t td�}t|jd� t|jd� dS )z"Test creation of a StringConverteriay��rZ N)r �intr �_status�default�r � converterr r r �
test_creation� s
z!TestStringConverter.test_creationc C s t � }t|jd� t|�d�d� t|jd� ddlm m} t|�|j �j
|�|j �j
k �}t|�d�d� t|jd| � t |�d�d� t|jd | � t|�d
�t
d
�� t|jd | � d D ] }|�|�}tt|�tu � t|d
� t|jd| � qddS )zTests the upgrade method.r �0rZ N� 17179869184l z0.g � �0jr* )r9 � ar9 � )r r ri �upgrade�numpy.core.numeric�core�numericrh �dtype�int_�itemsize�int64r �complexr �type�str)r rl �nx�
status_offsetre �resr r r � test_upgrade� s$
�z TestStringConverter.test_upgradec C sv t dd�}|�d� t|d�d� t|d�|j� t|d�|j� t|d�|j� z|d� W d S ty: Y d S w )
z Tests the use of missing values.)�missing�missed)�missing_valuesrn r r r� r� �missN)r rt r rj rE rk r r r � test_missing� s
�z TestStringConverter.test_missingc C s� t }tjdd� }z@t�|tddd�� t|tddd��}|d�}t|tddd�� |d�}t|tddd�� |d�}t|tddd�� W |t_dS |t_w ) zTests updatemapperNi� rZ z
2001-01-01i� z
2009-01-01i� r )rf r �_mapper�upgrade_mapperr r )r �
dateparser�_original_mapper�convertr r r r �test_upgrademapper� s z&TestStringConverter.test_upgrademapperc C s4 t jdd� }t t�}t|j|� tt|d�� dS )zAMake sure that string-to-object functions are properly recognizedNrj )r r� rf r r �hasattr)r �
old_mapper�convr r r �test_string_to_object� s z)TestStringConverter.test_string_to_objectc C sp t dddd�}|�d� t|jd� t|jt�t�� t dddd�}|�d� t|jd� t|jt�t�� dS )z+Make sure we don't lose an explicit defaultNr i���)r� rj z
3.14159265r )r rt r rj r} �nprx �floatrk r r r �test_keep_default� s �
�
z%TestStringConverter.test_keep_defaultc C s t tddd�}t|jd� dS )z'Check that we don't lose a default of 0r �N/A�rj r� N)r rh r rj rk r r r �test_keep_default_zero� s �z*TestStringConverter.test_keep_default_zeroc C s" t tddd�}t|jddh� dS )z*Check that we're not losing missing valuesr r� r� r N)r rh r r� rk r r r �test_keep_missing_values� s �
�z,TestStringConverter.test_keep_missing_valuesc C s: t tjdd�}d}t||�dk� d}t||�dk� dS )z/Check that int64 integer types can be specifiedr �rj z-9223372036854775807l�������� �9223372036854775807l ���� N)r r� r{ r �r rl �valr r r �test_int64_dtype� s
z$TestStringConverter.test_int64_dtypec C s& t tjdd�}d}t||�dk� dS )z0Check that uint64 integer types can be specifiedr r� �9223372043271415339l +&�| N)r r� �uint64r r� r r r �test_uint64_dtype s z%TestStringConverter.test_uint64_dtypeN)r3 r4 r5 r6 rm r� r� r� r� r� r� r� r� r� r r r r rg � s % rg c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestMiscFunctionsc C sh t �t�}tt|�d� t �ddtfg�}tt|�d� t �dtfddtfdgfg�}tt|�d� dS ) zTest has_nested_dtypeF)r8 z|S3r> r8 �BA)�BB�|S1TN)r� rx r� r r
rh �r �ndtyper r r �test_has_nested_dtype s
z'TestMiscFunctions.test_has_nested_dtypec C s0 t }tt|�t�t �� d}tt|�t�ddg�� tt|dd�t�ddg�� d}tt|dd �t�d
d g�� d}tt|d d �t�d
d g�� d}tt|d
d �t�dd g�� tt|ddd�t�d
dg�� dtfdt fg}tt|�t�dtfdt fg�� tt|dd �t�dtfdt fg�� tt|dd �t�dtfdt fg�� tt|dd �t�dtfdt fg�� tt t f}tt|�t�dtfdt fdt fg�� tt t f}tt|d d �t�dtfdt fdt fg�� t�t �}tt|d d �t�dd � d D ��� t�t �}tt|g d �dd�t�d d � d D ��� d!S )"zTest ndtype on dtypeszi4, f8)rS �i4)rT �f8z
field_%03i)r^ )� field_000r� )� field_001r� za, b)rF )r9 r� �r: r� za, b, cz, br9 zf%02i)rF r^ )�f00r� r8 r> za,br: rS za,b,crT rU r; c S � g | ]}|t f�qS r �r� ��.0�_r r r �
<listcomp>G � z5TestMiscFunctions.test_easy_dtype.<locals>.<listcomp>rR )r r r c S r� r r� r� r r r r� L r� )r� �f01�f02N)r� r r r� rx rh r� r r r �test_easy_dtype sh � � � � � � � � �
�
�
�
�z!TestMiscFunctions.test_easy_dtypec C s� t �ddg�}t|�}t|ttg� t �dddgfdtfg�}t|�}t|t �d�t �d�tg� t �dtd ffdtd
ffg�}t|�}t|ttg� t|d �}t|tgd tgd
� t �d d
g�}t|�}t|ttg� dS )zTesting flatten_dtype)r9 r� r� r9 )�aar� )�ab�|S2r: r� r� rp r* T))r9 r8 r� ))r: r> r� N)r� rx r r r� rh )r �dt�dt_flatr r r �test_flatten_dtypeN s
z$TestMiscFunctions.test_flatten_dtypeN)r3 r4 r5 r� r� r� r r r r r�
s 9r� )rc �datetimer �numpyr� �
numpy.testingr r r r �numpy.lib._iotoolsr r r r
r r r
r7 rf rg r� r r r r �<module> s E6}