gL i3)ZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZmZddlmZddlmZej.j1dZej4dZej4dZej4dZd Zd Zd Z d Z!ej.jEd ddgdZ#dZ$dZ%dZ&dZ'dZ(ejRdZ*ej.jVdZ,ej.jVej.jEdgddZ-ej.j]edej.jVejRdZ/ej`ddZ1d Z2d!Z3d"Z4d#Z5d$Z6d%Z7y)&N)using_string_dtype) DataFrame date_rangeread_csv read_excel read_feather read_json read_parquet read_pickle read_stata read_table)_test_decoratorsz=ignore:Passing a BlockManager to DataFrame:DeprecationWarningc#Ktjdddlm}ddlm}ddlm}Gdd|}|d|d ||jddd|jd<|jjyw) Nfsspecr)register_implementation)MemoryFileSystem) _registryc*eZdZdZdgZdfd ZxZS) fsspectest..TestMemoryFStestmemNcb|jdd|jd<t| di|y)Ntestr)poprsuper__init__)selfkwargs __class__s a/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/io/test_fsspec.pyrz)fsspectest..TestMemoryFS.__init__'s+!::fd3DIIaL G  &v &)returnN)__name__ __module__ __qualname__protocolrr __classcell__)rs@r TestMemoryFSr#sv ' 'r!r(rT)clobber) pytest importorskiprrfsspec.implementations.memoryrfsspec.registryrrrstoreclear)rrregistryr(s r fsspectestr1sp !.>5'''I|TB . LLD!LasBBc `tddgdtjgddgtddd S) Ng@tsz 2018-06-18)periods)intfloatstrdt)rnpnanrrr!r df1r?2s7 q6266]:\15   r!c#Ktjd}|jd}||jj yw)Nrmemory)r*r+ filesystemr.r/)rmemfss r cleared_fsrD>s;   *F   h 'E K KKsAAct|jdj}|jdd5}|j |dddt ddg}t j||y#1swY.xYw)NFindexz test/test.csvwbmemory://test/test.csvr<) parse_dates)r;to_csvencodeopenwritertmassert_frame_equal)rDr?textwdf2s r test_read_csvrTGso szzz& ' . . 0D $ /1   +$ @C#s# s A;;Bc,ddlm}tjtd5t ddddd}|j |dd|d tjt|5t d dddy#1swYQxYw#1swYyxYw) Nr)known_implementationsnosuchprotocolmatchznosuchprotocol://test/test.csvztest error message couldexistzunimportable.CouldExist)classerrzcouldexist://test/test.csv)r-rVr*raises ValueErrorrsetitem ImportError) monkeypatchrDrVerr_msgs r test_reasonable_errorrcPs5 z)9 :3123"G+G< {' 2/-.//33//s A>) B >B Bct|jddtddgd}tj||y)NrITrFr<rrJ index_col)rKrrOrP)rDr?rSs r test_to_csvrg_s4JJ'tJ4 +$1 MC#s#r!ctjdd}d|}|j|dt|dgd}t j ||y) Nopenpyxlxlsxzmemory://test/test.TrFr<rre)r*r+to_excelrrOrP)rDr?extpathrSs r test_to_excelrngsQ  # C  &DLLTL" Tv ;C#s#r! binary_modeFTctjd}d}|rdnd}|j||j5}|j|d|jrJ ddd|j dd}|j||5}t |d gd }|jrJ dddtj|y#1swYlxYw#1swY,xYw) NrrIrHrR)modeTrFrr<rre) r*r+rMrKclosedreplacerrOrP)rDror?rrmrq fsspec_objectrSs r test_to_csv_fsspec_objectrvrs   *F #D4CD T % * * ,(  = - '''''( <<S !D T %(   !''''' (#s#(( ((s"C C CC"ctddgi}|jdddid|jddk(sJtdddi |jddk(sJy Nartestmem://test/test.csvr csv_writeFstorage_optionsrGcsv_readr})rrKrrr1dfs r test_csv_optionsrsp C!: BII!FK3HPU ??1  ,, , &8LM ??1  ++ +r!ctddgi}|jdddid|jddk(sJtdddi |jddk(sJyrx)rrKrr rs r test_read_table_optionsrsp C!: BII!FK3HPU ??1  ,, ,(6::NO ??1  ++ +r!ctjdd}tddgi}d|}|j|ddid |jddk(sJt |dd i |jdd k(sJy) Nrirjryrztestmem://test/test.rrNFr|readr)r*r+rrkrr)r1 extensionrrms r test_excel_optionsrs  #I C!: B !) -DKKvw&7uKE ??1  (( (tff%56 ??1  '' 'r!cXtjd|jddddy)CRegression test for writing to a not-yet-existent GCS Parquet file. fastparquetrITN)rGengine compression)r*r+ to_parquet)rDr?s r test_to_parquet_new_filers+  &NN ]PTr!ctjdtddgi}|jdddddi|jddk(sJt dddd i |jdd k(sJy) rpyarrowryrrzNr parquet_writerrr} parquet_readrr}r*r+rrrr rs r test_arrowparquet_optionsrs  " C!: BMM!1  ??1  00 0!0 ??1  // /r!ctjdtddgi}|jdddddi|jddk(sJt dddd i |jdd k(sJy) rrryrrzNrrrrrrrs r test_fastparquet_optionsrs  & C!: BMM!1  ??1  00 0!0 ??1  // /r!c~tjdtjt d|j d|t |tjt d|j d|t |tjt d|j d|t |y)Ns3fss3:// /tips.csvrz /tips.csv.gzz /tips.csv.bz2r*r+rO assert_equalrname)s3_public_bucket_with_data tips_files3sos r test_from_s3_csvrs OO.334I >PT   OO.334L ASW    OO.334M BTX   r!r&)s3s3as3nctjdtjt |d|j d|t |y)Nrz://rrr)rrr&rs r test_s3_protocolsrsK OOj6;;>. // /r!cvtjdtjtd5t dddidddtjtd5t dddidddt j}tjtd5t |ddidddtddgi}tjtd5|jd ddidddy#1swYxYw#1swYxYw#1swYixYw#1swYyxYw) Nrr}rX localfileryTrr nonfsspecpath) r*r+r]r^rr ioBytesIOrr)byrs r test_non_fsspec_optionsrNs  " z): ;;sDk:; z): ;?[3+>? B z): ;2c4[12 C!: B z): ;D oT{ CDD;;?? 22DDs/D $D+D#,D/ DD #D,/D8)8rnumpyr=r*pandas._configrpandasrrrrrr r r r r pandas._testing_testingrO pandas.utilrtdmarkfilterwarnings pytestmarkfixturer1r?rDrTrcrgrn parametrizervrrrrr&skip_array_manager_not_yet_implementedr single_cpurrxfailrskip_if_installedrrrrrrrrr!r rs  -   . [[ ' 'C *$ /$$ 6$7$*,, (0&**0+0(.%9:;%'0PQ** +R h+ + ###"30Dr!