U sVc+@stddlmZddlZddlZddlmZddlmZmZddl m Z GdddZ Gddde Z Gd d d e ZGd d d e ZGd dde ZGddde ZejddZejddZeZeZeZeZegZeegZejeejjdgZeeeZdgZ ddgZ!dgZ"e!e e"Z#ejee#dddZ$ejee!dddZ%ejee ddd Z&ejee"dd!d"Z'd#d$Z(eje(d%e(d&dd'd(Z)d)d*d+gZ*d,d-d.gZ+d/d0gZ,d1d2e+DZ-eje*d3d4d5Z.eje-d3d6d7Z/ejd8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dag*d3dbdcZ0ejdddeZ1ejdfdgZ2dS)h) annotationsN)VERSIONS)read_csv read_tablec@sbeZdZUdZded<dZgZded<ddZd d Zd d d ddZ ddZ d d d ddZ dS) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescCs |}||j|jd|S)N)r low_memory)copyupdaterr )selfkwargsrC/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/io/parser/conftest.py update_kwargsszBaseParser.update_kwargscOs||}t||SN)rrr argsr rrrrs zBaseParser.read_csvz type[Warning]str) warn_typewarn_msgc Os:||}tj||dt||W5QRSQRXdSN)match)rtmassert_produces_warningrr rrrr rrrread_csv_check_warningss z"BaseParser.read_csv_check_warningscOs||}t||Sr)rrrrrrr)s zBaseParser.read_tablec Os:||}tj||dt||W5QRSQRXdSr)rrrrrrrrread_table_check_warnings-s z$BaseParser.read_table_check_warnings) __name__ __module__ __qualname__r__annotations__r rrrrrrrrrrrs    rc@seZdZdZdddgZdS)CParsercNhighZ round_triprrr rrrrrrr"8sr"c@seZdZdZdS)CParserHighMemoryFNrrr r rrrrr&=sr&c@seZdZdZdS)CParserLowMemoryTNr'rrrrr(Asr(c@seZdZdZdgZdS) PythonParserpythonNr%rrrrr)Esr)c@seZdZdZdgZdS) PyArrowParserpyarrowNr%rrrrr+Jsr+cCs |dddS)zG The directory path to the data files needed for parser tests. ioparserdatardatapathrrr csv_dir_pathOsr2cCstj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r-r/csvz test1.csv)ospathjoinr0rrrcsv1Wsr7Zmarksr*Zc_highZc_lowr,paramsidscCs8|}|jdkr4tdtdddl}|d|S)z) Fixture all of the CSV parsers. r,rN)paramrpytestZ importorskiprr,Z set_cpu_count)requestr.r,rrr all_parsersqs   r@cCs|S)z< Fixture all of the CSV parsers using the C engine. r=r?rrr c_parser_onlysrCcCs|S)zA Fixture all of the CSV parsers using the Python engine. rArBrrrpython_parser_onlysrDcCs|S)zB Fixture all of the CSV parsers using the Pyarrow engine. rArBrrrpyarrow_parser_onlysrEcCsg}g}tttD]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. valuesrr,rr8-r9) zip _all_parsers_all_parser_idshasattrrFrrr>mark single_cpur=append)r:r;r.Z parser_idZ precisionrLr=rrr,_get_all_parser_float_precision_combinationss    rOr:r;cCs|jS)zR Fixture for all allowable combinations of parser and float precision rArBrrrall_parsers_all_precisionss rP rG_utfZUTFcCs&g|]}tD]}||dq qS)z{0})_encoding_prefixes).0sepprefixrrr sr[)r:cCs|jS)zE Fixture for all possible integer values for a UTF encoding. rArBrrr utf_valuesr\cCs|jS)zD Fixture for all possible string formats of a UTF encoding. rArBrrr encoding_fmtsr])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)rUrU)-_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-1gQ?)z1,2E1g(@cCs|jS)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. rArBrrrnumeric_decimals4rxcCs\d|jkr|d}nd|jkr0|dd}ndS|jdkrXtjjdd}|j|dS)z> Fixture that xfails a test if the engine is pyarrow. r@rPrNr,pyarrow doesn't support this.)reason) fixturenamesgetfixturevaluerr>rLZxfailnodeZ add_marker)r?r.rLrrr pyarrow_xfail s    r~cCsLd|jkr|d}nd|jkr0|dd}ndS|jdkrHtddS)z= Fixture that skips a test if the engine is pyarrow. r@rPrNr,ry)r{r|rr>skip)r?r.rrr pyarrow_skips    r)3 __future__rr4r>Zpandas.compat._optionalrZpandasrrZpandas._testingZ_testingrrr"r&r(r)r+Zfixturer2r7Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyr=rLrMZ_pyarrow_parsers_onlyrIZ_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrJr@rCrDrErOrPZ _utf_valuesZ_encoding_sepsrWZ_encoding_fmtsr\r]rxr~rrrrrs   (                   /