Current File : //lib64/python3.9/site-packages/numpy/lib/tests/__pycache__/test__iotools.cpython-39.opt-1.pyc
a
z[yc�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 ��selfZstrg�test� r �C/usr/lib64/python3.9/site-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� ,� %)Z delimiterZcommentsNr 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+ r/ 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)Zcase_sensitive)r5 ZA_1�B�C�upper�lower)r6 Za_1r7 r8 ZfoobarN)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)rA rB rC )Z excludelist)Zdates_Zdata_Z
Other_DataZmask_N)r r= r )r r? � 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�r6 r7 r8 )r r7 r8 )�f0r7 r8 )r6 r7 r )r6 r7 rG )r rG r )�f1rG �f2N�r r �r ZnamelistrD 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 namesrF � ��nbfields)r6 � zg%i)rO �
defaultfmt)r6 r7 r8 Zg0Zg1NrJ rK 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' rN )rG rH rI )r r r rK r r r �test_validate_wo_names s z(TestNameValidator.test_validate_wo_namesN)r0 r1 r2 r@ rE rL rR rS r r r r r4 S s
r4 c C s t t�| d�d d� � S )Nz%Y-%m-%dr' )r �time�strptime)�sr r r �_bytes_to_date� s rW 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��rM 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 ]8}|�|�}tt|�tu � t|d
� t|jd| � q�dS )zTests the upgrade method.r �0rM NZ 17179869184l z0.g � Z0jr' )r6 � ar6 � )r r rZ �upgradeZnumpy.core.numericZcore�numericrY �dtypeZint_�itemsize�int64r �complexr �type�str)r r] ZnxZ
status_offsetrV �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 n typ Y n0 d S )
z Tests the use of missing values.)�missing�missed)�missing_valuesr_ r r rm rn ZmissN)r rc r r[ r> r\ 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_n|t_0 dS ) zTests updatemapperNi� rM z
2001-01-01i� z
2009-01-01i� r )rW r �_mapperZupgrade_mapperr r )r Z
dateparserZ_original_mapperZconvertr 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 recognizedNr[ )r rq rW r r �hasattr)r Z
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���)ro r[ z
3.14159265r )r rc r r[ ri �npre �floatr\ 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�r[ ro N)r rY r r[ r\ 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 ry rz r N)r rY r ro r\ 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 �r[ z-9223372036854775807l�������� Z9223372036854775807l ���� N)r rv rg r �r r] �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} Z9223372043271415339l +&�| N)r rv Zuint64r r~ r r r �test_uint64_dtype s z%TestStringConverter.test_uint64_dtypeN)r0 r1 r2 r3 r^ rl rp rr ru rx r{ r| r� r� r r r r rX � s %rX 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)r5 z|S3r9 r5 ZBA)ZBB�|S1TN)rv re rw r r
rY �r Zndtyper 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)rG �i4)rH �f8z
field_%03i)rQ )Z field_000r� )Z field_001r� za, b)r? )r6 r� �r7 r� za, b, cz, br6 zf%02i)r? rQ )�f00r� r5 r9 za,br7 rG za,b,crH rI r8 c S s g | ] }|t f�qS r �rw ��.0�_r r r �
<listcomp>G � z5TestMiscFunctions.test_easy_dtype.<locals>.<listcomp>rF )r r r c S s g | ] }|t f�qS r r� r� r r r r� L r� )r� Zf01Zf02N)rw r r rv re rY 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)r6 r� r� r6 )Zaar� )�ab�|S2r7 r� r� r` r' T))r6 r5 r� ))r7 r9 r� N)rv re r r rw rY )r ZdtZdt_flatr r r �test_flatten_dtypeN s
z$TestMiscFunctions.test_flatten_dtypeN)r0 r1 r2 r� r� r� r r r r r�
s 9r� )rT Zdatetimer Znumpyrv Z
numpy.testingr r r r Znumpy.lib._iotoolsr r r r
r r r
r4 rW rX r� r r r r �<module> s E6}
Mini Shell