§
    tŠtjk  ã                   óz  — d Z ddlZddlmZ ddlZddlZddlZddlmZ ddl	m
Z ej                             d¦  «        Z ej        ddg¬¦  «        d	„ ¦   «         Zej        d
„ ¦   «         Zej                             dg d¢¦  «        d„ ¦   «         Zej                             dddg¦  «        d„ ¦   «         Zd„ Zd„ Zej                             dddg¦  «        d„ ¦   «         Zej                             dg d¢¦  «        d„ ¦   «         Zd„ Zej                             dg d¢¦  «        d„ ¦   «         Zd„ Zd„ Zd„ ZdS ) zZ
Tests compressed data parsing functionality for all
of the parsers defined in parsers.py
é    N)ÚPath)Ú	DataFramez=ignore:Passing a BlockManager to DataFrame:DeprecationWarningTF)Úparamsc                 ó   — | j         S )N)Úparam)Úrequests    úe/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/io/parser/test_compression.pyÚbufferr
      s
   € àŒ=Ðó    c                 ó²   — | }t          |d¦  «        5 }|                     ¦   «         }d d d ¦  «         n# 1 swxY w Y   |                     |¦  «        }|||fS )NÚrb)ÚopenÚreadÚread_csv)Úall_parsersÚcsv1ÚparserÚfÚdataÚexpecteds         r	   Úparser_and_datar      s”   € à€Få	ˆd�DÑ	Ô	ð ˜QØ�vŠv‰xŒxˆðð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð à�Š˜tÑ$Ô$€Hà�4˜Ð!Ð!s   “4´8»8Úcompression)ÚzipÚinferÚzip2c                 óˆ  — |\  }}}| dz  }t          j        |d¬¦  «        5 }|                     d|¦  «         d d d ¦  «         n# 1 swxY w Y   |dk    r@t          |d¦  «        5 }|                     |d¬¦  «        }	d d d ¦  «         n# 1 swxY w Y   n|                     ||¬¦  «        }	t          j        |	|¦  «         d S )	Nútest_file.zipÚw©ÚmodeÚ	test_filer   r   r   ©r   )ÚzipfileÚZipFileÚwritestrr   r   ÚtmÚassert_frame_equal)
Útmp_pathr   r   r   r   r   ÚpathÚtmpr   Úresults
             r	   Útest_zipr,   %   sH  € à,Ñ€FˆD�(à�oÑ%€DÝ	Œ˜ CÐ	(Ñ	(Ô	(ð (¨CØ�Š�[ $Ñ'Ô'Ð'ð(ð (ð (ñ (ô (ð (ð (ð (ð (ð (ð (øøøð (ð (ð (ð (ð �fÒÐÝ�$˜ÑÔð 	; Ø—_’_ Q°E�_Ñ:Ô:ˆFð	;ð 	;ð 	;ñ 	;ô 	;ð 	;ð 	;ð 	;ð 	;ð 	;ð 	;øøøð 	;ð 	;ð 	;ð 	;øð —’ °;�Ñ?Ô?ˆåÔ˜& (Ñ+Ô+Ð+Ð+Ð+s#   ¢AÁA	ÁA	Á&B
Â
BÂBr   r   c                 óN  — |\  }}}| dz  }ddg}t          j        |d¬¦  «        5 }|D ]}	|                     |	|¦  «         Œ	 d d d ¦  «         n# 1 swxY w Y   t          j        t
          d¬¦  «        5  |                     ||¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )	Nzcombined_zip.zipr!   Úsecond_filer   r   zMultiple files©Úmatchr"   )r#   r$   r%   ÚpytestÚraisesÚ
ValueErrorr   )
r(   r   r   r   r   r   r)   Úinner_file_namesr*   Ú	file_names
             r	   Útest_zip_error_multiple_filesr6   6   sH  € à,Ñ€FˆD�(àÐ(Ñ(€DØ# ]Ð3Ðå	Œ˜ CÐ	(Ñ	(Ô	(ð *¨CØ)ð 	*ð 	*ˆIØ�LŠL˜ DÑ)Ô)Ð)Ð)ð	*ð*ð *ð *ñ *ô *ð *ð *ð *ð *ð *ð *øøøð *ð *ð *ð *õ 
Œ•zÐ)9Ð	:Ñ	:Ô	:ð 7ð 7Ø�Š˜¨+ˆÑ6Ô6Ð6ð7ð 7ð 7ñ 7ô 7ð 7ð 7ð 7ð 7ð 7ð 7ð 7øøøð 7ð 7ð 7ð 7ð 7ð 7s#   ¦AÁAÁAÁ5BÂBÂ!Bc                 ó  — |\  }}}| dz  }t          j        |d¬¦  «        5  	 d d d ¦  «         n# 1 swxY w Y   t          j        t          d¬¦  «        5  |                     |d¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr   r   r   z
Zero filesr/   r   r"   )r#   r$   r1   r2   r3   r   )r(   r   r   Ú_r)   s        r	   Útest_zip_error_no_filesr9   E   s  € Ø"�L€FˆAˆqà�oÑ%€DÝ	Œ˜ CÐ	(Ñ	(Ô	(ð ð Øðð ð ñ ô ð ð ð ð ð ð øøøð ð ð ð õ 
Œ•z¨Ð	6Ñ	6Ô	6ð 1ð 1Ø�Š˜¨%ˆÑ0Ô0Ð0ð1ð 1ð 1ñ 1ô 1ð 1ð 1ð 1ð 1ð 1ð 1ð 1øøøð 1ð 1ð 1ð 1ð 1ð 1s   ¢0°4·4ÁA;Á;A?ÂA?c                 ó4  — |\  }}}| dz  }|                      ¦   «          t          |d¦  «        5 }t          j        t          j        d¬¦  «        5  |                     |d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d d d ¦  «         d S # 1 swxY w Y   d S )Nzinvalid_file.zipr   zFile is not a zip filer/   r   r"   )Útouchr   r1   r2   r#   Ú
BadZipFiler   )r(   r   r   r8   r)   r   s         r	   Útest_zip_error_invalid_zipr=   P   s  € Ø"�L€FˆAˆqàÐ(Ñ(€DØ‡J‚J�L„L€LÝ	ˆd�DÑ	Ô	ð 2˜QÝŒ]�7Ô-Ð5MÐNÑNÔNð 	2ð 	2Ø�OŠO˜A¨5ˆOÑ1Ô1Ð1ð	2ð 	2ð 	2ñ 	2ô 	2ð 	2ð 	2ð 	2ð 	2ð 	2ð 	2øøøð 	2ð 	2ð 	2ð 	2ð2ð 2ð 2ñ 2ô 2ð 2ð 2ð 2ð 2ð 2ð 2ð 2øøøð 2ð 2ð 2ð 2ð 2ð 2s5   °!BÁA5Á)BÁ5A9	Á9BÁ<A9	Á=BÂBÂBÚfilenamez
test.{ext}c                 óô  — |\  }}}	|}
||
         }|€|n|                      |¬¦  «        }|r5|r3|                     t          j                             d¬¦  «        ¦  «         |r| |z  n| dz  }t          j        |
||¦  «         |rdn|
}|r@t          |d¦  «        5 }|                     ||¬¦  «        }d d d ¦  «         n# 1 swxY w Y   n|                     ||¬¦  «        }t          j	        ||	¦  «         d S )N)Úextz9Cannot deduce compression from buffer of compressed data.)Úreasonr!   r   r   r"   )
ÚformatÚapplymarkerr1   ÚmarkÚxfailr&   Úwrite_to_compressedr   r   r'   )r(   r   r   Úcompression_onlyr
   r>   Úcompression_to_extensionr   r   r   Úcompress_typer@   r)   r   r   r+   s                   r	   Útest_compressionrJ   Z   s~  € ð -Ñ€FˆD�(Ø$€Mà
" =Ô
1€CØ#Ð+ˆxˆx°·²ÀS°Ñ1IÔ1I€Hàð 
�Fð 
Ø×ÒÝŒK×ÒØRð ñ ô ñ	
ô 	
ð 	
ð #+ÐFˆ8�hÑÐ°¸;Ñ0F€DÝÔ˜=¨$°Ñ5Ô5Ð5Ø%Ð8�'�'¨=€Kàð @Ý�$˜ÑÔð 	A Ø—_’_ Q°K�_Ñ@Ô@ˆFð	Að 	Að 	Añ 	Aô 	Að 	Að 	Að 	Að 	Að 	Að 	Aøøøð 	Að 	Að 	Að 	Aøð —’ °;�Ñ?Ô?ˆåÔ˜& (Ñ+Ô+Ð+Ð+Ð+s   ÂC Ã CÃCr@   )NÚgzÚbz2c                 ó  — | }dddœ} |j         |fi |¤Ž}d|d<   |r8t          |d¬¦  «        5 } |j         |fi |¤Ž}d d d ¦  «         n# 1 swxY w Y   n|rd|z   nd	} |j         ||z   fi |¤Ž}t          j        ||¦  «         d S )
Nr   T)Ú	index_colÚparse_datesr   r   úutf-8©Úencodingú.Ú )r   r   r&   r'   )	r   r   r
   r@   r   Úkwargsr   r   r+   s	            r	   Útest_infer_compressionrV   ~   s
  € ð €FØ¨TÐ2Ð2€FàˆvŒ˜tÐ.Ð. vÐ.Ð.€HØ#€Fˆ=Ñàð 7Ý�$ Ð)Ñ)Ô)ð 	2¨QØ$�V”_ QÐ1Ð1¨&Ð1Ð1ˆFð	2ð 	2ð 	2ñ 	2ô 	2ð 	2ð 	2ð 	2ð 	2ð 	2ð 	2øøøð 	2ð 	2ð 	2ð 	2øð Ð&ˆc�C‰iˆi BˆØ �” ¨¡Ð6Ð6¨vÐ6Ð6ˆåÔ˜& (Ñ+Ô+Ð+Ð+Ð+s   ®A	Á	AÁAc                 ó  — | }|                      |¦  «        }t          j                             |d|› d�¦  «        }|                     ||dd¬¦  «        }t          ddgddgd	œ¦  «        }t          j        ||¦  «         d S )
NÚutfz_ex_small.zipr   ú	)rR   r   ÚsepÚ	Venezuelaõ   Hugo ChÃ¡vez FrÃ­asúHenrique Capriles R.©ÚCountryÚTwitter)rB   Úosr)   Újoinr   r   r&   r'   )	r   Úcsv_dir_pathÚ	utf_valueÚencoding_fmtr   rR   r)   r+   r   s	            r	   Útest_compression_utf_encodingrf   ‘   s™   € à€FØ×"Ò" 9Ñ-Ô-€HÝŒ7�<Š<˜Ð&D¨IÐ&DÐ&DÐ&DÑEÔE€Dà�_Š_˜T¨HÀ%ÈTˆ_ÑRÔR€FÝà# [Ð1Ø-Ð/EÐFð	
ð 	
ñô €Hõ Ô˜& (Ñ+Ô+Ð+Ð+Ð+r   Úinvalid_compression)ÚsfarkÚbz3Úzipperc                 óž   — | }d|i}d|› �}t          j        t          |¬¦  «        5   |j        di |¤Ž d d d ¦  «         d S # 1 swxY w Y   d S )Nr   zUnrecognized compression type: r/   r   )r   )r1   r2   r3   r   )r   rg   r   Úcompress_kwargsÚmsgs        r	   Útest_invalid_compressionrn   ¢   s±   € à€FØ$Ð&9Ð:€Oà
AÐ,?Ð
AÐ
A€Cå	Œ•z¨Ð	-Ñ	-Ô	-ð <ð <ØˆŒÐ;Ð;¨?Ð;Ð;Ð;ð<ð <ð <ñ <ô <ð <ð <ð <ð <ð <ð <ð <øøøð <ð <ð <ð <ð <ð <s   §AÁAÁ	Ac                 óª   — | }t           j                             |d¦  «        }|                     |¦  «        }t	          |j        ¦  «        dgk    sJ ‚d S )Nztar_csv.tar.gzÚa)ra   r)   rb   r   ÚlistÚcolumns)r   rc   r   r)   Údfs        r	   Útest_compression_tar_archivert   ­   sQ   € Ø€FÝŒ7�<Š<˜Ð&6Ñ7Ô7€DØ	�Š˜Ñ	Ô	€BÝ�”
ÑÔ ˜uÒ$Ð$Ð$Ð$Ð$Ð$r   c                 óP  — |}t          dddgi¦  «        }| dz  }| dz  }|                     |d¬¦  «         t          |¦  «                             t          |¦  «                             d¬	¦  «        d¬	¦  «         t          j        |                     |d ¬
¦  «        |¦  «         d S )Nrp   r   é   ztest.csvztest.csv.zipF©ÚindexrP   rQ   r"   )r   Úto_csvr   Ú
write_textÚ	read_textr&   r'   r   )r(   r   r   rs   Úpath_csvÚpath_zips         r	   Ú!test_ignore_compression_extensionr~   ´   s°   € Ø€FÝ	�C˜!˜Q˜�=Ñ	!Ô	!€Bà˜*Ñ$€HØ˜.Ñ(€Hà‡I‚Iˆh˜e€IÑ$Ô$Ð$Ýˆ�N„N×ÒÝˆX‰Œ× Ò ¨'Ð Ñ2Ô2¸Wð ñ ô ð õ Ô˜&Ÿ/š/¨(À˜/ÑEÔEÀrÑJÔJÐJÐJÐJr   c                 óÈ  — |}t          ddgddgdœ¦  «        }| dz  }|                     |d¬¦  «         t          j        |                     |¦  «        |¦  «         t          j        |d¦  «        5 }|                     |                     |                     ¦   «         d	         ¦  «        d
¬¦  «        }t          j        ||¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr[   r\   r]   r^   ztest.tar.gzFrw   zr:gzr   r   r"   )	r   ry   r&   r'   r   Útarfiler   ÚextractfileÚgetnames)r(   r   r   r   Útar_pathÚtarr+   s          r	   Útest_writes_tar_gzr…   Ã   s4  € Ø€FÝà# [Ð1Ø-Ð/EÐFð	
ð 	
ñô €Dð ˜-Ñ'€HØ‡K‚K� €KÑ&Ô&Ð&õ Ô˜&Ÿ/š/¨(Ñ3Ô3°TÑ:Ô:Ð:õ 
Œ�h Ñ	'Ô	'ð ,¨3Ø—’Ø�OŠO˜CŸLšL™NœN¨1Ô-Ñ.Ô.¸Gð !ñ 
ô 
ˆõ 	Ô˜f dÑ+Ô+Ð+ð	,ð ,ð ,ñ ,ô ,ð ,ð ,ð ,ð ,ð ,ð ,ð ,øøøð ,ð ,ð ,ð ,ð ,ð ,s   Á2ACÃCÃC)Ú__doc__ra   Úpathlibr   r€   r#   r1   Úpandasr   Úpandas._testingÚ_testingr&   rD   ÚfilterwarningsÚ
pytestmarkÚfixturer
   r   Úparametrizer,   r6   r9   r=   rJ   rV   rf   rn   rt   r~   r…   © r   r	   ú<module>r�      sA  ððð ð
 
€	€	€	Ø Ð Ð Ð Ð Ð Ø €€€Ø €€€à €€€à Ð Ð Ð Ð Ð Ø Ð Ð Ð Ð Ð àŒ[×'Ò'ØCñô €
ð
 €„˜˜e�}Ð%Ñ%Ô%ðð ñ &Ô%ðð „ð"ð "ñ „ð"ð „×Ò˜Ð(@Ð(@Ð(@ÑAÔAð,ð ,ñ BÔAð,ð  „×Ò˜¨°Ð(8Ñ9Ô9ð7ð 7ñ :Ô9ð7ð1ð 1ð 1ð2ð 2ð 2ð „×Ò˜ d¨LÐ%9Ñ:Ô:ð ,ð  ,ñ ;Ô:ð ,ðF „×Ò˜Ð 3Ð 3Ð 3Ñ4Ô4ð,ð ,ñ 5Ô4ð,ð$,ð ,ð ,ð" „×ÒÐ.Ð0JÐ0JÐ0JÑKÔKð<ð <ñ LÔKð<ð%ð %ð %ðKð Kð Kð,ð ,ð ,ð ,ð ,r   