i)ddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZej6j9dZej<dZej<d Z ej<d Z!d Z"d Z#d Z$dZ%ej6jMdddgdZ'dZ(dZ)dZ*ej6jWexrexre ddZ,dZ-dZ.ej6j^ej6jMdgddZ0ej6j^ej6jMdgdd Z1ej6jWedej6j^d!Z2ejfd"d#Z4d$Z5d%Z6d&Z7d'Z8d(Z9d)Z:y)*N)using_string_dtype) HAS_PYARROW)pa_version_under14p0) 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 ]/var/www/pod-logistic/pod-ai/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__)r!s@r" TestMemoryFSr&sv ' 'r#r*rT)clobber) pytest importorskiprrfsspec.implementations.memoryrfsspec.registryrrrstoreclear)rrregistryr*s r" fsspectestr3sp !.>5'''I|TB . LLD!LasBBc `tddgdtjgddgtddd S) Ng@tsz 2018-06-18)periods)intfloatstrdt)rnpnanrrr#r"df1rA5s7 q6266]:\15   r#c#Ktjd}|jd}||jj yw)Nrmemory)r,r- filesystemr0r1)rmemfss r" cleared_fsrFAs;   *F   h 'E K KKsAAc\t|jdj}|jdd5}|j |dddt ddg}|j }|djd|d<tj||y#1swYUxYw) NFindexz test/test.csvwbmemory://test/test.csvr>) parse_datesM8[us]) r=to_csvencodeopenwritercopyastypetmassert_frame_equal)rFrAtextwdf2expecteds r" test_read_csvrZJs szzz& ' . . 0D $ /1   +$ @CxxzHd^**84HTN#x( s B""B+c,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/r\r,raises ValueErrorrsetitem ImportError) monkeypatchrFr\err_msgs r"test_reasonable_errorriUs5 z)9 :3123"G+G< {' 2/-.//33//s A>) B >B Bc|jddtddgd}|j}|djd|d<t j ||y)NrKTrHr>rrL index_colrM)rNrrRrSrTrU)rFrArXrYs r" test_to_csvrmdsWJJ'tJ4 +$1 MCxxzHd^**84HTN#x(r#ctjdd}d|}|j|dt|dgd}|j }|dj d |d<t j||y) Nopenpyxlxlsxzmemory://test/test.TrHr>rrkrM)r,r-to_excelr rRrSrTrU)rFrAextpathrXrYs r" test_to_excelrtnst  # C  &DLLTL" Tv ;CxxzHd^**84HTN#x(r# binary_modeFTctjd}d}|rdnd}|j||j5}|j|d|jrJ ddd|j dd}|j||5}t |d gd }|jrJ ddd|j}|d jd |d <tj|y#1swYxYw#1swYSxYw) NrrKrJrW)modeTrHrr>rrkrM) r,r-rPrNclosedreplacerrRrSrTrU) rFrurArrsrw fsspec_objectrXrYs r"test_to_csv_fsspec_objectr|{s    *F #D4CD T % * * ,(  = - '''''( <<S !D T %(   !''''' (xxzHd^**84HTN#x((( ((s"C4D4C=D ctddgi}|jdddid|jddk(sJtdddi |jddk(sJy Nartestmem://test/test.csvr csv_writeFstorage_optionsrIcsv_readr)rrNrrr3dfs r"test_csv_optionsrsp C!: BII!FK3HPU ??1  ,, , &8LM ??1  ++ +r#ctddgi}|jdddid|jddk(sJtdddi |jddk(sJyr~)rrNrrrs 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) Nrorprrztestmem://test/test.rrQFrreadr)r,r-rrqrr )r3 extensionrrss r"test_excel_optionsrs  #I C!: B !) -DKKvw&7uKE ??1  (( (tff%56 ??1  '' 'r#zTODO(infer_string) fastparquet)reasoncXtjd|jddddy)CRegression test for writing to a not-yet-existent GCS Parquet file. fastparquetrKTN)rIengine compression)r,r- to_parquet)rFrAs r"test_to_parquet_new_filers-   &NN ]PTr#ctjdtddgi}|jdddddi|jddk(sJt dddd i |jdd k(sJy) rpyarrowrrrNr parquet_writerrr parquet_readrrr,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) rrrrrNrrrrrrrs r"test_fastparquet_optionsrs  & C!: BMM!1  ??1  00 0!0 ??1  // /r#compression_suffix)z.gzz.bz2ctjdtd|jd||}t|}t j ||y)Ns3fss3:// /tips.csvrr,r-rnamerT assert_equal)s3_bucket_public_with_datas3so tips_filer df_from_s3 df_from_locals r"test_from_s3_csvrsS  *//0 :L9MNJY'MOOJ .r#r()s3s3as3nctjdt|d|jd|}t|}t j ||y)Nrz://rrr)rrrr(rrs r"test_s3_protocolsrsQ  *C2778 BJY'MOOJ .r#ctjdtjdd|jd}|j|ddd|t |d|}t j ||y)Nrrrz /test.parquetF)rIrrrr)r,r-rrr rTr)s3_bucket_publicrrAfnrXs r"test_s3_parquetrsq  &  !&&'} 5BNN % 4QU r- FCOOCr#rc~d}tjt|5tddddy#1swYyxYw)NzH`Import fsspec` failed. Use pip or conda to install the fsspec package.r^rK)r,rcrfr)msgs r"test_not_present_exceptionrs3 TC {# .+)*+++s 3<ctjdtddgi}|jdddi|jddk(sJt dddi}|jddk(sJt j||y) Nrrrtestmem://mockfiler feather_writer feather_read)r,r-r to_featherrr rTrUr3routs r"test_feather_optionsrs  " C!: BMM&8QMR ??1  00 0 +fn=U VC ??1  // /"c"r#ctddgi}|jdddi|jddk(sJtdddi}|jddk(sJt j ||y)Nrrrr pickle_writer pickle_read)r to_picklerr rTrUrs r"test_pickle_optionsr!sx C!: BLL%7OLP ??1  // / *V]>. // /r#cvtjdtjtd5t dddidddtjtd5t dddidddt j}tjtd5t |ddidddtddgi}tjtd5|jd ddidddy#1swYxYw#1swYxYw#1swYixYw#1swYyxYw) Nrrr^ localfilerTrr nonfsspecpath) r,r-rcrdrr 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);rnumpyr?r,pandas._configr pandas.compatrpandas.compat.pyarrowrpandasrrrr r r r r rrpandas._testing_testingrT pandas.utilrtdmarkfilterwarnings pytestmarkfixturer3rArFrZrirmrt parametrizer|rrrxfailrrr single_cpurrrskip_if_installedrrrrrrrrr#r"rs  -%6   . [[ ' 'C *) /) ) 6)7).,, (E[E1E-E + 0&0(-/BC/D/%9:/;/%'0PQ R h+ + ###"30Dr#