U
|
ý°d! ã + @ st d dl mZ d dlZd dlZd dlmZ d dlmZmZ d dl m
|
Z G dd dZ G dd de Z G d d
|
d
|
e ZG d d d e ZG d d de ZG dd de Zejdd Zejdd ZeZeZeZeZegZeegZejeej j dgZ eee Z dgZ ddgZ!dgZ"e!e e"Z#eje e#ddd Z$ejee!dd d Z%ejee dd d Z&eje e"dd!d" Z'd#d$ Z(eje( d% e( d& dd'd( Z)d)d*d+gZ*d,d-d.gZ+d/d0gZ,d1d2 e+D Z-eje*d3d4d5 Z.eje-d3d6d7 Z/ejd8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dag*d3dbdc Z0ejddde Z1ejdfdg Z2dS )hé )Ú annotationsN)ÚVERSIONS)Úread_csvÚ
|
read_tablec @ sb e Zd ZU dZded< dZg Zded< dd Zd d
|
Zd d d ddZ dd Z
|
d d d ddZ dS )Ú
|
BaseParserNz
|
str | NoneÚengineTzlist[str | None]Úfloat_precision_choicesc C s | ¡ }| | j| jd¡ |S )N)r Ú
|
low_memory)ÚcopyÚupdater r )ÚselfÚkwargs© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/conftest.pyÚ update_kwargs s zBaseParser.update_kwargsc O s | |¡}t||S ©N)r r ©r Úargsr r r r r s
|
zBaseParser.read_csvz type[Warning]Ústr)Ú warn_typeÚwarn_msgc
|
O s: | |¡}tj||d t||W 5 Q R £ S Q R X d S ©N)Úmatch)r ÚtmÚassert_produces_warningr ©r r r r r r r r Úread_csv_check_warnings s
|
z"BaseParser.read_csv_check_warningsc O s | |¡}t||S r )r r r r r r r ) s
|
zBaseParser.read_tablec
|
O s: | |¡}tj||d t||W 5 Q R £ S Q R X d S r )r r r r r r r r Úread_table_check_warnings- s
|
z$BaseParser.read_table_check_warnings) Ú__name__Ú
|
__module__Ú __qualname__r Ú__annotations__r r r r r r r r r r r r s
|
|
r c @ s e Zd ZdZdddgZdS )ÚCParserÚcNÚhighZ
|
round_trip©r r r r r r r r r r" 8 s r" c @ s e Zd ZdZdS )ÚCParserHighMemoryFN©r r r r r r r r r& = s r& c @ s e Zd ZdZdS )ÚCParserLowMemoryTNr' r r r r r( A s r( c @ s e Zd ZdZdgZdS )Ú PythonParserÚpythonNr% r r r r r) E s r) c @ s e Zd ZdZdgZdS )Ú PyArrowParserÚpyarrowNr% r r r r r+ J s r+ c C s | dddS )zG
|
The directory path to the data files needed for parser tests.
|
ÚioÚparserÚdatar ©Údatapathr r r Ú csv_dir_pathO s r2 c C s t j | dddd¡S )zH
|
The path to the data file "test1.csv" needed for parser tests.
|
r- r/ Úcsvz test1.csv)ÚosÚpathÚjoinr0 r r r Úcsv1W s r7 ©Zmarksr* Zc_highZc_lowr, ©ÚparamsÚidsc C s8 | ¡ }|jdkr4t dtd ¡ ddl}| d¡ |S )z)
|
Fixture all of the CSV parsers.
|
r, r Né )Úparamr ÚpytestZ importorskipr r, Z set_cpu_count)Úrequestr. r, r r r Ú all_parsersq s
|
|
r@ c C s | ¡ S )z<
|
Fixture all of the CSV parsers using the C engine.
|
©r= ©r? r r r Ú c_parser_only s rC c C s | ¡ S )zA
|
Fixture all of the CSV parsers using the Python engine.
|
rA rB r r r Úpython_parser_only s rD c C s | ¡ S )zB
|
Fixture all of the CSV parsers using the Pyarrow engine.
|
rA rB r r r Úpyarrow_parser_only s rE c C s g } g }t ttD ]p\}}t|dr.|jd }|jD ]L}|jdkrJtjj nd}tj
|
| |f|d}| |¡ | | d| ¡ q4q| |dS )za
|
Return all allowable parser and float precision
|
combinations and corresponding ids.
|
Úvaluesr r, r r8 ú-r9 ) ÚzipÚ _all_parsersÚ_all_parser_idsÚhasattrrF r r r> ÚmarkÚ
|
single_cpur= Úappend)r: r; r. Z parser_idZ precisionrL r= r r r Ú,_get_all_parser_float_precision_combinations s
|
|
|
|
rO r: r; c C s | j S )zR
|
Fixture for all allowable combinations of parser
|
and float precision
|
rA rB r r r Úall_parsers_all_precisions® s rP é é é Ú rG Ú_ÚutfZUTFc C s& g | ] }t D ]}| | d q qS )z{0})Ú_encoding_prefixes)Ú.0ÚsepÚprefixr r r Ú
|
<listcomp>¿ s r[ )r: c C s | j S )zE
|
Fixture for all possible integer values for a UTF encoding.
|
rA rB r r r Ú utf_valueÄ s r\ c C s | j S )zD
|
Fixture for all possible string formats of a UTF encoding.
|
rA rB r r r Ú encoding_fmtÌ s r] )z-1,0ç ð¿)z-1,2e0g333333ó¿)z-1e0r^ )z+1e0ç ð?)z+1e+0r_ )z+1e-1ç¹?)z+,1e1r_ )z+1,e0r_ )z-,1e1r^ )z-1,e0r^ )z0,1r` )z1,r_ )z,1r` )z-,1g¹¿)z1_,r_ )z1_234,56ç
|
×£p=J@)z
|
1_234,56e0ra )rU rU )ú-_rb )ú-_1rc )ú-_1e0rd )Ú_1re )ú_1,rf )ú_1,_rg )Ú_1e0rh )ú1,2e_1ri )ú1,2e1_0rj )ú1,_2rk )ú,1__2rl )ú,1erm )ú-,1ern )ú 1_000,000_000ro )ú1,e1_2rp )úe11,2rq )ú1e11,2rr )ú1,2,2rs )ú1,2_1rt )ú 1,2e-10e1ru )ú--1,2rv )ú1a_2,1rw )z1,2E-1g¸
ëQ¸¾?)z1,2E1g (@c C s | j S )z¨
|
Fixture for all numeric formats which should get recognized. The first entry
|
represents the value to read while the second represents the expected result.
|
rA rB r r r Únumeric_decimalÔ s 4rx c C s\ d| j kr| d¡}n d| j kr0| d¡d }ndS |jdkrXtjjdd}| j |¡ dS )z>
|
Fixture that xfails a test if the engine is pyarrow.
|
r@ rP r Nr, ú pyarrow doesn't support this.)Úreason)Ú fixturenamesÚgetfixturevaluer r> rL ZxfailÚnodeZ
|
add_marker)r? r. rL r r r Ú pyarrow_xfail s
|
|
|
r~ c C sL d| j kr| d¡}n d| j kr0| d¡d }ndS |jdkrHt d¡ dS )z=
|
Fixture that skips a test if the engine is pyarrow.
|
r@ rP r Nr, ry )r{ r| r r> Úskip)r? r. r r r Ú pyarrow_skip s
|
|
|
r )3Ú
|
__future__r r4 r> Zpandas.compat._optionalr Zpandasr r Zpandas._testingZ_testingr r r" r& r( r) r+ Zfixturer2 r7 Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyr= rL rM Z_pyarrow_parsers_onlyrI Z_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrJ r@ rC rD rE rO rP Z _utf_valuesZ_encoding_sepsrW Z_encoding_fmtsr\ r] rx r~ r r r r r Ú<module> sÊ (
|
|
|
|
|
|
|
|
þ
|
|
|
ÿ
|
|
|
|
Õÿ/
|
|
|
