ZE # d dl Z d dlmZ d dlZd dlmZmZmZm Z d dl
m Z m Z m
Z
mZmZmZ G d d Z G d d Zd Z G d
d Z G d d
ZdS ) N)date)assert_ assert_equalassert_allclose
assert_raises) LineSplitter
NameValidatorStringConverterhas_nested_fields
easy_dtype
flatten_dtypec 6 e Zd ZdZd Zd Zd Zd Zd Zd Z dS ) TestLineSplitterz Tests the LineSplitter class.c d} t | }t |g d t d | }t |g d dS )z Test LineSplitter w/o delimiter 1 2 3 4 5 # test)12345 Nr r selfstrgtests q/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test__iotools.pytest_no_delimiterz"TestLineSplitter.test_no_delimiter se # |~~d##T444555 |B %%T44455555 c 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 s r test_space_delimiterz%TestLineSplitter.test_space_delimiter sg # |C &&T888999!|D!!$''TIs+,,,,,r c 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 23 4r& Nr r s r test_tab_delimiterz#TestLineSplitter.test_tab_delimiter! sn &!|D!!$''T777888$!|D!!$''T33344444r c d} t d | }t |g d d} t d | }t |g d d} t dd | }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 , %) delimitercommentsNr r s r test_other_delimiterz%TestLineSplitter.test_other_delimiter* s |C &&T888999# |C &&T888999 %:|dT:::4@@T88899999r c
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 s r test_constant_fixed_widthz*TestLineSplitter.test_constant_fixed_width9 s , |At$$T<<<===) |B %%T ./000) |B %%T ./00000r c d} t d | }t |g d d} t d | }t |g d d S )Nr2 )r1 r7 r1 )r r z4 56)r7 r7 )r r' r& r r s r test_variable_fixed_widthz*TestLineSplitter.test_variable_fixed_widthG sn ))|L))$//T222333)&|I&&t,,T00011111r N)
__name__
__module__ __qualname____doc__r r# r( r/ r5 r: r r r r st ##6 6 6- - -5 5 5
:
:
: 1 1 1 2 2 2 2 2r r c , e Zd Zd Zd Zd Zd Zd ZdS )TestNameValidatorc g d}t | }t |g d t d | }t |g d t d | }t |g d t d | }t |g d t t t d d S )
zTest case sensitivity)AabcF)case_sensitive)rC A_1BCupperlower)rD a_1rE rF foobarN)r validater r
ValueError)r namesr s r test_case_sensitivityz'TestNameValidator.test_case_sensitivityU s $$$ ''..T///000 E222;;EBBT111222 G444==eDDT111222 G444==eDDT111222 j-IIIIIIr c g d}t g d }| | }t |g d dS )zTest excludelist)datesdataz
Other Datamask)rT rU rV ) excludelist)dates_data_
Other_Datamask_N)r rO r )r rQ validatorr s r test_excludelistz"TestNameValidator.test_excludelistd sT 777!.G.G.GHHH !!%((TEEEFFFFFr c
d}t }t || g d d}t || g d d}t || g d d}t || g d dS ) zTest validate missing namesrD rE rF )r rE rF )f0rE rF )rD rE r )rD rE r` )r r` r )f1r` f2Nr r r namelistr\ s r test_missing_namesz$TestNameValidator.test_missing_namesk s "!OO YYx((///:::!YYx((*:*:*:;;;!YYx((*:*:*:;;;!YYx((*<*<*<=====r c d}t }t ||d d t ||dd g d d S )
zTest validate nb namesr_ nbfields)rD zg%i)rj
defaultfmt)rD rE rF g0g1Nrc rd s r test_validate_nb_namesz(TestNameValidator.test_validate_nb_namesw sg "!OO YYx!444f===YYx!FFF000 2 2 2 2 2r c d}t }t || du t ||d g d dS )zTest validate no namesNr1 ri )r` ra rb )r r r rd s r test_validate_wo_namesz(TestNameValidator.test_validate_wo_names sY !OO (##t+,,,YYx!4446H6H6HIIIIIr N)r; r< r= rR r] rf ro rq r? r r rA rA S sj
J
J
J G G G
>
>
>2 2 2J J J J Jr rA c J t t j | d d d S )Nz%Y-%m-%dr1 )r timestrptime)ss r _bytes_to_daterv s" q*--bqb1 22r c N e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d S )
TestStringConverterzTest StringConverterc t t d }t |j d t |j d dS )z"Test creation of a StringConverteriayrh N)r
intr _statusdefaultr converters r
test_creationz!TestStringConverter.test_creation s= #C00 Y&***Y&/////r c t }t |j d t | d d t |j d ddlmc m} t | |j j
| |j j
k }t | d d t |j d|z t | d d t |j d |z t | d
t d
t |j d |z d D ]b}| | }t t | t u t |d
t |j d|z cdS )zTests the upgrade method.r 0rh N 17179869184l z0.g 0jr1 )rD arD )r
r r{ upgradenumpy.core.numericcorenumericrz dtypeint_itemsizeint64r complexr typestr)r r~ nx
status_offsetru ress r test_upgradez TestStringConverter.test_upgrade s $%% Y&*** Y&&s++Q///Y&*** (''''''''BHHRW - - 6"(9K9K9T TUU
Y&&}55{CCCY&M(9::: ))$//555Y&M(9::: Y&&t,,gdmm<<<Y&M(9::: ? ?A##A&&C DII$ % % % c " " " *A
,= > > > > ? ?r c p t d }| d t |d d t |d |j t |d |j t |d |j |d d S # t $ r Y d S w xY w)
z Tests the use of missing values.)missingmissed)missing_valuesr r r r r missN)r
r r r| rP r} s r test_missingz TestStringConverter.test_missing s # 4> ? ? ? # YYs^^Q'''YYr]]I$5666YYy))9+<===YYx(()*;<<< If DD s B' '
B54B5c t }t j dd } t j |t ddd t |t ddd } |d }t |t ddd |d }t |t ddd |d }t |t ddd |t _ dS # |t _ w xY w) zTests updatemapperNi rh z
2001-01-01i z
2009-01-01i r )rv r
_mapperupgrade_mapperr r )r
dateparser_original_mapperconvertr s r test_upgrademapperz&TestStringConverter.test_upgrademapper s #
*21115
7 *:tD!Q7G7G H H H%j$tQ2B2BCCG7<((D tD!Q / / 0 0 07<((D tD!Q / / 0 0 072;;D tD!Q / / 0 0 0&6O # # #&6O # 6 6 6 6s CC, ,C:c t j dd }t t }t |j | t t |d dS )zAMake sure that string-to-object functions are properly recognizedNr| )r
r rv r r hasattr)r
old_mapperconvs r test_string_to_objectz)TestStringConverter.test_string_to_object sP $,QQQ/
~..T\:...i(()))))r c t ddd }| d t |j d t |j t j t t ddd }| d t |j d t |j t j t dS )z+Make sure we don't lose an explicit defaultNr i)r r| z
3.14159265r )r
r r r| r npr floatr} s r test_keep_defaultz%TestStringConverter.test_keep_default s #D,02 2 2 ,'''Y&---Y^RXe__555# Q0 0 0 ,'''Y&***Y^RXe__55555r c ^ t t dd }t |j d dS )z'Check that we don't lose a default of 0r N/Ar| r N)r
rz r r| r} s r test_keep_default_zeroz*TestStringConverter.test_keep_default_zero s6 #C38: : : Y&*****r c b t t dd }t |j ddh dS )z*Check that we're not losing missing valuesr r r r N)r
rz r r r} s r test_keep_missing_valuesz,TestStringConverter.test_keep_missing_values sE #C38: : : $r5k 3 3 3 3 3r c t t j d }d}t || dk d}t || dk dS )z/Check that int64 integer types can be specifiedr r| z-9223372036854775807l 9223372036854775807l N)r
r r r r r~ vals r test_int64_dtypez$TestStringConverter.test_int64_dtype s^ #BHa888 $ #"66777# #"5566666r c x t t j d }d}t || dk dS )z0Check that uint64 integer types can be specifiedr r 9223372043271415339l +&| N)r
r uint64r r s r test_uint64_dtypez%TestStringConverter.test_uint64_dtype s= #BIq999 # #"5566666r N)r; r< r= r> r r r r r r r r r r r? r r rx rx s 0 0 0 #? #? #?J 7 7 7 * * * 6 6 6 + + + 3 3 37 7 77 7 7 7 7r rx c e Zd Zd Zd Zd ZdS )TestMiscFunctionsc p t j t }t t | d t j ddt fg }t t | d t j dt
fddt fdgfg }t t | d dS ) zTest has_nested_dtypeF)rC z|S3rI rC BA)BB|S1TN)r r r r r rz r ndtypes r test_has_nested_dtypez'TestMiscFunctions.test_has_nested_dtype s % &v..666<#u 677&v..666C:tUm]-K'LMNN&v..55555r c t }t t | t j t d}t t | t j ddg t t |d t j ddg d}t t |d t j d
d g d}t t |d t j d
d g d}t t |d
t j dd g t t |dd t j d
dg dt
fdt fg}t t | t j dt
fdt fg t t |d t j dt
fdt fg t t |d t j dt
fdt fg t t |d t j dt
fdt fg t
t t f}t t | t j dt
fdt fdt fg t
t t f}t t |d t j dt
fdt fdt fg t j t }t t |d t j d d D t j t }t t |g d d t j d d D d S )!zTest ndtype on dtypeszi4, f8)r` i4)ra f8z
field_%03i)rl ) field_000r ) field_001r za, b)rQ )rD r rE r za, b, cz, brD zf%02i)rQ rl )f00r rC rI za,brE r` za,b,cra rb rF c g | ] }|t f S r? r .0_s r
z5TestMiscFunctions.test_easy_dtype..G s C C Ca5z C C Cr r_ )r r r c g | ] }|t f S r? r r s r r z5TestMiscFunctions.test_easy_dtype..L s @@@Qq%j@@@r )r f01f02N)r r r r r rz r s r test_easy_dtypez!TestMiscFunctions.test_easy_dtype s Z''% 999Z''X|\ :;; = = = Z<@@@X 24G HII K K K Zf555X{K 899 ; ; ; Zi888X{K 899 ; ; ; Ze444X|[ 9:: < < < ZcgFFFX{M :;; = = = *sEl+Z''C:U|2L)M)MNNNZe444XSzC< 899 ; ; ; Zc222XSzD%= 9:: < < < Zg666XSzC< 899 ; ; ; ue$Z''Xc{T5MD%= IJJ L L L ue$Zi888XSzC<#u FGG I I I % Zi888X C C? C C CDD F F F % v\\\g F F F H@@*?@@@ A A C C C C Cr c t j ddg }t | }t |t t g t j dddgfdt
fg }t | }t |t j d t j d t
g t j dt d ffdt
d
ffg }t | }t |t t
g t |d }t |t gd z t
gd
z z t j d d
g }t | }t |t t g dS )zTesting flatten_dtype)rD r r rD )aar )ab|S2rE r r r r1 T))rD rC r ))rE rI r N)r r r
r r rz )r dtdt_flats r test_flatten_dtypez$TestMiscFunctions.test_flatten_dtypeN sB X{K0
1
1 ##Wuen---
Xm] ;
?
? ##Wuen-----r N)r; r< r= r r r r? r r r r
sE 6 6 67C 7C 7Cr. . . . .r r )rs datetimer numpyr
numpy.testingr r r r numpy.lib._iotoolsr r r
r r r
r rA rv rx r r? r r r s @2 @2 @2 @2 @2 @2 @2 @2J1J 1J 1J 1J 1J 1J 1J 1Jl3 3 3z7 z7 z7 z7 z7 z7 z7 z7zW. W. W. W. W. W. W. W. W. W.r