o թZh"@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@sdeZdZUdZded<dZgZded<ddZd d Zddd dddZ ddZ dddddZ dS) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescCs |}||j|jd|S)N)r low_memory)copyupdaterr )selfkwargsrV/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/io/parser/conftest.py update_kwargsszBaseParser.update_kwargscO||}t|i|SN)rrr argsr rrrr zBaseParser.read_csv)raise_on_extra_warningscheck_stacklevel warn_type type[Warning]warn_msgstrrboolcOsN||}tj||||dt|i|WdS1s wYdS)N)matchrr)rtmassert_produces_warningr)r rrrrrr rrrread_csv_check_warningss  $z"BaseParser.read_csv_check_warningscOrr)rrrrrrr4rzBaseParser.read_table)rcOsL||}tj|||dt|i|WdS1swYdS)N)rr)rrrr)r rrrrr rrrread_table_check_warnings8s  $z$BaseParser.read_table_check_warnings)rrrrrr)rrrr) __name__ __module__ __qualname__r__annotations__r rrrr rr!rrrrrs    rc@seZdZdZgdZdS)CParserc)NhighZ round_tripNr"r#r$rrrrrrr&Js r&c@eZdZdZdS)CParserHighMemoryFNr"r#r$r rrrrr+Or+c@r*)CParserLowMemoryTNr,rrrrr.Sr-r.c@eZdZdZdgZdS) PythonParserpythonNr)rrrrr0W r0c@r/) PyArrowParserpyarrowNr)rrrrr3\r2r3cCs |dddS)zG The directory path to the data files needed for parser tests. ioparserdatardatapathrrr csv_dir_pathas r:cCstj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r5r7csvz test1.csv)ospathjoinr8rrrcsv1isr?Zmarksr1Zc_highZc_lowr4paramsidscCs8|}|jdkrtdtdddl}|d|S)z) Fixture all of the CSV parsers. r4rN)paramrpytestZ importorskiprr4Z set_cpu_count)requestr6r4rrr all_parserss   rHcC|S)z< Fixture all of the CSV parsers using the C engine. rErGrrr c_parser_onlyrLcCrI)zA Fixture all of the CSV parsers using the Python engine. rJrKrrrpython_parser_onlyrMrNcCrI)zB Fixture all of the CSV parsers using the Pyarrow engine. rJrKrrrpyarrow_parser_onlyrMrOcCsg}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. valuesrr4rr@-rA) zip _all_parsers_all_parser_idshasattrrPrrrFmark single_cpurEappend)rBrCr6Z parser_id precisionrVrErrr,_get_all_parser_float_precision_combinationss     rZrBrCcC|jS)zR Fixture for all allowable combinations of parser and float precision rJrKrrrall_parsers_all_precisionss r\) )rQ_utfZUTFcCs$g|]}tD] }||dqqS)z{0})_encoding_prefixes).0sepprefixrrr s rg)rBcCr[)zE Fixture for all possible integer values for a UTF encoding. rJrKrrr utf_valuerhcCr[)zD Fixture for all possible string formats of a UTF encoding. rJrKrrr encoding_fmtrirj)*)z-1,0)z-1,2e0g333333)z-1e0rk)z+1e0?)z+1e+0rl)z+1e-1皙?)z+,1e1rl)z+1,e0rl)z-,1e1rk)z-1,e0rk)z0,1rm)z1,rl)z,1rm)z-,1g)z1_,rl)z1_234,56 ףp=J@)z 1_234,56e0rn)rara)-_ro)-_1rp)-_1e0rq)_1rr)_1,rs)_1,_rt)_1e0ru)1,2e_1rv)1,2e1_0rw)1,_2rx),1__2ry),1erz)-,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. rJrKrrrnumeric_decimals4rcCs^d|jvr |d}nd|jvr|dd}ndS|jdkr-tjjdd}||dSdS)z Fixture that xfails a test if the engine is pyarrow. Use if failure is do to unsupported keywords or inconsistent results. rHr\rNr4zpyarrow doesn't support this.reason) fixturenamesgetfixturevaluerrFrVZxfailZ applymarker)rGr6rVrrr pyarrow_xfails    rcCsRd|jvr |d}nd|jvr|dd}ndS|jdkr'tjdddSdS)z Fixture that skips a test if the engine is pyarrow. Use if failure is do a parsing failure from pyarrow.csv.read_csv rHr\rNr4z,https://github.com/apache/arrow/issues/38676r)rrrrFskip)rGr6rrr pyarrow_skip0s    r)3 __future__rr<rFZpandas.compat._optionalrZpandasrrZpandas._testingZ_testingrrr&r+r.r0r3Zfixturer:r?Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyrErVrWZ_pyarrow_parsers_onlyrSZ_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrTrHrLrNrOrZr\Z _utf_valuesZ_encoding_sepsrcZ_encoding_fmtsrhrjrrrrrrrsx   :                /