U
|
ý°d` ã @ s d Z ddlZddlmZ ddlZddlmZ ddlmZ ddl m
|
Z ej d¡Zejdd Zd d
|
Zd d Zd d Zej dddddddg¡ej ddddg¡dd Zej jdd d idfd d iefd d idfd didfd d iefd d iefgd!d"d#d$d%d&gd'd(d) ZdS )*zf
|
Tests that dialects are properly handled during parsing
|
for all of the parsers defined in parsers.py
|
é N)ÚStringIO)Ú ParserWarning)Ú DataFrameZ pyarrow_skipc C s d} ddddddd}| |fS )NZweirdFú~ú:é )Ú doublequoteÚ
|
escapecharÚ delimiterÚskipinitialspaceÚ quotecharÚquoting© )Ú dialect_nameÚdialect_kwargsr r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/test_dialect.pyÚcustom_dialect s úr c C s\ | }d}t ¡ }t j|_|jt||d}d}| t|¡}|jdddd t ||¡ d S )Nz0label1,label2,label3
|
index1,"a,c,e
|
index2,b,d,f
|
©Údialectz/label1,label2,label3
|
index1,a,c,e
|
index2,b,d,f
|
Úaz"aT)Zinplace) ÚcsvÚexcelÚ
|
QUOTE_NONEr Úread_csvr ÚreplaceÚtmÚassert_frame_equal)Ú all_parsersÚparserÚdataZdiaÚdfÚexpr r r Ú test_dialect! s r" c C s^ d}| }d}t ddgddgd}tj|dd $ |jt||d
|
}t ||¡ W 5 Q R X d S ) NZ mydialectz+fruit:vegetable
|
apple:broccoli
|
pear:tomato
|
ZappleZpearZbroccoliZtomato)ZfruitZ vegetabler )r
|
r )r r Úwith_csv_dialectr r r )r r r r r! r r r r Útest_dialect_str7 s r$ c C sJ G dd d}d}| }d}t jt|d |jt||d W 5 Q R X d S )Nc @ s e Zd ZdS )z,test_invalid_dialect.<locals>.InvalidDialectN)Ú__name__Ú
|
__module__Ú __qualname__r r r r ÚInvalidDialectG s r( za
|
1zInvalid dialect)Úmatchr )ÚpytestZraisesÚ
|
ValueErrorr r )r r( r r Úmsgr r r Útest_invalid_dialectF s r- Úargr r r r r Úvaluer ÚdefaultÚotherc C s¾ |\}}| }t dgdgd}d}d } i }
|
|d k rv|dkrH|| |
|
|< n.|dkrjddlm} | | |
|
|< n t} d |
|
|< tj|f|0 |j| d
|
t|fd|i|
|
} t | |¡ W 5 Q R X d S ) Né é ©r Úbúa:b
|
1:2r r0 r )Úparser_defaultsZblahzConflicting values for) r Z pandas.io.parsers.base_parserr7 r r r# Úread_csv_check_warningsr r ) r r r. r/ r r r Úexpectedr Ú warning_klassÚkwdsr7 Úresultr r r Ú&test_dialect_conflict_except_delimiterR s2 ýüûr= zkwargs,warning_klassÚsepú,Ú.r
|
r zsep-override-truezsep-override-falsezdelimiter-no-conflictzdelimiter-default-argzdelimiter-conflictzdelimiter-conflict2)Úidsc
|
C sj |\}}| }t dgdgd}d}tj|f|0 |j|dt|fd|i|} t | |¡ W 5 Q R X d S )Nr2 r3 r4 r6 z"Conflicting values for 'delimiter'r )r r r# r8 r r )
|
r r Úkwargsr: r r r r9 r r< r r r Ú test_dialect_conflict_delimitery s ýüûrC )Ú__doc__r Úior r* Z pandas.errorsr Zpandasr Zpandas._testingZ_testingr ÚmarkZ usefixturesZ
|
pytestmarkZfixturer r" r$ r- Z parametrizer= rC r r r r Ú<module> sH
|
þ "
|
|
|
|
|
|
ú úö
|
