3
���h�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 sF d}t � |�}t|dddddg� t d�|�}t|dddddg� dS ) z Test LineSplitter w/o delimiterz 1 2 3 4 5 # test�1�2�3�4�5� N)r r )�self�strg�test� r �>/tmp/pip-build-5_djhm0z/numpy/numpy/lib/tests/test__iotools.py�test_no_delimiter s
z"TestLineSplitter.test_no_delimiterc C sD d}t d�|�}t|ddddddg� t d �|�}t|d
dg� d S ) zTest space delimiterz 1 2 3 4 5 # test� r r r r r r z z1 2 3 4N)r r )r r r r r r �test_space_delimiter s
z%TestLineSplitter.test_space_delimiterc C sH d}t d�|�}t|dddddg� d}t d�|�}t|d d
dg� d S ) zTest tab delimiterz 1 2 3 4 5 6� r r r r z5 6z 1 2 3 4 5 6z1 2z3 4N)r r )r r r r r r �test_tab_delimiter! s z#TestLineSplitter.test_tab_delimiterc C sz d}t d�|�}t|ddddddg� d }t d�|�}t|ddddddg� d
}t d d d
�|�}t|ddddddg� dS )z Test LineSplitter on delimiterz
1,2,3,4,,5�,r r r r r r z 1,2,3,4,,5 # tests 1,2,3,4,,5 % test� ,� %)� delimiter�commentsN)r r )r r r r r r �test_other_delimiter* s z%TestLineSplitter.test_other_delimiterc C sd d}t d�|�}t|dddddddg� d }t d
�|�}t|d g� d }t d �|�}t|d g� d
S )z'Test LineSplitter w/ fixed-width fieldsz 1 2 3 4 5 # test� r r r r r r z 1 3 4 5 6# test� z1 3 4 5 6� N)r r )r r r r r r �test_constant_fixed_width9 s z*TestLineSplitter.test_constant_fixed_widthc C sF d}t d �|�}t|ddddg� d}t d �|�}t|dd d
g� d S )
Nz 1 3 4 5 6# testr$ � r r z4 5�6� z3 4z5 6)r$ r( r( r$ )r( 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� ddddg}t � j|�}t|ddddg� t dd�j|�}t|dddd g� t d
d�j|�}t|dddd g� t d d�j|�}t|dd ddg� ttt d
d� dS )zTest case sensitivity�A�a�b�cF)Zcase_sensitiveZA_1�B�C�upper�lowerZa_1ZfoobarN)r �validater r �
ValueError)r �namesr r r r �test_case_sensitivityU s z'TestNameValidator.test_case_sensitivityc C s< ddddg}t dddgd�}|j|�}t|dddd g� d
S ) zTest excludelist�dates�dataz
Other Data�mask)Z excludelistZdates_Zdata_Z
Other_DataZmask_N)r r9 r )r r; � validatorr r r r �test_excludelistd s
z"TestNameValidator.test_excludelistc C sj d }t � }t||�dddg� d
}t||�dddg� d }t||�dddg� d }t||�dddg� dS )
zTest validate missing namesr2 r3 r4 r �f0�f1�f2N)r2 r3 r4 )r r3 r4 )r2 r3 r )r rB r )r r )r �namelistr@ r r r �test_missing_namesk s z$TestNameValidator.test_missing_namesc C s> d }t � }t||dd�d
� t||ddd�dddd d
g� d S )zTest validate nb namesr2 r3 r4 � )�nbfields� zg%i)rH �
defaultfmtZg0Zg1N)r2 r3 r4 )r2 )r r )r rE r@ r r r �test_validate_nb_namesw s
z(TestNameValidator.test_validate_nb_namesc C s6 d}t � }t||�dk� t||dd�dddg� dS )zTest validate no namesNr$ )rH rB rC rD )r r r )r rE r@ r r r �test_validate_wo_names s z(TestNameValidator.test_validate_wo_namesN)r, r- r. r< rA rF rK rL r r r r r0 S s
r0 c C s t tj| d�d d� � S )Nz%Y-%m-%dr$ )r �time�strptime)�sr r r �_bytes_to_date� s rP 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 StringConverteri�� rG Niay��iay��)r �intr �_status�default)r � converterr r r �
test_creation� s
z!TestStringConverter.test_creationc C s t � }t|jd� t|jd�d� t|jd� ddljj}t|j|j �j
|j|j �j
k �}t|jd�d� t|jd| � t |jd�d� t|jd | � t|jd
�t
d
�� t|jd | � x@dD ]8}|j|�}tt|�tk� t|d � t|jd| � q�W dS )zTests the upgrade method.r �0rG NZ 17179869184l z0.g � Z0jr$ r2 � a� )r2 rY )r r rS �upgradeZnumpy.core.numeric�core�numericrR �dtype�int_�itemsize�int64r �complexr �type�str)r rU ZnxZ
status_offsetrO �resr r r � test_upgrade� s"
z TestStringConverter.test_upgradec C sx t d d�}|jd� t|d�d� t|d�|j� t|d�|j� t|d�|j� y |d� W n tk
rr Y nX dS )
z Tests the use of missing values.�missing�missed)�missing_valuesrW r r ZmissN)rg rh )r r[ r rT r: )r rU r r r � test_missing� s
z TestStringConverter.test_missingc C s� t }tjdd� }zxtj|tddd�� t|tddd��}|d�}t|tddd�� |d�}t|tddd�� |d�}t|tddd�� W d|t_X dS ) zTests updatemapperNi� rG z
2001-01-01i� z
2009-01-01i� r )rP r �_mapperZupgrade_mapperr r )r Z
dateparserZ_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 recognizedNrT )r rk rP 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�}|jd� t|jd� t|jtjt�� t dddd�}|jd� t|jd� t|jtjt�� dS ) z+Make sure we don't lose an explicit defaultNr i� )ri rT z
3.14159265r i���i���)r r[ r rT rc �npr^ �float)r rU 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 zN/A)rT ri N)r rR r rT )r rU 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 zN/A)rT ri r N)r rR r ri )r rU 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 )rT z-9223372036854775807l ���� Z9223372036854775807Nl�������� )r rq ra r )r rU �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 )rT Z9223372043271415339l +&�| N)r rq Zuint64r )r rU rv r r r �test_uint64_dtype s z%TestStringConverter.test_uint64_dtypeN)r, r- r. r/ rV rf rj rm rp rs rt ru rw rx r r r r rQ � s %rQ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestMiscFunctionsc C sh t jt�}tt|�d� t jd
dtfg�}tt|�d� t jdtfddtfd gfg�}tt|�d� d S ) zTest has_nested_dtypeFr1 �|S3r5 ZBA�BB�|S1TN)r1 rz )r{ r| )rq r^ rr r r
rR )r �ndtyper r r �test_has_nested_dtype s
z'TestMiscFunctions.test_has_nested_dtypec C s2 t }tt|�tjt �� d}tt|�tjd d!g�� tt|dd�tjd"d#g�� d}tt|d
d �tjd$d%g�� d}tt|dd �tjd&d'g�� d}tt|dd �tjd(d)g�� tt|d dd�tjd*d+g�� dtfdt fg}tt|�tjdtfdt fg�� tt|dd �tjd tfd
t fg�� tt|d d �tjd tfdt fg�� tt|dd �tjd tfd
t fg�� tt t f}tt|�tjdtfdt fdt fg�� tt t f}tt|dd �tjd tfd
t fdt fg�� tjt �}tt|dd �tjdd� d,D ��� tjt �}tt|dddgdd�tjd d� d-D ��� d S ).zTest ndtype on dtypeszi4, f8rB �i4rC �f8z
field_%03i)rJ � field_000� field_001za, b)r; r2 r3 za, b, cz, bzf%02i)r; rJ �f00r1 r5 za,bza,b,crD r4 c S s g | ] }|t f�qS r )rr )�.0�_r r r �
<listcomp>G s z5TestMiscFunctions.test_easy_dtype.<locals>.<listcomp>r c S s g | ] }|t f�qS r )rr )r� r� r r r r� L s �f01�f02N)rB r )rC r� )r� r )r� r� )r2 r )r3 r� )r2 r )r3 r� )rB r )r3 r� )r2 r )r� r� )r2 r3 r4 )r� r� r� )rr r r rq r^ rR )r r} r r r �test_easy_dtype sN
z!TestMiscFunctions.test_easy_dtypec C s� t jddg�}t|�}t|ttg� t jdddgfdtfg�}t|�}t|t jd�t jd�tg� t jdtdffdtd ffg�}t|�}t|ttg� t|d
�}t|tgd tgd � t jddg�}t|�}t|ttg� d
S )zTesting flatten_dtyper2 r� r3 �aa�|S1�ab�|S2rX r$ Tr1 r5 N)r2 r� )r3 r� )r� r� )r� r� �r2 r1 )r� r� �r3 r5 )r� r� )rq r^ r r rr rR )r �dtZdt_flatr r r �test_flatten_dtypeN s
z$TestMiscFunctions.test_flatten_dtypeN)r, r- r. r~ r� r� r r r r ry
s 9ry )rM �datetimer Znumpyrq Z
numpy.testingr r r r Znumpy.lib._iotoolsr r r r
r r r
r0 rP rQ ry r r r r �<module> s E6} |