U
|
P ±d7 ã @ s d dl Z d dlmZ d dlZd dlmZmZmZm Z d dl
|
m Z m Z m Z mZmZmZ G dd dZG dd dZd d
|
ZG d d d ZG d d dZdS )é 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 delimiterú 1 2 3 4 5 # testÚ1Ú2Ú3Ú4Ú5Ú N©r r ©ÚselfÚstrgÚtest© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\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 delimiterr ú r r r r r r z z1 2 3 4Nr 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 ú5 6z 1 2 3 4 5 6z1 2ú3 4Nr 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ÚcommentsNr 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 ú 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 sF d}t d|}t|ddddg d}t d|}t|ddd g d S )
|
Nr* )r) é r. r) r r z4 5Ú6)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- r1 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)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}| |¡}t|dddd g d
|
S ) zTest excludelistÚdatesÚdataz
|
Other DataÚmask)Z excludelistZdates_Zdata_Z
|
Other_DataZmask_N)r r? r )r rA Ú 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||d dd
|
g d S ) zTest validate missing names©r8 r9 r: r8 r9 r: )r r9 r: Úf0)r8 r9 r )r rI r Úf1Úf2N©r r ©r ÚnamelistrF r r r Útest_missing_namesk s z$TestNameValidator.test_missing_namesc C s> d}t }t||ddd t||ddddd d
|
d d g d S )zTest validate nb namesrH é ©Únbfields)r8 é zg%i)rR Ú
|
defaultfmtr8 r9 r: Zg0Zg1NrL rM r r r Útest_validate_nb_namesw s ÿz(TestNameValidator.test_validate_nb_namesc C s6 d}t }t||dk t||dddddg dS )zTest validate no namesNr) rQ rI rJ rK )r r r rM r r r Útest_validate_wo_names s z(TestNameValidator.test_validate_wo_namesN)r2 r3 r4 rB rG rO rU rV r r r r r6 S s
|
r6 c C s t t | d¡d d
S )Nz%Y-%m-%dr) )r ÚtimeÚstrptime)Úsr r r Ú_bytes_to_date s rZ 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þÿrP 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|tk t|d t|jd| qÈdS )zTests the upgrade method.r Ú0rP NZ 17179869184l z0.g é Z0jr) )r8 ó ar8 é )r r r] ÚupgradeZnumpy.core.numericÚcoreÚnumericr\ ÚdtypeÚint_ÚitemsizeÚint64r Úcomplexr ÚtypeÚstr)r r` ZnxZ status_offsetrY Úresr r r Ú test_upgrade s"
|
|
z TestStringConverter.test_upgradec C sx t dd}| d¡ t|dd t|d|j t|d|j t|d|j z |d W n tk
|
rr Y nX d S )
|
z Tests the use of missing values.)ÚmissingÚmissed)Úmissing_valuesrb r r rr rs ZmissN)r rf r r^ r@ r_ r r r Ú test_missing» s
|
|
z TestStringConverter.test_missingc C s t }tjdd
}zxt |tddd¡ t|tddd}|d}t|tddd |d}t|tddd |d}t|tddd W 5 |t_X dS ) zTests updatemapperNiÐ rP z
|
2001-01-01iÑ z
|
2009-01-01iÙ r )rZ 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 recognizedNr^ )r rv rZ 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üÿÿ)rt r^ z
|
3.14159265r )r rf r r^ rn Únpri Ú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^ rt N)r r\ 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 r r r N)r r\ r rt 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 r| rl 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 r| Zuint64r r r r r Útest_uint64_dtype s z%TestStringConverter.test_uint64_dtypeN)r2 r3 r4 r5 ra rq ru rx r{ r~ r r r r r r r r r[ s %r[ 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)r7 z|S3r; r7 ZBA)ÚBBú|S1TN)r| ri r} r r
|
r\ ©r Zndtyper r r Útest_has_nested_dtype s
|
z'TestMiscFunctions.test_has_nested_dtypec C s2 t }tt|t t ¡ d}tt|t ddg¡ tt|ddt ddg¡ d}tt|dd t d
|
d g¡ d}tt|d d t d
|
d g¡ d}tt|d d t dd g¡ tt|dddt 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 gddt d d d D ¡ d!S )"zTest ndtype on dtypeszi4, f8)rI Úi4)rJ Úf8z
|
field_%03i)rT )Z field_000r )Z field_001r za, b)rA )r8 r ©r9 r za, b, cz, br8 zf%02i)rA rT )Úf00r r7 r; za,br9 rI za,b,crJ rK r: c S s g | ] }|t fqS r ©r} ©Ú.0Ú_r r r Ú
|
<listcomp>G s z5TestMiscFunctions.test_easy_dtype.<locals>.<listcomp>rH r c S s g | ] }|t fqS r r r r r r r L s )r Zf01Zf02N)r} r r r| ri r\ 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)r8 r r r8 )Zaar )Úabú|S2r9 r r rc r) T))r8 r7 r ))r9 r; r N)r| ri r r r} r\ )r ÚdtZdt_flatr r r Útest_flatten_dtypeN s
|
z$TestMiscFunctions.test_flatten_dtypeN)r2 r3 r4 r r r r r r r r
|
s 9r )rW Údatetimer Únumpyr| Z numpy.testingr r r r Znumpy.lib._iotoolsr r r r
|
r r r r6 rZ r[ r r r r r Ú<module> s E6}
|
