Edit File: test__iotools.cpython-37.pyc
B ��Fd�4 � @ s� d dl mZmZmZ d dlZd dlZd dlmZ d dlZ d dl m Z m Z m Z mZmZmZ d dlmZmZmZmZmZmZ G dd� de �ZG dd � d e �Zd d � ZG d d � d e �ZG dd� de �Z e dkr�e � dS )� )�division�absolute_import�print_functionN)�date)�run_module_suite�TestCase�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 delimiters 1 2 3 4 5 # test� 1� 2� 3� 4� 5� N)r r )�self�strg�test� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/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 delimiters 1 2 3 4 5 # test� r r r r � r s s 1 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 delimiters 1 2 3 4 5 6� r r r r s 5 6s 1 2 3 4 5 6s 1 2s 3 4N)r r )r r r r r r �test_tab_delimiter% s z#TestLineSplitter.test_tab_delimiterc C sP d}t d�|�}t|ddddddg� d }t d�|�}t|ddddddg� d S ) z Test LineSplitter on delimiters 1,2,3,4,,5� ,r r r r r r s 1,2,3,4,,5 # testN)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 fieldss 1 2 3 4 5 # test� r r r r r r s 1 3 4 5 6# test� s 1 3 4 5 6� N)r r )r r r r r r �test_constant_fixed_width8 s z*TestLineSplitter.test_constant_fixed_widthc C sF d}t d�|�}t|ddddg� d}t d�|�}t|ddd g� d S ) Ns 1 3 4 5 6# test)r& � r* r& r r s 4 5� 6)r* r* � s 3 4s 5 6)r r )r r r r r r �test_variable_fixed_widthF 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 � �|�}t|ddddg� t dd��|�}t|dddd g� t d d��|�}t|dddd g� t d d��|�}t|dd ddg� ttt d d� dS )zTest case sensitivity�A�a�b�cF)�case_sensitiveZA_1�B�C�upper�lowerZa_1ZfoobarN)r �validater r � ValueError)r �namesr r r r �test_case_sensitivityT s z'TestNameValidator.test_case_sensitivityc C s< ddddg}t dddgd�}|�|�}t|dddd g� d S ) zTest excludelist�dates�dataz Other Data�mask)� excludelistZdates_Zdata_Z Other_DataZmask_N)r r< r )r r> � validatorr r r r �test_excludelistc s z"TestNameValidator.test_excludelistc C sj d}t � }t||�dddg� d}t||�dddg� d}t||�dddg� d}t||�d dd g� d S ) zTest validate missing names)r4 r5 r6 r4 r5 r6 )r r5 r6 �f0)r4 r5 r )r rF r �f1�f2N)r r )r �namelistrD r r r �test_missing_namesj s z$TestNameValidator.test_missing_namesc C s> d}t � }t||dd�d� t||ddd�dd d d d g� d S )zTest validate nb names)r4 r5 r6 � )�nbfields)r4 � zg%i)rL � defaultfmtr4 r5 r6 Zg0�g1N)r r )r rI rD r r r �test_validate_nb_namesv s z(TestNameValidator.test_validate_nb_namesc C s6 d}t � }t||�dk� t||dd�dddg� dS )zTest validate no namesNr& )rL rF rG rH )r r r )r rI rD r r r �test_validate_wo_names~ s z(TestNameValidator.test_validate_wo_namesN)r. r/ r0 r? rE rJ rP rQ r r r r r2 R s r2 c C sH t jd dkr,tt�| �d�d�d d� � S tt�| d�d d� � S d S )Nr r& �latin1z%Y-%m-%d)�sys� version_infor �time�strptime�decode)�sr r r �_bytes_to_date� s rY 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��rK 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 | � t|�d �d � t|jt|j�d � dS )zTests the upgrade method.r � 0rK Ns 17179869184l s 0.g � s 0jZ0jr& � a)r r r\ �upgrade�numpy.core.numeric�core�numericr[ �dtype�integer�itemsize�int64r �complex�len�_mapper)r r^ �nxZ status_offsetr r r � test_upgrade� s z TestStringConverter.test_upgradec C sx t dd�}|�d� t|d�d� t|d�|j� t|d�|j� t|d�|j� y |d� W n tk rr Y nX d S ) z Tests the use of missing values.)s missings missed)�missing_valuesr` r r s missings missed�missN)r rc r r] r= )r r^ r r r � test_missing� s z TestStringConverter.test_missingc C s| t }t�|tddd�� t|tddd��}|d�}t|tddd�� |d�}t|tddd�� |d�}t|tddd�� dS ) zTests updatemapperi� rK s 2001-01-01i� s 2009-01-01i� r N)rY r �upgrade_mapperr r )r Z dateparser�convertr r r r �test_upgrademapper� s z&TestStringConverter.test_upgrademapperc C s2 t t�}t|jd d d�d� tt|d�� dS )zAMake sure that string-to-object functions are properly recognized���r y r] N)r rY r rm r �hasattr)r �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���)rp r] s 3.14159265r )r rc r r] �type�nprg �float)r r^ 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 s N/A)r] rp N)r r[ r r] )r r^ r r r �test_keep_default_zero� s z*TestStringConverter.test_keep_default_zeroc C s& t tddd�}t|jtddg�� dS )z*Check that we're not losing missing valuesr s N/A)r] rp r N)r r[ r rp �set)r 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] s -9223372036854775807l�������� s 9223372036854775807l ���� N)r r{ rj 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] s 9223372043271415339l +&�| N)r r{ �uint64r )r r^ r� r r r �test_uint64_dtype� s z%TestStringConverter.test_uint64_dtypeN)r. r/ r0 r1 r_ ro rr ru ry r} r~ r� r� r� r r r r rZ � s rZ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestMiscFunctionsc C sj t �t j�}tt|�d� t �ddtfg�}tt|�d� t �dtfddtfdgfg�}tt|�d� dS ) zTest has_nested_dtypeF)r3 z|S3r8 r3 ZBA)ZBBz|S1TN)r{ rg r| r r r[ )r �ndtyper r r �test_has_nested_dtype s z'TestMiscFunctions.test_has_nested_dtypec C s2 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|d d d gdd�t�d d � d D ��� d!S )"zTest ndtype on dtypeszi4, f8)rF �i4)rG �f8z field_%03i)rN )Z field_000r� )Z field_001r� za, b)r> )r4 r� )r5 r� za, b, cz, br4 zf%02i)r> rN )�f00r� r3 r8 za,br5 rF za,b,crG rH r6 c S s g | ] }|t f�qS r )r| )�.0�_r r r � <listcomp>> s z5TestMiscFunctions.test_easy_dtype.<locals>.<listcomp>)r4 r5 r6 r c S s g | ] }|t f�qS r )r| )r� r� r r r r� C s )r� Zf01Zf02N)r| r r r{ rg r[ )r r� r r r �test_easy_dtype sN 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)r4 r� )r5 r� r4 )�aaz|S1)�abz|S2r5 z|S1z|S2ra r& T))r4 r3 r� ))r5 r8 r� N)r{ rg r r r| r[ )r �dtZdt_flatr r r �test_flatten_dtypeE s z$TestMiscFunctions.test_flatten_dtypeN)r. r/ r0 r� r� r� r r r r r� s 9r� �__main__) � __future__r r r rS rU �datetimer �numpyr{ Z numpy.testingr r r r r r Znumpy.lib._iotoolsr r r r r r r r2 rY rZ r� r. r r r r �<module> s @6rY