Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/formats/__pycache__/test_to_csv.cpython-37.pyc
Назад
B �5�g�^ � @ sn d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl m Z G dd� d�Z dd� Zdd� ZdS ) � N)� DataFrame�compatc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zej�dejdggdggd gd �d fejdgd ggdgdggd d gd �d!fg�ej�d"eejg�d#d$� ��Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zejj e � � d/d0�d1d2� �Z d3d4� Z ej�d5d6d7g�ej�d8d6d7g�d9d:� ��Z!d;d<� Z"d=d>� Z#ej�d?d@dAg�ej�dBdCdDdEg�dFdG� ��Z$ej�dHdIg�dJdK� �Z%dLdM� Z&dNdO� Z'ej�dPdQdRdSg�dTdU� �Z(ej�dVdWdXg�dYdZ� �Z)ej�dVdWdXg�d[d\� �Z*dCS )]� TestToCSVc C s� t d dg�}d}t�d��:}|j|d d d� t|��}|�� |ksFt�W d Q R X W d Q R X t dd g�}d}t�d��:}|j|d d d� t|��}|�� |ks�t�W d Q R X W d Q R X d S )N� z"" 1.0 ztest.csv)�header�indexz1.0 "" )r �tm� ensure_clean�to_csv�open�read�AssertionError)�selfZdf1Z expected1�path�fZdf2Z expected2� r �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/formats/test_to_csv.py� test_to_csv_with_single_column s $ z(TestToCSV.test_to_csv_with_single_columnc C sN t dddddgi�}t�d��&}|�|� t�tj|dd�|� W d Q R X d S ) N�colZAAAAAu ÄÄÄÄÄu ßßßßßu 聞聞聞聞聞ztest.csvr )� index_col)r r r r �assert_frame_equal�pd�read_csv)r �dfr r r r � test_to_csv_defualt_encoding- s z&TestToCSV.test_to_csv_defualt_encodingc C s� t dddgi�}d}t�d��8}|j|dd� t|��}|�� |ksHt�W d Q R X W d Q R X d}t�d��:}|j|ddd � t|��}|�� |ks�t�W d Q R X W d Q R X t�d��0}tjt d d �� |j|dd d � W d Q R X W d Q R X d S ) Nr r � z"","col" "0","1" "1","2" ztest.csv)�quotingz$$,$col$ $0$,$1$ $1$,$2$ �$)r � quotecharr )�match) r r r r r r r �pytest�raises� TypeError)r r �expectedr r r r r �test_to_csv_quotechar6 s $ $ z TestToCSV.test_to_csv_quotecharc C s� t dddgi�}d}t�d��:}|j|ddd� t|��}|�� |ksJt�W d Q R X W d Q R X d d lm} t�d��.}t j |d d �� |j|d d� W d Q R X W d Q R X d S )Nr za"az"bb"z!"","col" "0","a""a" "1","""bb""" ztest.csvr T)r � doublequoter )�Error� escapechar)r F)r% ) r r r r r r r �_csvr& r r! )r r r# r r r& r r r �test_to_csv_doublequoteR s $ z!TestToCSV.test_to_csv_doublequotec C s� t dddgi�}d}t�d��<}|j|dddd � t|��}|�� |ksLt�W d Q R X W d Q R X t dd d gi�}d }t�d��:}|j|d dd� t|��}|�� |ks�t�W d Q R X W d Q R X d S )Nr za"az"bb"z!"","col" "0","a\"a" "1","\"bb\"" ztest.csvr F�\)r r% r' za,az,bb,z,col 0,a\,a 1,\,bb\, � )r r' )r r r r r r r )r r r# r r r r r �test_to_csv_escapechare s $ z TestToCSV.test_to_csv_escapecharc C s8 t dddgi�}dddg}t�|�}|�� |ks4t�d S )Nr r r z,colz0,1z1,2)r r � convert_rows_list_to_csv_strr r )r r � expected_rowsr# r r r �test_csv_to_string~ s z TestToCSV.test_csv_to_stringc C s2 t dgdgdgd��}ddg}t�|�}|�� |ks8t�ddg}t�|�}|jd d d �|ks`t�dd g}t�|�}|jd d�|ks�t�ddg}t�|�}|jd d d d�|ks�t�t ddgddgdd��}dddg}t�|�}|jddd�|ks�t�|�d�jdd �|k�st�|�dd g�jdd �|k�s.t�d S ) Nr �ag333333$@)Zcol1Zcol2Zcol3z,col1,col2,col3z 0,1,a,10.1z;col1;col2;col3z 0;1;a;10,1�,�;)�decimal�sepz 0,1,a,10.10z%.2f)� float_formatz 0;1;a;10,10)r3 r4 r5 r g�������?g������@gffffff @)r0 �b�cza,b,cz 0^0,2^2,1z 1^1,3^3,1F�^)r r3 )r3 r6 )r r r- r r � set_index)r r r. Zexpected_defaultZexpected_european_excelZ expected_float_format_defaultZexpected_float_formatr# r r r �test_to_csv_decimal� s( z TestToCSV.test_to_csv_decimalc C sh t ddgddgdd��}dddg}t�|�}|�d �jd d �|ksFt�|�d d g�jd d �|ksdt�d S ) Nr r g������@gffffff @)r0 r6 r7 za,b,cz0,2.20,1z1,3.30,1r0 z%.2f)r5 r6 )r r r- r9 r r )r r r. r# r r r �test_to_csv_float_format� s z"TestToCSV.test_to_csv_float_formatc C sv t dtjgddgddgd��}dddg}t�|�}|�d �jd d �|ksLt�|�d d g�jd d �|ksjt�t tjddgddgd��}dd dg}t�|�}|�d �jd d �|ks�t�|�d d g�jd d �|ks�t�t dddgddgd��}dddg}t�|�}|�d �jd d �|k�st�|�d d g�jd d �|k�s8t�t� d tj dg�jdd �}t�ddddg�}||k�srt�d S )Nr r r r+ )r0 r6 r7 za,b,cz0.0,0,2z_,1,3r0 �_)�na_repr6 z_,0,2z0,0,2z0,1,3r7 �ZZZZZz,0z0,az1,ZZZZZz2,c) r �np�NaNr r- r9 r r r �Series�NA)r r r. r# �csvr r r �test_to_csv_na_rep� s$ z TestToCSV.test_to_csv_na_repc C s@ t �ddddg�}tjdtjdg|d�jdd �}||ks<t�d S ) Nz,0z0,az1,ZZZZZz2,cr0 r7 )Zdtyper> )r= )r r- r rA rB r r )r Znullable_string_dtyper# rC r r r �"test_to_csv_na_rep_nullable_string� s z,TestToCSV.test_to_csv_na_rep_nullable_stringc C sL t dtjdddd�i�}t dtjdddd�i�}ddd d d d g}t�|�}|�� |ksZt�ddd dddg}t�|�}|jdd�|ks�t�ddddddg}t�|�}|jdd�|ks�t�dddddd g}t�|�}|�� |ks�t�|jdd�|ks�t�d |d < d |d < d!d"g}t�|�}|�tjdd#d$�d g�}|� � jdd�|k�sHt�d S )%N�AZ20130101� �s)Zperiods�freq�dz,Az0,2013-01-01 00:00:00z1,2013-01-01 00:00:01z2,2013-01-01 00:00:02z3,2013-01-01 00:00:03z4,2013-01-01 00:00:04z1,2013-01-02 00:00:00z2,2013-01-03 00:00:00z3,2013-01-04 00:00:00z4,2013-01-05 00:00:00z%Y-%m-%d %H:%M:%S)Z date_formatz 0,2013-01-01z 1,2013-01-01z 2,2013-01-01z 3,2013-01-01z 4,2013-01-01z%Y-%m-%dz 1,2013-01-02z 2,2013-01-03z 3,2013-01-04z 4,2013-01-05r �Br �CzA,B,Cz2013-01-01,0,1.0Z1h)�keyrI ) r r Z date_ranger r- r r �groupbyZGrouperZmean) r Zdf_secZdf_dayr. Zexpected_default_secZexpected_ymdhms_dayZexpected_ymd_secZexpected_default_dayZdf_sec_groupedr r r �test_to_csv_date_format� sR z!TestToCSV.test_to_csv_date_formatc C sX t dgtj�dgdgg�d�}dddg}t�|�}|�� |ksBt�dddg}t�|�}|jd d �|ksjt�t dgtj�dgdgg�tj�dgdgg�d �}d d dg}t�|�}|�� |ks�t�dddg}t�|�}|jd d �|ks�t�t dgtj�dgdgg�d�}dddg}t�|�}|�� |k�s*t�dddg}t�|�}|jd d �|k�sTt�d S )Nr r )�columnsz,1z,2z0,1�1�2F)r )rP r z,,1z,,2z1,2,1�foo�barz,fooz,bar)r r � MultiIndexZ from_arraysr r- r r )r r Zexp_rows�expr r r �test_to_csv_multi_index s0 z!TestToCSV.test_to_csv_multi_indexz ind,expectedg �?r �x)Zlevels�codes�namesz x,data 1.0,1 g @�yzx,y,data 1.0,2.0,1 �klassc C s0 |t jdg|dd��jddd�}||ks,t�d S )Nr �data)�name� T)�line_terminatorr )r rA r r )r �indr# r\ �resultr r r �$test_to_csv_single_level_multi_index? s z.TestToCSV.test_to_csv_single_level_multi_indexc C sp dddgidddgig}t |�}d}t�d��8}|j|dd � t|��}|�� |ksXt�W d Q R X W d Q R X d S ) NrZ rS rT �baz�quxz-,names 0,"['foo', 'bar']" 1,"['baz', 'qux']" z str_test.csv�ascii)�encoding)r r r r r r r )r � str_arrayr �expected_asciir r r r r � test_to_csv_string_array_asciiV s z(TestToCSV.test_to_csv_string_array_asciic C sp dddgidddgig}t |�}d}t�d��8}|j|dd � t|��}|�� |ksXt�W d Q R X W d Q R X d S ) NrZ rS rT rd re z-,names 0,"['foo', 'bar']" 1,"['baz', 'qux']" zunicode_test.csvzutf-8)rg )r r r r r r r )r rh r Z expected_utf8r r r r r � test_to_csv_string_array_utf8d s z'TestToCSV.test_to_csv_string_array_utf8c C s: dddgdddgd�}t |�}t�d��f}tj�d �}d | d | d | d | }|j|dd� t|d��}|�� |ks�t �W d Q R X W d Q R X t�d��@}d}|j|ddd� t|d��}|�� |ks�t �W d Q R X W d Q R X t�d��B}d}|j|ddd� t|d��}|�� |k�s"t �W d Q R X W d Q R X d S )Nr r r+ �abczd efzg h i)�intZstr_lfz lf_test.csvzutf-8s int,str_lfs 1,abcs 2,"d ef"s 3,"g h i"F)r �rbs% int,str_lf 1,abc 2,"d ef" 3,"g h i" r_ )r` r s) int,str_lf 1,abc 2,"d ef" 3,"g h i" z ) r r r �os�linesep�encoder r r r ) r r] r r � os_linesep�expected_noargr � expected_lf� expected_crlfr r r �test_to_csv_string_with_lfr s$ $ $ z$TestToCSV.test_to_csv_string_with_lfc C s: dddgdddgd�}t |�}t�d��f}tj�d �}d | d | d | d | }|j|dd� t|d��}|�� |ks�t �W d Q R X W d Q R X t�d��@}d}|j|ddd� t|d��}|�� |ks�t �W d Q R X W d Q R X t�d��B}d}|j|ddd� t|d��}|�� |k�s"t �W d Q R X W d Q R X d S )Nr r r+ rl zd efz g h i)rm Zstr_crlfz crlf_test.csvzutf-8s int,str_crlfs 1,abcs 2,"d ef"s 3,"g h i"F)r rn s+ int,str_crlf 1,abc 2,"d ef" 3,"g h i" r_ )r` r s/ int,str_crlf 1,abc 2,"d ef" 3,"g h i" z ) r r r ro rp rq r r r r ) r r] r r rr rs r rt ru r r r � test_to_csv_string_with_crlf� s$ $ $ z&TestToCSV.test_to_csv_string_with_crlfc C sf t ddgddggddgd�}dd d g}t�|�}|jtjd d � |�� }|j|ksVt�tjj rbt�d S ) NrS rT rd re Zname_1Zname_2)rP z,name_1,name_2z 0,foo,barz 1,baz,quxrf )rg ) r r r- r �sys�stdoutZ readouterr�outr �closed)r Zcapsysr r. ri Zcapturedr r r �test_to_csv_stdout_file� s z!TestToCSV.test_to_csv_stdout_filez�Especially in Windows, file stream should not be passedto csv writer without newline='' option.(https://docs.python.org/3.6/library/csv.html#csv.writer))�reasonc C s� t ddddgi�}d}t�d��Z}t|d�� }|�d� |j|d d d � W d Q R X t|��}|�� |kslt�W d Q R X W d Q R X d S ) Nr0 rX r[ �zzmanual header x y z ztest.txt�wzmanual header )r r )r r r r �writer r r )r r r# r r r r r � test_to_csv_write_to_open_file� s z(TestToCSV.test_to_csv_write_to_open_filec C s� t ddddgi�}dddg}dt�|� }t�d��f}t|ddd �� }|�d� |j|d d d � W d Q R X t|d �� }|�� t|d �ks�t �W d Q R X W d Q R X d S ) Nr0 rX r[ r~ zmanual header ztest.txtr � )�newline)r r rn zutf-8) r r r- r r r� r r �bytesr )r r r. r# r r r r r �/test_to_csv_write_to_open_file_with_newline_py3� s z9TestToCSV.test_to_csv_write_to_open_file_with_newline_py3�to_inferTF� read_inferc C s� |}|dkr t �|� d�� d}|dkr2|d7 }n||7 }tddgi�}|rPdn|}|r\dn|}t�|��0} |j| |d � tj| d |d �} t�| |� W d Q R X d S ) N�zipz is not supported for to_csvztest.�gzip�gzrF r �infer)� compressionr )r r� ) r �skipr r r r r r r ) r �compression_onlyr� r� r� �filenamer Zto_compressionZread_compressionr rb r r r �test_to_csv_compression� s z!TestToCSV.test_to_csv_compressionc C sp |}t ddgi�}d}||dkr$dn|7 }t�|��2}|j|d|id� tj|dd �}t�||� W d Q R X d S ) N�ABCr zto_csv_compress_as_dict.r� r� �method)r� r )r )r r r r r r r )r r� r� r r� r Zread_dfr r r � test_to_csv_compression_dict s z&TestToCSV.test_to_csv_compression_dictc C s\ t ddgi�}ddi}d}t�d��.}tjt|d�� |j||d� W d Q R X W d Q R X d S ) Nr� r Z some_optionTzmust have key 'method'zout.zip)r )r� )r r r r r! � ValueErrorr )r r r� �msgr r r r �-test_to_csv_compression_dict_no_method_raises s z7TestToCSV.test_to_csv_compression_dict_no_method_raisesr� r� r� � archive_nameNztest_to_csv.csvztest_to_csv.zipc C s� ddl m} tddgi�}t�d��|}|j|||d�d� ||��T}|d krP|n|}tj�|�}t |j �dksrt �tj�|j d j �}||ks�t �W d Q R X W d Q R X d S )Nr )�ZipFiler� r zto_csv_archive_name.zip)r� r� )r� ) �zipfiler� r r r r ro r �basename�len�filelistr r� ) r r� r� r� r r ZzpZexpected_arcnameZ archived_filer r r �test_to_csv_zip_arguments s z#TestToCSV.test_to_csv_zip_arguments� df_new_typeZInt64c C sV t dtd�gd i�}|�|�}ddddg}t�|�}|jdddd�}||ksRt�d S )Nr7 �nanr+ ZmynullFrf )r r= rg )r �float�astyper r- r r )r r� r r. r# rb r r r � test_to_csv_na_rep_long_string3 s z(TestToCSV.test_to_csv_na_rep_long_stringc C sR t �ddg��d�}t�� }|�|� |�� }dddg}t�|�}||ksNt �d S )Nr ztimedelta64[ns]z,0z0,0 days 00:00:00.000000001z1,0 days 00:00:00.000000001) r rA r� �io�StringIOr �getvaluer r- r )r rH �bufrb r. r# r r r � test_to_csv_timedelta_precision? s z)TestToCSV.test_to_csv_timedelta_precisionc C s� t �tdd��jdd�}t�ddddd g�}||ks8t�t �d d g�jd d�}t�dd dg�}||ksjt�t �ddg�jdd�}t�dddg�}||ks�t�d S )N� � �-)r= z,0z0,8z1,9z2,10z3,11TFr� z0,Truez1,Falseg�������?g������@�.z0,1.1z1,2.2)r rA �ranger r r- r )r rb r# r r r �test_na_rep_truncatedM s z TestToCSV.test_na_rep_truncated�errors� surrogatepass�ignore�replacec C sB dg}t j|t �|�d�}t�d��}|j||d� W d Q R X d S )Nu �foo)r ztest.csv)r� )r rA �Indexr r r )r r� r] Zserr r r r �test_to_csv_errors[ s z TestToCSV.test_to_csv_errors�mode�wbr c C s\ t �� }t �� �B}t|dd��}|j||d� W dQ R X t �|tj|dd�� W dQ R X dS )z� Binary file objects should work (if 'mode' contains a 'b') or even without it in most cases. GH 35058 and GH 19827 zw+b)r� Nr )r )r � makeDataFramer r r r r r )r r� r r �handler r r �test_to_csv_binary_handlee s z#TestToCSV.test_to_csv_binary_handlec C s� d� d�}t�|�}tj|dd�}t�� }|j||ddd� |�d� |�� �|�sXt �t � � �J}t |d��4}t � j||dd� |�d� |�� �d �s�t �W d Q R X W d Q R X d S ) z{ Binary file objects should honor a specified encoding. GH 23854 and GH 13068 with binary handles u a, b, 🐟z utf-8-sig)rg F)r� rg r r zw+b)r� rg s ""N)rq r� �BytesIOr r r �seekr� � startswithr r r r r r )r r� �content�bufferr r r� r r r �"test_to_csv_encoding_binary_handles s z,TestToCSV.test_to_csv_encoding_binary_handle)+�__name__� __module__� __qualname__r r r$ r) r, r/ r: r; rD rE rO rW r �markZ parametrizer rU r rA rc rj rk rv rw r| Zxfailr Zis_platform_windowsr� r� r� r� r� r� r� r� r� r� r� r� r r r r r sT $ ?$ "' r c C sH t �� }t �� �.}|j|| dd� t �tj|| dd�|� W d Q R X d S )Nr )r� � chunksizer )r� r )r r� r r r r r )r� r r r r r �&test_to_csv_iterative_compression_name� s r� c C s\ t �� }t�� �B}|j|| dd� |�d� t �tj|| dd�|� |j rNt �W d Q R X d S )Nr )r� r� r )r� r ) r r� r� r� r r� r r r r{ r )r� r r� r r r �(test_to_csv_iterative_compression_buffer� s r� )r� ro rx �numpyr? r Zpandasr r r Zpandas._testingZ_testingr r r� r� r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка