Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/__pycache__/test_to_csv.cpython-37.pyc
Назад
B �5�gy� � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlmZ d dl Z d dl m Z m Z m Z mZmZmZmZmZmZ d dlmZ d dlm mZ d dlm Z dddgZ d d d d d dddgZ G dd� d�Z dS )� N)�StringIO)� ParserError) � DataFrame�Index� MultiIndex�NaT�Series� Timestamp� date_range�read_csv� to_datetime)� get_handleZfloat16�float32�float64Zuint8Zuint16Zuint32Zuint64Zint8Zint16Zint32�int64c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z e j j dd� �Ze j j dd� �Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Ze j j d-d.� �Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?d@� Z%dAdB� Z&dCdD� Z'e j �(dEe)dFdGdHgdIdJdKggdLdMgdNdOdPgdQ�dRfe)dSdTdUggdNdOdPgdV�dWfe)dXdYdZd[gg dNdOdPgdV�d\fe)dXdYd]d^gg dNdOdPgdV�d_fg�d`da� �Z*dbdc� Z+ddde� Z,dfdg� Z-dhdi� Z.djdk� Z/dldm� Z0dndo� Z1dpdq� Z2drds� Z3dRS )t�TestDataFrameToCSVc K s" ddd�}|j f |� t|f|�S )Nr T)� index_col� parse_dates)�updater )�self�path�kwargs�params� r �\/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/methods/test_to_csv.pyr ) s zTestDataFrameToCSV.read_csvc C sJ t �d���4}tj|d d d�< |�|� |j|ddgd� |j|dd� |j|dd� |j�d �|_|�|� | �|�}t �||� |j|d d � | j|d d �}t |j �t |j �d ks�t �|j|dd� | j|d d �}t � |j |j � tttd �t�d �d�ttd�t�d�d�d��}|�|� | �|�}t �||� W d Q R X d S )NZ__tmp_to_csv_from_csv1__�A� �B)�columnsF)�header)�indexr )� index_label)r � � � )�s1�s2)�tm� ensure_clean�np�nan�to_csvr � _with_freqr �assert_frame_equal�lenr �AssertionError�assert_almost_equal�valuesr r �range�arange)r � float_frame�datetime_framer �recons�dmr r r �test_to_csv_from_csv1/ s, z(TestDataFrameToCSV.test_to_csv_from_csv1c C s0 t �d���}ttj�dd�dddgdddgd�}|�|� | �|�}t �||� t � d d d g�}ttj�dd�|dddgd�}|�|� | j|d d d gdd�}t j||dd� t ddddg�}|j||d� | �|�}|� � }||_ t �||� d} tjt| d�� |j|ddgd� W d Q R X W d Q R X d S )NZ__tmp_to_csv_from_csv2__r# �a�b�x�y�z)r r )r r" r$ )r r" r$ r r" r$ F)r r )� check_namesZAA�X�Y�Z)r z Writing 4 cols but got 2 aliases)�match)r' r( r r) �random�randnr+ r r- r � from_tuplesr �copyr �pytest�raises� ValueError) r r4 r �df�resultZmidxZ col_aliases�rs�xp�msgr r r �test_to_csv_from_csv2V s( " z(TestDataFrameToCSV.test_to_csv_from_csv2c C s� t �d���}ttj�dd��}ttj�dd��}|�|� |j|ddd� t�||g�}t |dd�}d d � |j D �|_ d d � |j D �|_ t � ||� W d Q R X d S ) NZ__tmp_to_csv_from_csv3__r# r" r9 F)�moder r )r c S s g | ] }t |��qS r )�int)�.0�labelr r r � <listcomp>� s z<TestDataFrameToCSV.test_to_csv_from_csv3.<locals>.<listcomp>c S s g | ] }t |��qS r )rQ )rR rS r r r rT � s ) r' r( r r) rC rD r+ �pd�concatr r r- )r r Zdf1�df2rM rL r r r �test_to_csv_from_csv3v s z(TestDataFrameToCSV.test_to_csv_from_csv3c s� t �d���}tjdd�� td� fdd�td�D �it� fdd�td�D �d d �d �}|�|� t|d d �}t� |j �|_ |j � d �|_ t� |d �|d< t j ||d d� W d Q R X d S )NZ__tmp_to_csv_from_csv4__r" )�secondsZdt_datac s g | ] }|� �qS r r )rR �i)�dtr r rT � s z<TestDataFrameToCSV.test_to_csv_from_csv4.<locals>.<listcomp>r# c s g | ] }|� �qS r r )rR rZ )r[ r r rT � s �dt_index)�name)r )r T)Zcheck_index_type) r' r( rU � Timedeltar r2 r r+ r Z to_timedeltar �renamer- )r r rJ rK r )r[ r �test_to_csv_from_csv4� s " z(TestDataFrameToCSV.test_to_csv_from_csv4c sf t �d��R}��|� t|ddgd�� � �fdd�}|d�� d< |d�� d< t �� �� W d Q R X d S ) NZ__tmp_to_csv_from_csv5__r r )r r c s$ t � | �j�d�j��| jj�S )N�UTC)r r[ � tz_convert�tz)�c)rK �timezone_framer r �<lambda>� s z:TestDataFrameToCSV.test_to_csv_from_csv5.<locals>.<lambda>r �C)r' r( r+ r r- )r re r � converterr )rK re r �test_to_csv_from_csv5� s z(TestDataFrameToCSV.test_to_csv_from_csv5c C sz d}t |d �}t�|d�}|j}|d |d g}t�� �"}|j|||d� t|dd�}W d Q R X tj|| |dd � d S ) Nr g @r# r$ r )r � chunksize)r F)r> )rQ r' �makeCustomDataframer r( r+ r r- )r rj �NrJ �cs�colsr �rs_cr r r �test_to_csv_cols_reordering� s z.TestDataFrameToCSV.test_to_csv_cols_reorderingc sV d� fdd� }d� t � d �}t�|d�}dddg|_||d � ddg}|||� d S ) Nc s� t �� ��}| j||� d� t|dd�}|d k r�| jjr>||_n | j�|�\}}| j�|�|_x^|D ]<}| | }|| }t|t �r�t � ||� qbt j ||dd� qbW n| j|_t j | |dd� W d Q R X d S )N)r rj r )r F)r> ) r' r( r+ r r Z is_uniqueZget_indexer_non_uniqueZtake� isinstancer Zassert_series_equalr- ) rJ rn r ro Zindexer�missingrd Zobj_dfZobj_rs)rj r r � _check_df� s z?TestDataFrameToCSV.test_to_csv_new_dupe_cols.<locals>._check_dfr g @r# r9 r: )N)rQ r' rk r )r rs rl rJ rn r )rj r �test_to_csv_new_dupe_cols� s z,TestDataFrameToCSV.test_to_csv_new_dupe_colsc C s~ d dd�}d}||d �}||d d�}t �d��B}t||d��}|j||d� | �|��t�}t j||d d � W d Q R X d S ) Nc S sr |d krt | d �}ttdd| d��}|rnx$tj�dt|�|�D ] }t||< q@W tj�d�}t|| < t||< |S )Ng�������?�2000Z5min)�freq�periodsr �d )rQ �listr r) rC �randintr. r )�nZnnat�srZ r r r �make_dtnat_arr� s z<TestDataFrameToCSV.test_to_csv_dtnat.<locals>.make_dtnat_arri� r r z1.csv)r9 r: )rj F)r> )N)r' r( r r+ r �applyr r- )r r} rj r% r& �pthrJ r6 r r r �test_to_csv_dtnat� s z$TestDataFrameToCSV.test_to_csv_dtnatc s� d � �fdd� }d}d� d}� | }xzdd|d ||d |d d| d d| d d| d| d d| d |d ||d gD ] }|t j||d d d �d d � q~W x�d D ]�\}}x�dD ]�}� | }x|dd|d ||d |d d| d d| d d| d| d d| d |d ||d gD ] }|t j||||d �||� �qW q�W q�W x�dD ]�}� | }xvd|d |d ||d |d d| d d| d d| d| d d| d |d ||d gD ]}|t �||�� �q�W �qLW x�d|d |d ||d |d gD ]~}t �|d�}t|j�} ddg| d d�< ddg| dd �< t|j�} ddg| d d�< ddg| dd �< | |_| |_||dd� �q�W |tt�d�d�� |t j� d d ddd�dd� x�dD ]�}t� | �}x�d|d |d ||d |d d| d d| d d| d| d d| d |d ||d gD ]R}|t j||dd�dd� |t j||dd�dd� |t j||ddd�ddd � �q"W �q�W d S ) NFc s� ddi}|rj|d k r$t t|��|d< t t|��|d< t�d��$}| j|d�d� �j|f|��W d Q R X n<d|d< t�d��$}| j|d�d� �j|f|��W d Q R X d d � � |r�| j�_|�r|�s�fd d �t|d �D �}t�t �j �g| �} | �_ �j d d �|d d �f �ddddddd�} |�rH|dk�r~d}t j � fdd ��j D �|d��_ t j � fdd �| j D �|d�| _ n�|dk�r�d}t j dd � �j D �|d��_ t j dd � | j D �|d�| _ n�|dk�rd}t �j �} t j dd � | D �|d��_ t j t tt| j �� ��|d�| _ n.| �|�}t j �j |d��_ t j | j |d�| _ |�rl|dk�r�d}t j � fdd ��jD �|d��_t j � fd d �| jD �|d�| _n�|dk�r�d}t j d d � �jD �|d��_t j d d � | jD �|d�| _n�|dk�r>d}t �j�} t j d d � | D �|d��_| j�� } t j d d � | D �|d�| _n.| �|�}t j �j|d��_t j | j|d�| _tj| �dd!� d S )"Nr Fr r Z__tmp_to_csv_moar__�utf8)�encodingrj r c S s t | t�s| �d�S | S )Nr� )rq �str�decode)r; r r r �_to_uni s zFTestDataFrameToCSV.test_to_csv_moar.<locals>._do_test.<locals>._to_unic s g | ]}� j d d �|f j�qS )N)�ilocr1 )rR rZ )r6 r r rT s zITestDataFrameToCSV.test_to_csv_moar.<locals>._do_test.<locals>.<listcomp>r" rZ �f�O)rZ r� r| �ur[ �pr� c s g | ] }� |��qS r r )rR rS )r� r r rT & s )�dtypec s g | ] }� |��qS r r )rR rS )r� r r rT ) s r[ c S s g | ] }t |��qS r )r )rR rS r r r rT . s c S s g | ] }t |��qS r )r )rR rS r r r rT 1 s r� c S s g | ] }t |��qS r )r )rR rS r r r rT 7 s c s g | ] }� |��qS r r )rR rS )r� r r rT D s c s g | ] }� |��qS r r )rR rS )r� r r rT G s c S s g | ] }t |��qS r )r )rR rS r r r rT L s c S s g | ] }t |��qS r )r )rR rS r r r rT O s c S s g | ] }t |��qS r )r )rR rS r r r rT U s c S s g | ] }t |��qS r )r )rR rS r r r rT Y s )r> )ry r2 r' r( r+ r r r � from_arraysr r� r) �arrayr �mapr Z to_timestamp�getr- ) rJ Zr_dtypeZc_dtype�rnlvl�cnlvl�dupe_colr r Z delta_lvl�ixZtype_mapZidx_listZcol_list)rj r )r� r6 r �_do_test� s� z5TestDataFrameToCSV.test_to_csv_moar.<locals>._do_testrx i� � r$ � r" r[ r| )� r_idx_type� c_idx_type))rZ rZ )r| r| )r� r[ )r� r� )r" r$ r# r� r# Zdupe���ZrdupeT)r� )r )� r_idx_nlevels)r� )r$ r# r� )� c_idx_nlevels)r� )r� r� )r� r� )NNNNF) r' rk ry r r r r) r3 rQ ) r r� rl Zncols�baseZnrowsr� r� rJ rn r� r )rj r r �test_to_csv_moar� s� c & z#TestDataFrameToCSV.test_to_csv_moarc C sz t j|d< dd� }|j�|�|d< t�� �F}|�|� | �|�}tj||dd� tjt � |�t � |�dd� W d Q R X d S )N�Gc S s t jt jgt j�� dk S )Ng �?)r) �infr* rC Zrand)r; r r r rf � � zETestDataFrameToCSV.test_to_csv_from_csv_w_some_infs.<locals>.<lambda>�HF)r> ) r) r* r r� r' r( r+ r r- �isinf)r r4 r� r r6 r r r � test_to_csv_from_csv_w_some_infs� s z3TestDataFrameToCSV.test_to_csv_from_csv_w_some_infsc C sn t j|d< t j |d< t�� �F}|�|� | �|�}tj||dd� tjt �|�t �|�dd� W d Q R X d S )N�E�FF)r> )r) r� r' r( r+ r r- r� )r r4 r r6 r r r � test_to_csv_from_csv_w_all_infs� s z2TestDataFrameToCSV.test_to_csv_from_csv_w_all_infsc C s� t �d��z}tdddgdddgd��}|j|d d � t|�}t �||� td d d gdd�|d< |j|d d � t|�}t �||� W d Q R X d S )NZ__tmp_to_csv_no_index__r" r$ r# r� r � )�c1�c2F)r � � � r )r� �c3)r' r( r r+ r r- r )r r rJ rK r r r �test_to_csv_no_index� s z'TestDataFrameToCSV.test_to_csv_no_indexc C sD t dddgdddgd��}d|d < |�� |jd d d gd �ks@t�d S ) Nr9 r: rd �aaZbb�cc)r r" �txt�testr r" )r )r r+ r/ )r rJ r r r � test_to_csv_with_mix_columns s z/TestDataFrameToCSV.test_to_csv_with_mix_columnsc C s� t ddgddggddgd�}t ddgddggdd gd�}t�d ��p}|j|dd gd � | �|�}t�||� |j|d dd gd � | �|�}|jdd�}|d ks�t�t�||� W d Q R X d S )Nr" r$ r# r� r r )r r? r@ Z__tmp_to_csv_headers__)r F)r r T)Zinplace)r r' r( r+ r r- Z reset_indexr/ )r Zfrom_dfZto_dfr r6 Z return_valuer r r �test_to_csv_headers s z&TestDataFrameToCSV.test_to_csv_headersc C sB |}|j }t�t|�d ��dd�}tj|ddgd�}||_ t�d���2}|j |dd� |j |d d gd � |� |� | j |d d gdd�}tj ||dd� |j j |j j ks�t �||_ |} | j }|t�t|��g}t�|�| _ | j |ddgd� | j |d d gd�} tj | | dd� | � |� | j |d d�} t| j�t| j�d k�sDt �| j |dd� | j |d d�} t�| j|j� ||_ W d Q R X t�d���F}d#dd�} tjddddd�}|� |� t |d d ddgd d gd�} t� || � tjddd dd�}|� |� t |d d ddgd d�} t� || � tjddddd�}|� |� t |d d ddgd d dgd�} t� || � | � }|j |dd� t |d d gd�} t� || � | d �}|j |dd� t |d d gd�} tj| jj � �s�t �|jj | j_ t� || � | � }|� |� t |d d gd gd�} t� || � | d �}|� |� t |d d gd gd�} t� || � | d �}|� |� xJd D ]B} d | � d �}tjt|d �� t |tt| ��d d� W d Q R X �qXW d!}tjt|d �� |j |dd"gd � W d Q R X W d Q R X t�d��V}| d d � � |� | � |�} | d d � }g |_ t�| j|j� t| �d k�s4t �W d Q R X d S )$Nr$ ����first�second)�namesZ__tmp_to_csv_multiindex__F)r r r )r r r" )r r )r> �time�foo)r! )r )r c S s< | dkrddg} t tjjdddd�tjdd d g| d �d d �S )NTr� r� r r� )r# r# )�size)�bahr� )r� �bar)Zban�baz)r� r )r r� )r r) rC rz r rE )r� r r r � _make_frameS s z>TestDataFrameToCSV.test_to_csv_multiindex.<locals>._make_framer r# r� )r� r� )r r T)r� r� zlen of z, but only 5 lines in file)rB z%cannot specify cols with a MultiIndexr� )N)r r) r3 r. Zreshaper r� r' r( r+ r r- r� r/ r r0 r1 rk �comZall_nonerG rH r ry r2 � TypeErrorZassert_index_equal)r r4 r5 �frameZ old_indexZarraysZ new_indexr rJ Ztsframer6 r� rK rZ rN �expr r r �test_to_csv_multiindex s� && z)TestDataFrameToCSV.test_to_csv_multiindexc C sv t td�td�d�t�dd�d�}t�d��@}|�|� | j|dd�}|� � }|j � t �|_ t� ||� W d Q R X d S )N�abcr# )r r r )r z!__tmp_to_csv_interval_index__.csv)r )r ry r2 rU Zinterval_ranger' r( r+ r rF r �astyper� r- )r rJ r rK �expectedr r r �test_to_csv_interval_index� s " z-TestDataFrameToCSV.test_to_csv_interval_indexc C s� t tj�dd��tj��}tj|d< t�d��J}|j |dd� t |��(}|� � }|d � d�d dksjt �W d Q R X W d Q R X d S ) Nr" r� z!__tmp_to_csv_float32_nanrep__.csvi� )�na_rep�,r$ Z999)r r) rC rD r� r r* r' r( r+ �open� readlines�splitr/ )r rJ r r� �linesr r r �test_to_csv_float32_nanrep� s z-TestDataFrameToCSV.test_to_csv_float32_nanrepc C sT t dddgdddgd��}t�d��&}|�|� | �|�}t�||� W d Q R X d S ) Nr" r$ r# z5,6z7,8z9,0)r r z __tmp_to_csv_withcommas__.csv)r r' r( r+ r r- )r rJ r rW r r r �test_to_csv_withcommas� s z)TestDataFrameToCSV.test_to_csv_withcommasc C sF dd� }t tj�dd�d|d�d�}t tj�dd��d�d|d �d�}t d |j|d �d �}t d |j|d�d �}t td�|j|d�d �}tj|jdd�dd�f< t j |||||gdd�}i }xHdtj fd tj fd tj fdtfgD ]"\} } x|| �D ] } | || < q�W q�W t�� �0} |�| � t| d||d�d�} t�| |� W d Q R X d S )Nc s � fdd�t d�D �S )Nc s g | ]}� � |d ���qS )Z03dr )rR rZ )r] r r rT � s zMTestDataFrameToCSV.test_to_csv_mixed.<locals>.create_cols.<locals>.<listcomp>r )r2 )r] r )r] r � create_cols� s z9TestDataFrameToCSV.test_to_csv_mixed.<locals>.create_colsrx r r �float)r� r r rQ T�bool)r r r� �object�20010101�date� �2 r" r# )�axisr )r r� r )r r) rC rD r� r r r* r� rU rV r r Zbool_r� r' r( r+ r r- )r r� �df_float�df_int�df_bool� df_object�df_dtrJ Zdtypesr{ r� rd �filenamerL r r r �test_to_csv_mixed� s6 z$TestDataFrameToCSV.test_to_csv_mixedc C s� t tj�dd�ttd��ttd�� dd�}t�� �0}|�|� t |dd�}|j |_ t� ||� W d Q R X t tj�dd�dd �}t tj�dd��� d �}t d |j td�d �}t d |j td�d �}t td�|j td�d �}tj|||||gdd d�}dddgd |_ t�� �P}|�|� t |dd�}x dD ]} t|| �|| < �q(W |j |_ t� ||� W d Q R X d} t�| d�}dddg|_ t�� �8}|�|� t |dd�}|jddid�}t� ||� W d Q R X d S )Ni� r� � r )r r� r )r r# )r� r T)r r r� r� r" )r� Z ignore_indexr$ r )z0.4z1.4z2.4r� r9 r: za.1)r )r r) rC rD ry r2 r' r( r+ r r r- r� r r rU rV r rk r_ ) r rJ r� rK r� r� r� r� r� rZ rl r r r �test_to_csv_dups_cols� s@ z(TestDataFrameToCSV.test_to_csv_dups_colsc C s� t dtd�i�}|jd |d< |jd |d< |jd |d< xFd D ]>}t�� �,}|j||d � t|d d �}t�||� W d Q R X q@W d S ) Nr i�� g �?r g @rg g @�D)i' iP� i�� )rj r )r )r r2 r r' r( r+ r r- )r r� rj r� rL r r r �test_to_csv_chunking% s z'TestDataFrameToCSV.test_to_csv_chunkingc C sX t tj�dd�d d d�}t�� �.}|j|ddd� t|d d�}t�||� W d Q R X d S )Nr" i�� )r r F)r r )r ) r r) rC rD r' r( r+ r r- )r rJ r� rL r r r �!test_to_csv_wide_frame_formatting2 s z4TestDataFrameToCSV.test_to_csv_wide_frame_formattingc C sj t d�}| j|d d�}td||jd i�}t�� �,}|�|� t|dd�}tj||dd� W d Q R X d S )Nz a,1.0 b,2.0)r �tr )r F)r> )r r r r r' r( r+ r- )r �f1rJ Znewdfr r6 r r r �test_to_csv_bug; s z"TestDataFrameToCSV.test_to_csv_bugc C s| t ddddgi�}t�� �X}|j|dd� t|ddd�}t�||� |j|dd d � t|d dd�}t�||� W d Q R X d S ) Nu c/σr" r$ r# zUTF-8)r� r )r r� F)r� r )r r' r( r+ r r- )r rJ r rW r r r �test_to_csv_unicodeG s z&TestDataFrameToCSV.test_to_csv_unicodec C sl t d�}tddddgdddd ggdd d d gdd gd �}|j|dd� |�d� t|ddd�}t�||� d S )N� u אZd2Zd3Zd4Za1Za2�a3Za4u בu גu ד)r r zUTF-8)r� r )r r� )r r r+ �seekr r' r- )r �bufrJ rW r r r � test_to_csv_unicode_index_colT s z0TestDataFrameToCSV.test_to_csv_unicode_index_colc C s: t � }|�|� |�d� t|dd�}tj||dd� d S )Nr )r F)r> )r r+ r� r r' r- )r r4 r� r6 r r r �test_to_csv_stringiob s z'TestDataFrameToCSV.test_to_csv_stringioc C s� t dddgdddggddgd d d gd �}t�� �T}|j|d d� t|dd�}t dddgdddggddgd d d gd �}t�||� W d Q R X d S )Ng��~�Ϛ�?g9DܜJ�?g >�D�)�?g�.4�i�(@g33333�@g���̤�Ar r r? r@ rA )r r z%.2f)Z float_formatr )r g� ��Q��?gq= ףp�?g= ףp=�?g�p= ף(@)r r' r( r+ r r- )r rJ r� rL rM r r r �test_to_csv_float_formatj s z+TestDataFrameToCSV.test_to_csv_float_formatc C sb t dddgdddgd��}t� }|j|dtjd d � |�� }d d d dg}t�|�}||ks^t�d S )Nr" r$ r# r� r� r� )r r Fzutf-8)r �quotingr� z"A","B"z1,"foo"z2,"bar"z3,"baz") r r r+ �csv�QUOTE_NONNUMERIC�getvaluer' � convert_rows_list_to_csv_strr/ )r rJ r� rK � expected_rowsr� r r r �!test_to_csv_unicodewriter_quoting~ s z4TestDataFrameToCSV.test_to_csv_unicodewriter_quotingc C sd t dddgi�}xNdD ]F}t� }|j|tj|dd� |�� }dddg}t�|�}||kst�qW d S )Nr Zhelloz {"hello"})Nzutf-8F)r� r� r ) r r r+ r� � QUOTE_NONEr� r' r� r/ )r rJ r� r� rK r� r� r r r �test_to_csv_quote_none� s z)TestDataFrameToCSV.test_to_csv_quote_nonec C sb t dddgdddgd�dd d gd �}t� }|j|d d � ddddg}t�|�}|�� |ks^t�d S )Nr" r$ r# r� r r� )r r �one�two�three)r F)r! zA,Bzone,1,4ztwo,2,5z three,3,6)r r r+ r' r� r� r/ )r rJ r� r� r� r r r �"test_to_csv_index_no_leading_comma� s $ z5TestDataFrameToCSV.test_to_csv_index_no_leading_commac C s8 t dddgdddgd�dd d gd �}t�� �@}|j|d d � d}t|dd��}|�� |ks^t�W d Q R X W d Q R X t�� �@}|j|dd � d}t|dd��}|�� |ks�t�W d Q R X W d Q R X t�� �f}|�|� tj� d�}d| d | d | d | }t|dd��}|�� |k�s t�W d Q R X W d Q R X d S )Nr" r$ r# r� r r� )r r r� r� r� )r z )�line_terminators# ,A,B one,1,4 two,2,5 three,3,6 �rb)rP � s ,A,B one,1,4 two,2,5 three,3,6 zutf-8s ,A,Bs one,1,4s two,2,5s three,3,6) r r' r( r+ r� �readr/ �os�linesep�encode)r rJ r r� r� Z os_linesepr r r � test_to_csv_line_terminators� s" $ $ $ z/TestDataFrameToCSV.test_to_csv_line_terminatorsc C s� t t�ddddddddg��}t ddddddddg�}t� }|j|dd� t� }|j|dd� |�� |�� ksrt�td|i�}td|i�}t� }|�|� t� }|�|� |�� |�� ks�t�d S )Nr9 r: rd F)r r| )r rU Z Categoricalr r+ r� r/ r )r r| r& �resr� rJ rW r r r � test_to_csv_from_csv_categorical� s z3TestDataFrameToCSV.test_to_csv_from_csv_categoricalc C s: |j d d�}t|t�st�tt|�dd�}t�||� d S )N)Z path_or_bufr )r )r+ rq r� r/ r r r' r- )r r4 Zcsv_strr6 r r r �test_to_csv_path_is_none� s z+TestDataFrameToCSV.test_to_csv_path_is_nonez df,encodingg��~�Ϛ�?g9DܜJ�?g >�D�)�?g�.4�i�(@g33333�@g���̤�Ar r r? r@ rA )r r Nr� �defZghi)r �asciir �{ u 你好u 世界�gb2312u Γειά σουu ΚόσμεZcp737c C s t �� ��}|j|||d� t||d|d�}t �||� t|d||d��"}|j|j|d� |jjrdt�W d Q R X t|||ddd�}t �||� t � ||��4}|� � � |p�d�}x|j D ]} | |ks�t�q�W W d Q R X t � ||�� }t �|t|d|d �� W d Q R X W d Q R X d S ) N)� compressionr� r )r r r� �w)r� T)r r� r Zsqueezer� )r r� ) r' r( r+ r r- r �handle�closedr/ �decompress_filer r� r ) r rJ r� r r� rK Zhandles�fh�text�colr r r �test_to_csv_compression� s. z*TestDataFrameToCSV.test_to_csv_compressionc C sf t �d���P}|j}t||�d�d�|d�}|j|dd� t|dd�}|�d d � �}|j�d d � �|_t � ||� |j|d d� t|dd�}|�d d � �}|j�dd � �|_t � ||� |j }|j|dd� t|dd�}|�dd � �}|j �dd � �|_ t � ||� t dgd dddg �}td|i|d�} | j|d d� t|ddgdd�}t � || � W d Q R X d S )NZ__tmp_to_csv_date_format__r" )r r )r z%Y%m%d)� date_formatr )r c S s t | �d��S )Nz%Y%m%d)rQ �strftime)r; r r r rf + r� z<TestDataFrameToCSV.test_to_csv_date_format.<locals>.<lambda>c S s t | �d��S )Nz%Y%m%d)rQ r )r; r r r rf . r� z%Y-%m-%dc S s | � d�S )Nz%Y-%m-%d)r )r; r r r rf 8 r� c S s | � d�S )Nz%Y-%m-%d)r )r; r r r rf ; r� c S s t | �d��S )Nz%Y%m%d)rQ r )r; r r r rf G r� c S s | � d�S )Nz%Y%m%d)r )r; r r r rf K r� r r� z 2000-01-01z1/1/2000z1-1-2000r )r r ) r' r( r r �shiftr+ r Zapplymapr� r- �Tr r ) r r5 r r\ r� Zdatetime_frame_intZdatetime_frame_strZdatetime_frame_columnsZ nat_indexZ nat_framer r r �test_to_csv_date_format s@ z*TestDataFrameToCSV.test_to_csv_date_formatc C s� t �d���}tdddddd�}x�||t�d� gD ]p}|�d �}tjtt |��d d �}t d |i|d �}|j |d d � t |dd�}t |jd d��d�|_t �||� q2W W d Q R X tddddd�}|�d �}d |j_t d|d�|d �}t �d��\}|j |d d � t |dd�}t |jd d��d�|_t |d d d��d�|d< t �||� W d Q R X |�t� t �d��&}|�|� t�|�}t �||� W d Q R X d S )NZcsv_date_format_with_dstz2013-10-26 23:00z2013-10-27 01:00z Europe/Londonr� Zinfer)rc rv Z ambiguousZ10sr )r� r )r Tr )r )�utcz 2015-01-01z 2015-12-31z Europe/Paris)rv rc r" )r1 �idxr z datetime64[ns, Europe/Paris])r' r( r rU r^ r, r) r� r2 r. r r+ r r r rb r- �_dataZ_freqr� r� Z to_pickleZ read_pickle)r r �timesrZ Z time_rangerJ rK r r r r � test_to_csv_with_dst_transitions[ sB z3TestDataFrameToCSV.test_to_csv_with_dst_transitionsc C sL t ddgddgdtjgddgd��}d d d g}t�|�}|�� }||ksLt�|jd d �}||ksdt�d d d g}t�|�}|jtjd �}||ks�t�d ddg}t�|�}|jtj d �}||ks�t�d ddg}t�|�}|jtj d �}||ks�t�d}t j tj |d�� |jtjd � W d Q R X t j tj |d�� |jtjd d� W d Q R X d d dg}t�|�}|jtjdd�}||k�szt�dd dg}t�|�}|jtjdd�}||k�s�t�ddg}t�|�}tt|��}t� }|j|d dd � |�� |k�s�t�t d d gd d!gd"d#gd$��}|�dd%g�}d&d'd(g}t�|�}|jtj d �|k�sHt�d S ))NTFg �?g������ @�* r9 zb,c)Zc_boolZc_float�c_intZc_stringz ,c_bool,c_float,c_int,c_stringz0,True,1.0,42.0,az1,False,3.2,,"b,c")r� z("","c_bool","c_float","c_int","c_string"z"0","True","1.0","42.0","a"z"1","False","3.2","","b,c"z0,True,1.0,42.0,"a"z1,False,3.2,"","b,c"z%need to escape, but no escapechar set)rB )r� � escapecharz1,False,3.2,,b!,c�!z ,c_bool,c_ffloat,c_int,c_stringz1,False,3.2,,bf,cr� za,b,cz 1,"test ",3zutf-8)r� r r" r$ r# r� r r� )r9 r: rd r: z "a","b","c"z "1","3","5"z "2","4","6")r r) r* r' r� r+ r/ r� � QUOTE_MINIMAL� QUOTE_ALLr� rG rH �Errorr� r r r� Z set_index) r rJ r� r� rK rN Z text_rowsr r� r r r �test_to_csv_quoting� sv z&TestDataFrameToCSV.test_to_csv_quotingc C s� dddg}t j|dd�}tdddg|d �}|�� }d d d d g}t�|�}||ksTt�d}|j|d�}d dddg}t�|�}||ks�t�dtdg}t j|dd�}tdddg|d �}|�� }d d dd g}t�|�}||ks�t�d S )Nz 1990-01-01z 2000-01-01z 3005-01-01r� )rv r� r r� )r z,0z 1990-01-01,4z 2000-01-01,5z 3005-01-01,6z%m-%d-%Y)r z 01-01-1990,4z 01-01-2000,5z 01-01-3005,6z,5)rU Z PeriodIndexr r+ r' r� r/ r )r �datesr rJ rK r� r� r r r r � test_period_index_date_overflow� s& z2TestDataFrameToCSV.test_period_index_date_overflowc C sp t �ddddg�}tddddgd d d d gg�}||_d dddg}|j|d�}dddg}t�|�}||kslt�d S )N)r9 r" )r9 r$ )r: r" )r: r$ r" r$ r# r� r r� r� r� r9 r: rd �d)r z,a,b,c,dz 0,1,2,3,4z 1,5,6,7,8)r rE r r r+ r' r� r/ )r r rJ r rK r� r� r r r �test_multi_index_header s z*TestDataFrameToCSV.test_multi_index_headerc C sR t dddg�}tdddgg|d�}|jddgd�}d}|jd d �}t�||� d S ) N)r" )r$ )r# r" r$ r# )r z ,1,3 0,1,3 r )r� )r r Zreindexr+ r' r0 )r r rJ r� rK r r r �$test_to_csv_single_level_multi_index s z7TestDataFrameToCSV.test_to_csv_single_level_multi_indexc C s� t dddgi�}dddg}t�|�}t�d��<}|j|dd� tj|d d ��}|�� �d �}W d Q R X W d Q R X ||ks|t�d S ) Nr9 r" r$ �1�2z__test_gz_lineend.csv.gzF)r �gzip)r zutf-8) r r' r� r( r+ r r r� r/ )r rJ r� r� r r� rK r r r �test_gz_lineend# s "z"TestDataFrameToCSV.test_gz_lineendc C s< t dtddd�i�}t� }|�|� |�� }d|ks8t�d S )Nr9 z1/1/2000r� )rw z 2000-01-01)r r r r+ r� r/ )r r� r� rK r r r �test_to_csv_numpy_16_bug/ s z+TestDataFrameToCSV.test_to_csv_numpy_16_bugc C s6 t d d g�jd dddd��dd�}d}||ks2t�d S )NFr� )r r r� z r z"" "" )r r+ �replacer/ )r rK r� r r r �test_to_csv_na_quoting8 s z)TestDataFrameToCSV.test_to_csv_na_quoting)4�__name__� __module__� __qualname__r r8 rO rX r` ri rp rt rG �markZslowr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r Z parametrizer r r r r' r) r+ r, r0 r1 r3 r r r r r ( sp ' ) \ .2 & (<1b r ) r� �ior r �numpyr) rG Z pandas.errorsr ZpandasrU r r r r r r r r r Zpandas._testingZ_testingr' Zpandas.core.common�core�commonr� Zpandas.io.commonr ZMIXED_FLOAT_DTYPESZMIXED_INT_DTYPESr r r r r �<module> s( ,
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка