o Ti@s,ddlmZddlZddlZddlmZddlmZmZddl m Z GdddZ Gddde Z Gd d d e ZGd d d e ZGd dde ZGddde ZejddZejddZeZeZeZeZegZeegZejeejjdgZgeeeZdgZ ddgZ!dgZ"ge!e e"Z#ejee#dddZ$ejee!dddZ%ejee ddd Z&ejee"dd!d"Z'd#d$Z(eje(d%e(d&dd'd(Z)gd)Z*gd*Z+d+d,gZ,d-d.e+DZ-eje*d/d0d1Z.eje-d/d2d3Z/ejgd4d/d5d6Z0ejd7d8Z1ejd9d:Z2dS);) annotationsN)VERSIONS)read_csv read_tablec@sVeZdZUdZded<dZgZded<ddZd d ZdddZ ddZ dddZ dS) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescCs |}||j|jd|S)N)r low_memory)copyupdaterr )selfkwargsr_/var/www/html/evchargy.com/venv/lib/python3.10/site-packages/pandas/tests/io/parser/conftest.py update_kwargsszBaseParser.update_kwargscO||}t|i|SN)rrr argsr rrrr zBaseParser.read_csv warn_type type[Warning]warn_msgstrcOJ||}tj||dt|i|WdS1swYdSN)match)rtmassert_produces_warningrr rrrr rrrread_csv_check_warnings  $z"BaseParser.read_csv_check_warningscOrr)rrrrrrr)rzBaseParser.read_tablecOrr)rrrrrrrrread_table_check_warnings-r!z$BaseParser.read_table_check_warnings)rrrr) __name__ __module__ __qualname__r__annotations__r rrrr rr"rrrrrs    rc@seZdZdZgdZdS)CParserc)NhighZ round_tripNr#r$r%rrrrrrr'8s r'c@eZdZdZdS)CParserHighMemoryFNr#r$r%r rrrrr,=r,c@r+)CParserLowMemoryTNr-rrrrr/Ar.r/c@eZdZdZdgZdS) PythonParserpythonNr*rrrrr1E r1c@r0) PyArrowParserpyarrowNr*rrrrr4Jr3r4cCs |dddS)zG The directory path to the data files needed for parser tests. ioparserdatardatapathrrr csv_dir_pathOs r;cCstj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r6r8csvz test1.csv)ospathjoinr9rrrcsv1Wsr@Zmarksr2Zc_highZc_lowr5paramsidscCs8|}|jdkrtdtdddl}|d|S)z) Fixture all of the CSV parsers. r5rN)paramrpytestZ importorskiprr5Z set_cpu_count)requestr7r5rrr all_parsersqs   rIcC|S)z< Fixture all of the CSV parsers using the C engine. rFrHrrr c_parser_onlyrMcCrJ)zA Fixture all of the CSV parsers using the Python engine. rKrLrrrpython_parser_onlyrNrOcCrJ)zB Fixture all of the CSV parsers using the Pyarrow engine. rKrLrrrpyarrow_parser_onlyrNrPcCsg}g}tttD]8\}}t|dr|jd}|jD]&}|jdkr%tjj nd}tj ||f|d}| || |d|qq ||dS)za Return all allowable parser and float precision combinations and corresponding ids. valuesrr5rrA-rB) zip _all_parsers_all_parser_idshasattrrQrrrGmark single_cpurFappend)rCrDr7Z parser_id precisionrWrFrrr,_get_all_parser_float_precision_combinationss     r[rCrDcC|jS)zR Fixture for all allowable combinations of parser and float precision rKrLrrrall_parsers_all_precisionss r]) )rR_utfZUTFcCs$g|]}tD] }||dqqS)z{0})_encoding_prefixes).0sepprefixrrr s rh)rCcCr\)zE Fixture for all possible integer values for a UTF encoding. rKrLrrr utf_valuericCr\)zD Fixture for all possible string formats of a UTF encoding. rKrLrrr encoding_fmtrjrk)*)z-1,0)z-1,2e0g333333)z-1e0rl)z+1e0?)z+1e+0rm)z+1e-1皙?)z+,1e1rm)z+1,e0rm)z-,1e1rl)z-1,e0rl)z0,1rn)z1,rm)z,1rn)z-,1g)z1_,rm)z1_234,56 ףp=J@)z 1_234,56e0ro)rbrb)-_rp)-_1rq)-_1e0rr)_1rs)_1,rt)_1,_ru)_1e0rv)1,2e_1rw)1,2e1_0rx)1,_2ry),1__2rz),1er{)-,1er|) 1_000,000_000r})1,e1_2r~)e11,2r)1e11,2r)1,2,2r)1,2_1r) 1,2e-10e1r)--1,2r)1a_2,1r)z1,2E-1gQ?)z1,2E1g(@cCr\)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. rKrLrrrnumeric_decimals4rcCs`d|jvr |d}nd|jvr|dd}ndS|jdkr.tjjdd}|j|dSdS)z> Fixture that xfails a test if the engine is pyarrow. rIr]rNr5pyarrow doesn't support this.)reason) fixturenamesgetfixturevaluerrGrWZxfailnodeZ add_marker)rHr7rWrrr pyarrow_xfail s    rcCsPd|jvr |d}nd|jvr|dd}ndS|jdkr&tddSdS)z= Fixture that skips a test if the engine is pyarrow. rIr]rNr5r)rrrrGskip)rHr7rrr pyarrow_skips    r)3 __future__rr=rGZpandas.compat._optionalrZpandasrrZpandas._testingZ_testingrrr'r,r/r1r4Zfixturer;r@Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyrFrWrXZ_pyarrow_parsers_onlyrTZ_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrUrIrMrOrPr[r]Z _utf_valuesZ_encoding_sepsrdZ_encoding_fmtsrirkrrrrrrrsx   (                /