a Gb.@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@sReZdZUdZded<dZgZded<ddZd d Zd d d ddZ ddZ dS) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescCs |}||j|jd|S)N)r low_memory)copyupdaterr )selfkwargsr>lib/python3.9/site-packages/pandas/tests/io/parser/conftest.py update_kwargsszBaseParser.update_kwargscOs||}t|i|SN)rrr argsr rrrrs zBaseParser.read_csvz type[Warning]str) warn_typewarn_msgcOsJ||}tj||dt|i|WdS1s<0YdS)N)match)rtmZassert_produces_warningr)r rrrr rrrread_csv_check_warningss z"BaseParser.read_csv_check_warningscOs||}t|i|Sr)rrrrrrr)s zBaseParser.read_table) __name__ __module__ __qualname__r__annotations__r rrrrrrrrrrs    rc@seZdZdZgdZdS)CParserc)NZhighZ round_tripNrrrrrrrrrr.src@seZdZdZdS)CParserHighMemoryFNrrrr rrrrr!3sr!c@seZdZdZdS)CParserLowMemoryTNr"rrrrr#7sr#c@seZdZdZdgZdS) PythonParserpythonNr rrrrr$;sr$c@seZdZdZdgZdS) PyArrowParserpyarrowNr rrrrr&@sr&cCs |dddS)zG The directory path to the data files needed for parser tests. ioparserdatardatapathrrr csv_dir_pathEsr-cCstj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r(r*Zcsvz test1.csv)ospathjoinr+rrrcsv1Msr1Zmarksr%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_parsersgs   r:cCs|S)z< Fixture all of the CSV parsers using the C engine. r7r9rrr c_parser_onlywsr=cCs|S)zA Fixture all of the CSV parsers using the Python engine. r;r<rrrpython_parser_onlysr>cCs|S)zB Fixture all of the CSV parsers using the Pyarrow engine. r;r<rrrpyarrow_parser_onlysr?cCsg}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'rr2-r3) zip _all_parsers_all_parser_idshasattrr@rrr8mark single_cpur7append)r4r5r)Z parser_idZ precisionrFr7rrr,_get_all_parser_float_precision_combinationss    rIr4r5cCs|jS)zR Fixture for all allowable combinations of parser and float precision r;r<rrrall_parsers_all_precisionss rJ) )rA_utfZUTFcCs&g|]}tD]}||dq qS)z{0})_encoding_prefixes).0sepprefixrrr srU)r4cCs|jS)zE Fixture for all possible integer values for a UTF encoding. r;r<rrr utf_valuesrVcCs|jS)zD Fixture for all possible string formats of a UTF encoding. r;r<rrr encoding_fmtsrW)*)z-1,0)z-1,2e0g333333)z-1e0rX)z+1e0?)z+1e+0rY)z+1e-1皙?)z+,1e1rY)z+1,e0rY)z-,1e1rX)z-1,e0rX)z0,1rZ)z1,rY)z,1rZ)z-,1g)z1_,rY)z1_234,56 ףp=J@)z 1_234,56e0r[)rOrO)-_r\)-_1r])-_1e0r^)_1r_)_1,r`)_1,_ra)_1e0rb)1,2e_1rc)1,2e1_0rd)1,_2re),1__2rf),1erg)-,1erh) 1_000,000_000ri)1,e1_2rj)e11,2rk)1e11,2rl)1,2,2rm)1,2_1rn) 1,2e-10e1ro)--1,2rp)1a_2,1rq)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. r;r<rrrnumeric_decimals4rrcCs\d|jvr|d}nd|jvr0|dd}ndS|jdkrXtjjdd}|j|dS)z> Fixture that xfails a test if the engine is pyarrow. r:rJrNr'pyarrow doesn't support this.)reason) fixturenamesgetfixturevaluerr8rFZxfailZnodeZ add_marker)r9r)rFrrr pyarrow_xfails    rwcCsLd|jvr|d}nd|jvr0|dd}ndS|jdkrHtddS)z= Fixture that skips a test if the engine is pyarrow. r:rJrNr'rs)rurvrr8skip)r9r)rrr pyarrow_skips    ry)3Z __future__rr.r8Zpandas.compat._optionalrZpandasrrZpandas._testingZ_testingrrrr!r#r$r&Zfixturer-r1Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyr7rFrGZ_pyarrow_parsers_onlyrCZ_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrDr:r=r>r?rIrJZ _utf_valuesZ_encoding_sepsrQZ_encoding_fmtsrVrWrrrwryrrrrsv                  /