Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/excel/__pycache__/test_writers.cpython-37.pyc
Назад
B �5�g�� � @ sP d dl mZm Z mZ d dlmZ d dlmZ d dlZd dlZd dl Z d dl Z d dl m mZ d dlZd dlmZmZmZmZmZ d dlmZ d dlmZm Z m Z m Z m Z m Z e j!dd� �Z"e j!d d � �Z#e j$�%d e j&d e�'d �e�'d�gd�e j&de�'d �e�'d�gd�e j&de�'d�e�'d�gd�e j&d e�'d�e�'d�gd�e j&de�'d�d�g�G dd� d��Z(e j$�%de j&d d e�'d �e�'d�gd�e j&d de�'d �e�'d�gd�e j&dde�'d�e�'d�gd�e j&dd e�'d�e�'d�gd�e j&dde�'d�d�g�e j$�)d �G dd� d���Z*G dd � d �Z+e�'d�e�'d �G d d � d ���Z,dS ) � )�date�datetime� timedelta)�partial)�BytesION)� DataFrame�Index� MultiIndex� get_option� set_option)� ExcelFile� ExcelWriter�_OpenpyxlWriter� _XlsxWriter� _XlwtWriter�register_writerc c s t �| �� }|V W dQ R X dS )z9 Fixture to open file for use in each test case. N)�tm� ensure_clean)�ext� file_path� r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/excel/test_writers.py�path$ s r c c s8 d|� d�� d�}t|�}t|| � dV t||� dS )aM Fixture to set engine for use in each test case. Rather than requiring `engine=...` to be provided explicitly as an argument in each test, this fixture sets a global option to dictate which engine should be used to write Excel files. After executing the test it rolls back said change to the global option. z io.excel.�.z.writerN)�stripr r )�enginer Z option_nameZ prev_enginer r r � set_engine- s r r z.xlsx�openpyxlZxlrd)�marksz.xlsmz.xls�xlwt� xlsxwriterz.ods�odfc @ s e Zd Zej�ddeejgd �fdedejgd i�fg�dd� �Z ej�ddedgejgd �fdeejgd �fg�d d � �Z d d � Z d d� Z dd� Z ej�dddg�ej�dddg�ej�dddg�ej�dddg�dd� ����Zdd� Zd d � ZdS ) � TestRoundTripzheader,expectedN� r z Unnamed: 0� c C sx d}t dddgdddgdddgdd d gg�}t�|��,}|j||d d d � tj||d g|d�}W d Q R X t�||� d S )NZ no_header� � �d � �� r$ i, r# i� F)�index�headerr )� sheet_name�usecolsr+ )r r r �to_excel�pd� read_excel�assert_frame_equal)�selfr r+ �expected�filename�dfr �resultr r r �!test_read_one_empty_col_no_headerK s ( z/TestRoundTrip.test_read_one_empty_col_no_headerc C sx d}t dddgdddgdddgdd d gg�}t�|��,}|j|dd d d � tj||dg|d�}W d Q R X t�||� d S )NZ with_headerr% r& r' r( r) r$ i, r# i� FT)r* r+ r )r, r- r+ )r r r r. r/ r0 r1 )r2 r r+ r3 r4 r5 r r6 r r r �#test_read_one_empty_col_with_header\ s ( z1TestRoundTrip.test_read_one_empty_col_with_headerc C s� t ddgddgddggddgd �}t�|���}t|��(}|j|d d d d � |j|d d d� W d Q R X ddg|_t|��2}tj|d d ddgd�}tj|d d ddgd�}W d Q R X t� ||� t� ||� W d Q R X d S )Nr& �foor( �barr$ Zbaz�a�b)�columnsZ Data_no_headF)r+ r* ZData_with_head)r* �A�B)r, r+ �names)r, � index_colr@ ) r r r r r. r= r r/ r0 r1 )r2 r Zrefdf�pth�writer�readerZ xlsdf_no_headZxlsdf_with_headr r r �"test_set_column_names_in_parameterl s " z0TestRoundTrip.test_set_column_names_in_parameterc s� dd� � dddg}� fdd�|D �}t t||��}t�|��n}t|��(}x |�� D ]\}}|�||� qRW W d Q R X tj||dd �}x |D ]} t� || || � q�W W d Q R X d S ) Nc S s& dddgdddg }}t ||| gd�S )N� � �! r& r( r$ )r= )r )Zcol_sheet_name�d�ir r r �tdf� s zDTestRoundTrip.test_creating_and_reading_multiple_sheets.<locals>.tdfZAAAZBBBZCCCc s g | ] }� |��qS r r )�.0�s)rK r r � <listcomp>� s zKTestRoundTrip.test_creating_and_reading_multiple_sheets.<locals>.<listcomp>r )r, rA ) �dict�zipr r r �itemsr. r/ r0 r1 ) r2 r Zsheets�dfsrB �ewZ sheetnamer5 Z dfs_returnedrM r )rK r �)test_creating_and_reading_multiple_sheets� s z7TestRoundTrip.test_creating_and_reading_multiple_sheetsc C s� t �|���}tddiddiddiddid��}tddiddiddiddid��}|�|� tj|ddgdd�}t �||� tddiddiddiddid��}tddiddiddiddid ��}|�|� tj|ddgdd�}t �||� W d Q R X d S ) Nr r& r$ � ))�One�x)�Two�X)rX �Y)�Zeror% ))rV rW )rX rY )rX rZ )r[ zUnnamed: 4_level_1)r+ rA ))�Begr% )�MiddlerW )�TailrY )r^ rZ ))r\ zUnnamed: 1_level_1)r] rW )r^ rY )r^ rZ )r r r r. r/ r0 r1 )r2 r r r5 r3 �actualr r r �&test_read_excel_multiindex_empty_level� s6 z4TestRoundTrip.test_read_excel_multiindex_empty_level� c_idx_namesTF� r_idx_names� c_idx_levelsr& � r_idx_levelsc C s( t �|���}|dkr$|r$t�d� |p.|dk}t �dd||||�}|�|� tj|tt |��tt |��d�} t j || |d� t j |j dd d �f< |�|� tj|tt |��tt |��d�} t j || |d� t j |j dd d �f< |�|� tj|tt |��tt |��d�} t j || |d� W d Q R X d S )Nr& z?Column index name cannot be serialized unless it's a MultiIndex� )rA r+ )� check_namesr ���)r r �pytest�skip�makeCustomDataframer. r/ r0 �list�ranger1 �np�nan�iloc) r2 r ra rb rc rd rB rf r5 Zactr r r � test_excel_multindex_roundtrip� s6 z,TestRoundTrip.test_excel_multindex_roundtripc C s� t dddgtjddd�d��}|�� }|d j�d�|d< t�|��p}|�|� tj |d d �}t� ||� tj |dgd d �}t� ||� d d � }tj |dg|d d�}t� ||� W d Q R X d S )Nr& r( r$ z 2012-01-01)�periods)�col� date_stringsrs z%m/%d/%Yr )rA )� parse_datesrA c S s t �| d�S )Nz%m/%d/%Y)r �strptime)rW r r r �<lambda> � z;TestRoundTrip.test_read_excel_parse_dates.<locals>.<lambda>)rt � date_parserrA ) r r/ Z date_range�copy�dt�strftimer r r. r0 r1 )r2 r r5 Zdf2rB �resrx r r r �test_read_excel_parse_dates� s z)TestRoundTrip.test_read_excel_parse_datesc C s� t �td�tjt�d�ddd�g�}ttd�|d�}t�|��"}|� |� tj |ddgd�}W d Q R X ttd�t �td�d d d d gg��}t� ||� d S ) Nr# z 2020-01-01Z6M)�startrq �freq)r* r r& )rA z(2020-01-31, 2020-07-31]z(2020-07-31, 2021-01-31]z(2021-01-31, 2021-07-31]z(2021-07-31, 2022-01-31]) r � from_arraysrl r/ Zinterval_range� Timestampr r r r. r0 r1 )r2 r Zmidxr5 rB r6 r3 r r r �"test_multiindex_interval_datetimes s" z0TestRoundTrip.test_multiindex_interval_datetimes)�__name__� __module__� __qualname__rh �mark� parametrizer rm rn r7 r8 rE rT r` rp r} r� r r r r r" > s 68/ .r" z engine,extc @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z e j � de je je je jg�dd� �Ze j � de je je jg�dd� �Ze j � de je jg�dd� �Zdd� Zdd� Zdd� Z d d � Z d d � Z d d!� Z d"d#� Z d$d%� Z!d&d'� Z"d(d)� Z#d*d+� Z$d,d-� Z%d.d/� Z&d0d1� Z'd2d3� Z(d4d5� Z)d6d7� Z*d8d9� Z+d:d;� Z,e j � d<d=d>g�e j � d?d@dAdBg�e j � dCd@dAdBg�dDdE� ���Z-dFdG� Z.dHdI� Z/dJdK� Z0e j � dLdMdNg�dOdP� �Z1dQdR� Z2dSdT� Z3dUdV� Z4dWdX� Z5dYdZ� Z6d[d\� Z7d]d^� Z8d_d`� Z9dadb� Z:dcdd� Z;dedf� Z<dgdh� Z=didj� Z>e j � dkdle?g�dmdn� �Z@dodp� ZAdqdr� ZBdlS )s�TestExcelWriterc C s� d}d}t j|dfd�}t jd|fd�}t|�}t|�}d}tjt|d�� |�|� W d Q R X tjt|d�� |�|� W d Q R X d S )Ni i@ r& )�shapezsheet is too large)�match)rm Zzerosr rh �raises� ValueErrorr. ) r2 r Zbreaking_row_countZbreaking_col_countZrow_arrZcol_arrZrow_dfZcol_df�msgr r r �test_excel_sheet_sizeN s z%TestExcelWriter.test_excel_sheet_sizec C sz t tj�dd��}|�|� t|��}tj|ddd�}W d Q R X t� ||� d}t j t |d�� t�|d� W d Q R X d S )N� r( r )r, rA z Worksheet named '0' not found)r� �0) r rm �random�randnr. r r/ r0 r r1 rh r� r� )r2 r r �gt�xlr5 r� r r r � test_excel_sheet_by_name_raise` s z.TestExcelWriter.test_excel_sheet_by_name_raisec C s� t |��8}|�|d� |�� }|jd d d� |_|�|d� W d Q R X t|��>}tj|ddd�}tj|ddd�}t�||� t�||� W d Q R X d S )NZData1rg ZData2r )r, rA ) r r. ry r= r r/ r0 r r1 )r2 �framer rC �frame2rD Zfound_dfZ found_df2r r r �!test_excel_writer_context_managerm s z1TestExcelWriter.test_excel_writer_context_managerc C s� |� � }tj|d d d�< |�|d� |j|dddgd� |j|ddd� |j|ddd� |�|d� tj|dd d �}t�||� |j|ddd� tj|dd d �}|j|_t�||� |j|dd d � tj|dd d gd �}t�||� |j|ddd � tj|dd dgd �}t�||� |j|ddd � tj|dd ddgd �}t�||� |�|d� tj|d d�}t�||� |�|d� tj|d d�}t�||� |d }|�|� tj|d d�}t�|� � |� d S )Nr> re �test1r? )r= F)r+ )r* r )r, rA ZNA)Zna_rep)r, rA Z na_valuesZ88�X g V@�Sheet1)rA r� ) ry rm rn r. r/ r0 r r1 r* Zto_frame)r2 r� r �reconsrM r r r �test_roundtrip{ sB z TestExcelWriter.test_roundtripc C sP |� � }d|d< |�|d� t|��}tj|ddd�}W d Q R X t�||� d S )Nr: r9 r� r )r, rA )ry r. r r/ r0 r r1 )r2 r� r Z mixed_framerD r� r r r � test_mixed� s zTestExcelWriter.test_mixedc C s` |}t jt�|j�d d�}||_|�|d� t|��}t j|ddd�}W d Q R X t� ||� d S )N)r r� r )r, rA ) r/ � DatetimeIndexrm �asarrayr* r. r r0 r r1 )r2 �tsframer r5 r* rD r� r r r � test_ts_frame� s z TestExcelWriter.test_ts_framec C s^ |� � }tj|d d d�< |�|d� |j|dddgd� |j|ddd� |j|ddd� d S ) Nr> re r� r? )r= F)r+ )r* )ry rm rn r. )r2 r� r r r r �test_basics_with_nan� s z$TestExcelWriter.test_basics_with_nan�np_typec C s� t tjjdddd�|d�}|�|d� t|��}tj|ddd�}W d Q R X |�tj �}t � ||� tj|ddd�}t � ||� |�t �}|j �t �|_ |j�t �|_t jtd d �� tj|dd dd �}W d Q R X t � ||� d S ) Ni����r� )r� r( )�size)�dtyper� r )r, rA zconvert_float is deprecated)r� F)r, � convert_floatrA )r rm r� �randintr. r r/ r0 �astype�int64r r1 �floatr= r* �assert_produces_warning� FutureWarning) r2 r� r r5 rD r� Z int_frameZrecons2Z float_framer r r �test_int_types� s z TestExcelWriter.test_int_typesc C sZ t tj�d�|d�}|�|d� t|�� }tj|ddd��|�}W d Q R X t � ||� d S )Nr� )r� r� r )r, rA ) r rm r� Z random_sampler. r r/ r0 r� r r1 )r2 r� r r5 rD r� r r r �test_float_types� s z TestExcelWriter.test_float_typesc C sZ t ddddg|d�}|�|d� t|�� }tj|ddd��|�}W d Q R X t�||� d S )Nr& r TF)r� r� )r, rA )r r. r r/ r0 r� r r1 )r2 r� r r5 rD r� r r r �test_bool_types� s z TestExcelWriter.test_bool_typesc C s\ t dtjfddtj fg�}|�|d� t|��}tj|ddd�}W d Q R X t�||� d S )Nr& )r( r$ re r� r )r, rA ) r rm �infr. r r/ r0 r r1 )r2 r r5 rD r� r r r �test_inf_roundtrip� s z"TestExcelWriter.test_inf_roundtripc C s. t jt�|j�d d�}||_|�� }tj|d d d�< |�|d� |j|dddgd� |j|ddd� |j|ddd � t|�� }|�|d� |�|d � W d Q R X t |��>}t j |dd d �}t � ||� t j |d d d �}t � ||� W d Q R X d t |j�k�st�d|jd k�st�d |jd k�s*t�d S )N)r r> re r� r? )r= F)r+ )r* Ztest2r )r, rA r( r& )r/ r� rm r� r* ry rn r. r r r0 r r1 �len� sheet_names�AssertionError)r2 r� r� r r* rC rD r� r r r � test_sheets s& zTestExcelWriter.test_sheetsc C s� |� � }tj|d d d�< |�|d� |j|dddgd� |j|ddd� |j|ddd� td d d d g�}|j|d|d� t|��}tj|dd d�}W d Q R X |� � }||_t � ||� d S )Nr> re r� r? )r= F)r+ )r* ZAArY rZ �Zr )r, rA ) ry rm rn r. r r r/ r0 r= r r1 )r2 r� r Z col_aliasesrD �rs�xpr r r �test_colaliases# s z TestExcelWriter.test_colaliasesc C s" |� � }tj|d d d�< |�|d� |j|dddgd� |j|ddd� |j|ddd� ttj�d d ��d k}|j|dd g|d � t|�� }tj |dd d�� tj �}W d Q R X d g|j _ |j j |j j ks�t�ttj�d d ��d k}|j|dd ddg|d � t|�� }tj |dd d�� tj �}W d Q R X d g|j _ |j j |j j k�sHt�ttj�d d ��d k}|j|dd |d � t|�� }tj |dd d�� tj �}W d Q R X d g|j _ t�||� t�� |j|dddddgd|d� |� � }|�ddg�}t|��}tj |dd dgd�}W d Q R X t�||� d S )Nr> re r� r? )r= F)r+ )r* r� r( r �test)Z index_label� merge_cells)r, rA �dummyZdummy2�C�D)r= r* r� r& )ry rm rn r. r r� r� r r/ r0 r� r� r* r@ r� r r1 �bool� set_index)r2 r� r� r r5 rD r� r r r �test_roundtrip_indexlabels5 sT z*TestExcelWriter.test_roundtrip_indexlabelsc C sr t tj�dd��}d|j_|j||d� t|�� }tj ||j d dd�}W d Q R X t � ||� |jjdksnt �d S )Nr� r# r9 )r� r )r, rA )r rm r� r� r* �namer. r r/ r0 r� r r1 r� )r2 r� r r5 �xfr6 r r r � test_excel_roundtrip_indexnamen s z.TestExcelWriter.test_excel_roundtrip_indexnamec C sz t jt�|j�d d�}||_|�� }dd� |jD �|_|j|d|d� t|��}t j|ddd�}W d Q R X t � ||� d S )N)r c S s g | ] }|� � �qS r )r )rL rW r r r rN � s zATestExcelWriter.test_excel_roundtrip_datetime.<locals>.<listcomp>r� )r� r )r, rA ) r/ r� rm r� r* ry r. r r0 r r1 )r2 r� r� r r* ZtsfrD r� r r r � test_excel_roundtrip_datetimez s z-TestExcelWriter.test_excel_roundtrip_datetimec C s\ t tddd�tddd�gtddd d d d �tdd dddd�ggddgddgd�}t tddd�tddd�gtddd d d d �tdd dddd�ggddgddgd�}t�|���}t|��}|�|d� W d Q R X t|ddd��}|�|d� W d Q R X t|��} tj | ddd�} W d Q R X t|��} tj | ddd�} W d Q R X W d Q R X t� | | � t� | |� d S )Ni� r& � i� � � i� re � � rH r# r( � � ZDATEZDATETIMErY rZ )r* r= r� z DD.MM.YYYYzDD.MM.YYYY HH-MM-SS)Z date_formatZdatetime_formatr )r, rA ) r r r r r r r. r r/ r0 r1 ) r2 r r r r5 Z df_expected� filename2Zwriter1Zwriter2Zreader1Zrs1Zreader2Zrs2r r r � test_excel_date_datetime_format� s2 $ $ $ z/TestExcelWriter.test_excel_date_datetime_formatc C s� t tjjdddd�tjd�}|�� }t�|d d�|d< t�|d d��t �|d< |� |d� t |��}tj |ddd �}W d Q R X t �||� d S ) Ni����r� )� r& )r� )r� r �newr� )r, rA )r rm r� r� r� ry r/ �cutr� �strr. r r0 r r1 )r2 r r5 r3 rD r� r r r � test_to_excel_interval_no_labels� s z0TestExcelWriter.test_to_excel_interval_no_labelsc C s� t tjjdddd�tjd�}|�� }tj|d dddd d d d d dddg d�}||d< t�t |��|d< |� |d� t |��}tj |ddd�}W d Q R X t �||� d S )Ni����r� )r� r& )r� )r� r r> r? r� r� �E�F�G�H�I�J)�labelsr� r� )r, rA )r rm r� r� r� ry r/ r� ZSeriesrk r. r r0 r r1 )r2 r r5 r3 Z intervalsrD r� r r r � test_to_excel_interval_labels� s $ z-TestExcelWriter.test_to_excel_interval_labelsc C s� t tjjdddd�dgtjd�}|�� }|d �dd� �|d < |d �d d� �|d < |�|d � t|��}t j |d d d �}W d Q R X t � ||� d S )Ni����r� )r� r& )r� r> )r= r� c S s t | d�S )N)�seconds)r )rW r r r rv � rw z9TestExcelWriter.test_to_excel_timedelta.<locals>.<lambda>r� c S s t | d��� d S )N)r� i�Q )r � total_seconds)rW r r r rv � rw r� r )r, rA ) r rm r� r� r� ry �applyr. r r/ r0 r r1 )r2 r r5 r3 rD r� r r r �test_to_excel_timedelta� s z'TestExcelWriter.test_to_excel_timedeltac C sX |j ddd��� }|�|d� t|��}tj|ddd�}W d Q R X t�||�d�� d S )N�MZperiod)�kindZsht1r )r, rA ) ZresampleZmeanr. r r/ r0 r r1 Z to_period)r2 r� r r� rD r� r r r �test_to_excel_periodindex� s z)TestExcelWriter.test_to_excel_periodindexc C s� t �t|j�d ��dd�}tj|ddgd�}||_|j|ddd� |j|dd d gd � |j|d|d � t|��}t j |dd dgd�}W d Q R X t � ||� d S )Nr( rg �first�second)r@ r� F)r+ r> r? )r= )r� r r& )r, rA ) rm �aranger� r* �reshaper r� r. r r/ r0 r r1 )r2 r� r� r �arrays� new_indexrD r5 r r r �test_to_excel_multiindex� s z(TestExcelWriter.test_to_excel_multiindexc C sb t d ddgdddgtj�d�d��}|�ddg�}|j||d � tj|d d gd �}t� ||� d S ) Nr( r$ r� r� � )r> r? r� r> r? )r� r r& )rA ) r rm r� �sampler� r. r/ r0 r r1 )r2 r� r r5 Zdf1r r r �"test_to_excel_multiindex_nan_label s $z2TestExcelWriter.test_to_excel_multiindex_nan_labelc C s� t �t|j�d ��dd�}tj|ddgd�}||_t�dddd g�}||_d d g}|s\d }|j |d |d � t |�� }t j |d |d d gd�} W d Q R X |s�|jj dddd�} dd� t| � D �|_t�|| � d S )Nr( rg r� r� )r@ )�( r& )r� r( )�2 r& )r� r( r r& r� )r� )r, r+ rA F)ZsparsifyZadjoinr@ c S s g | ]}d � tt|���qS )r )�join�mapr� )rL �qr r r rN s zATestExcelWriter.test_to_excel_multiindex_cols.<locals>.<listcomp>)rm r� r� r* r� r r� � from_tuplesr= r. r r/ r0 �formatrP r r1 ) r2 r� r� r r� r� Znew_cols_indexr+ rD r5 �fmr r r � test_to_excel_multiindex_cols s z-TestExcelWriter.test_to_excel_multiindex_colsc C s� |j t�t|j ��g}t�|�|_ ddg|j _|j|d|d� t|��}t j |dddgd�}W d Q R X t � ||� |j jdks�t �d S ) N�timer9 r� )r� r r& )r, rA )r� r9 )r* rm r� r� r r� r@ r. r r/ r0 r r1 r� )r2 r� r� r r� rD r� r r r � test_to_excel_multiindex_dates# s z.TestExcelWriter.test_to_excel_multiindex_datesc C sz t ddgddgddgd��}|�� }t�dd g�}||_|j|d d d � t|��}tj|d d �}W d Q R X t � ||� d S )Nr� r� r� r� r� �< )r; r< �c)�F �P )�Z r' r� F)r* )r, ) r ry r r� r* r. r r/ r0 r r1 )r2 r Zframe1r� Z multi_indexrD Zframe3r r r �'test_to_excel_multiindex_no_write_index0 s z7TestExcelWriter.test_to_excel_multiindex_no_write_indexc C s� t dddgdddggddgd d d gd �}|j|d dd� t|��}tj|d dd�}W d Q R X t dddgdddggddgd d d gd �}t�||� d S )Ng��~�Ϛ�?g9DܜJ�?g >�D�)�?g�.4�i�(@g33333�@g���̤�Ar> r? rY rZ r� )r* r= r� z%.2f)� float_formatr )r, rA g� ��Q��?gq= ףp�?g= ףp=�?g�p= ף(@)r r. r r/ r0 r r1 )r2 r r5 rD r6 r3 r r r �test_to_excel_float_formatE s z*TestExcelWriter.test_to_excel_float_formatc C sr t dddgdddggddgd d d gd �}t�d | ��2}|j|ddd� tj|ddd�}t�||� W d Q R X d S )Nu ƒu Ɠu Ɣu ƕu Ɩu Ɨu Aƒr? u XƓrZ r� )r* r= z __tmp_to_excel_float_format__.Z TestSheet�utf8)r, �encodingr )r, rA )r r r r. r/ r0 r1 )r2 r r5 r4 r6 r r r � test_to_excel_output_encodingW s z-TestExcelWriter.test_to_excel_output_encodingc C s� t �d| ���}z2yt|d�}W n tk r> t�d� Y nX W d |�� X tdddgddd ggd d gd d dgd�}|j|ddd� t |��}t j |ddd�}W d Q R X W d Q R X tdddgddd ggd d gd d dgd�}t � ||� d S )Nu ƒu.�wbz$No unicode file names on this systemg��~�Ϛ�?g9DܜJ�?g >�D�)�?g�.4�i�(@g33333�@g���̤�Ar> r? rY rZ r� )r* r= r� z%.2f)r� r )r, rA g� ��Q��?gq= ףp�?g= ףp=�?g�p= ף(@) r r �open�UnicodeEncodeErrorrh ri �closer r. r r/ r0 r1 ) r2 r r r4 �fr5 rD r6 r3 r r r � test_to_excel_unicode_filenamed s$ $z.TestExcelWriter.test_to_excel_unicode_filename� use_headersTF� r_idx_nlevelsr& r( r$ � c_idx_nlevelsc s0 d� �fdd� }|rdnd }|t dg�||�}|jdks<t�|jd tjk sPt�d} d} tj| | ||d �} |d kr�d } tj t | d �� || |d d� W d Q R X n�|| |�}|r�|j| | | fks�t�n|j| d | | fks�t�xJt t |j ��D ]8} x2t t |j��D ] }|j| |f tjk �st��qW q�W d S )NTr c s< | j �|� |d� t���}tj||jd |d�S Q R X d S )N)r+ r� r* r )r, r+ )r. r r/ r0 r� )�datar+ Z parser_hdrr* r� )r� r r r � roundtrip� s z;TestExcelWriter.test_excel_010_hemstring.<locals>.roundtrip)r& r( )r r re r$ )r r r& z_Writing to Excel with MultiIndex columns and no index \('index'=False\) is not yet implemented.)r� F)r* )Tr T)r r� r� ro rm rn r rj rh r� �NotImplementedErrorrl r� r* r= )r2 r� r r r r r Z parser_headerr| ZnrowsZncolsr5 r� �rr� r )r� r r �test_excel_010_hemstring� s( z(TestExcelWriter.test_excel_010_hemstringc C s� t dddgdddgdddggdddgd�}|�|d� t dddgdddgdddggdddgd�}tj|dd d �}t�||� tj|dd d d �}t�||� t dddd gddddggddddgd�}|�|d� tj|dd d �}t dddd gddddggddddgd�}t�||� |j|dddd� tj|dd d�}t dddd gddddgg�}t�||� d}tjt|d�� tj|dd dd� W d Q R X d S )Nr& r( r$ r> r? )r= r� zB.1r )r, rA T)r, rA �mangle_dupe_colsr# re � rU � zA.1F)r* r+ )r, r+ z3Setting mangle_dupe_cols=False is not supported yet)r� )r, r+ r ) r r. r/ r0 r r1 rh r� r� )r2 r r5 r3 r6 r� r r r �test_duplicated_columns s, * ( ( & z'TestExcelWriter.test_duplicated_columnsc C sj t dddgdddgd��}|j|dddgd� tj|ddd �}t�|d |d � t�|d |d � d S ) Nr& r( )r> r? r� r? r> )r= r )r, r+ )r r. r/ r0 r �assert_series_equal)r2 r � write_frame� read_framer r r �test_swapped_columnsA s z$TestExcelWriter.test_swapped_columnsc C sz t dddgdddgd��}tjtdd�� |j|dddgd � W d Q R X tjtd d�� |j|ddd gd � W d Q R X d S ) Nr& r( )r> r? zNot all names specified)r� r� r? r� )r= z.'passes columns are not ALL present dataframe'r� )r rh r� �KeyErrorr. )r2 r r r r r �test_invalid_columnsK s z$TestExcelWriter.test_invalid_columnsz#to_excel_index,read_excel_index_col)Tr )FNc C sd t dddgdddgdddgd��}|j|dddg|d� |ddg }tj|d|d �}t�||� d S ) Nr& r( r$ )r> r? r� Zcol_subset_bugr> r? )r= r* )r, rA )r r. r/ r0 r r1 )r2 r Zto_excel_indexZread_excel_index_colr r3 r r r r �test_write_subset_columnsW s " z)TestExcelWriter.test_write_subset_columnsc C sv t dddgdddgd��}|�|d� tj|ddd�}d |jd < d |jd < d |jd < tj|dd dd �}t�||� d S )N�onez#one�twoz#two)r> r? �test_cr )r, rA )r& r )r& r& )r( r& �#)r, �commentrA )r r. r/ r0 ro r r1 )r2 r r5 �result1�result2r r r �test_comment_argl s z TestExcelWriter.test_comment_argc C sT t dddgdddgd��}|�|d� tj|dd�}tj|dd d�}t�||� d S ) Nr z#oner z#two)r> r? r )r, )r, r! )r r. r/ r0 r r1 )r2 r r5 r" r# r r r �test_comment_default s z$TestExcelWriter.test_comment_defaultc C sb t dddgdddgd��}|�|d� t dd dgdd d gd��}tj|dddd �}t�||� d S ) Nr z#oner z#two)r> r? r r r )r, r! rA )r r. r/ r0 r r1 )r2 r r5 r3 r6 r r r �test_comment_used� s z!TestExcelWriter.test_comment_usedc C sT t ddgddgd��}|j|dd� t dgd gd��}tj|d d �}t�||� d S ) N�1z#2�2�3)r; r< F)r* r& r( r )r! )r r. r/ r0 r r1 )r2 r r5 r3 r6 r r r �test_comment_empty_line� s z'TestExcelWriter.test_comment_empty_linec C s t dddddd�t dddddd�t ddddd d �t dddd dd �t dddd dd�t dddddd�t dddddd�t dddddd�t dddddd �t dddddd�t ddddd d �g }td |i�}|�|d � |�d �s�|�d!�r�t�d"� tj|d dd#�}t� |d |d � d S )$Ni� r& r� r( r$ �- �8 r# � �1 r �* rU �9 �# r� �) r� rF � � � �5 � �% r � r� �4 r> r� �xlsxZxlsmz*Defaults to openpyxl and fails - GH #38644)r, r+ ) r r r. �endswithrh ri r/ r0 r r )r2 r Z datetimesr r r r r �test_datetimes� s" z TestExcelWriter.test_datetimesc C sn t � �^}ttj�dd��}t||d��}|�|� W d Q R X |�d� tj |dd�}t � ||� W d Q R X d S )Nr� r( )r r )rA ) r r rm r� r� r r. �seekr/ r0 r r1 )r2 r Zbior5 rC Z reread_dfr r r � test_bytes_io� s z TestExcelWriter.test_bytes_ioc C s~ t dddgddd�gdddgd d d gd ��}|�|d � tj|d ddd�}|�� }|j�t�|_|j� d�|_t � ||� d S )Nr; r< r� �er( )rI r r& g @�apple�bananaZcherry)�mixed�numericr� r� r )r, r+ rA r� ) r r. r/ r0 ry rC r� r� rD r� r r1 )r2 r r5 �readr3 r r r �test_write_lists_dict� s z%TestExcelWriter.test_write_lists_dictc C sF t ddgddgd��}|�|d� tj|ddd�}|}t�||� d S ) Nr& r( r$ r# )�renderr r� r )rA )r r. r/ r0 r r1 )r2 r r5 rE r3 r r r �test_render_as_column_name� s z*TestExcelWriter.test_render_as_column_namec C sV t ddggddgd�}|�ddd��}|�|� tj|dgdgd d �}t�||� d S ) Nr9 r: �col1�col2)r= TF)r9 r: r )Z true_valuesZ false_valuesrA )r �replacer. r/ r0 r r1 )r2 r r5 r3 r r r r �!test_true_and_false_value_options� s z1TestExcelWriter.test_true_and_false_value_optionsc C sJ t ddgddggddgd�}|j|dd d � tj|d d �}t�||� d S ) Nr& r( r$ r# rI rJ )r= r� )r& r& )Z freeze_panesr )rA )r r. r/ r0 r r1 )r2 r r3 r6 r r r �test_freeze_panes� s z!TestExcelWriter.test_freeze_panesc C sJ t �� }t|j|d�}ttjdd�}t j||d|� �d�}t �||� d S )N)r r )rA r9 )r )r � makeDataFramer r. r/ r0 Zround_trip_pathlibr1 )r2 r r r5 rC rD r6 r r r �test_path_path_lib� s z"TestExcelWriter.test_path_path_libc C sJ t �� }t|j|d�}ttjdd�}t j||d|� �d�}t �||� d S )N)r r )rA r9 )r )r rN r r. r/ r0 Zround_trip_localpathr1 )r2 r r r5 rC rD r6 r r r �test_path_local_path s z$TestExcelWriter.test_path_local_pathc C s� t �t�d�t�d�ft�d�t�d�fg�}tt�d�|d�}|�|� tj t dd�� tj |dd gdd d �}W d Q R X |j j d d � |jd D �d d � |jd D �gdd gd�|_ |j�tj�|_t�||� d S )NZ2018Z2018Q1Z2018Q2)r( r( )r= zconvert_float is deprecated)r� r r& F)r+ rA r� c S s g | ] }t |��qS r )r� )rL rJ r r r rN s zCTestExcelWriter.test_merged_cell_custom_objects.<locals>.<listcomp>)�level)r r� r/ ZPeriodr rm Zonesr. r r� r� r0 r= Z set_levelsZlevelsr* r� �float64r1 )r2 r� r �mir3 r6 r r r � test_merged_cell_custom_objects s &z/TestExcelWriter.test_merged_cell_custom_objectsr� Nc C s� |}t jd|d�}t|g|d�}tjtdd�� |�|� W d Q R X |�� }t|g|d�}tjtdd�� |�|� W d Q R X d S )NZ2019)�tz)r� zExcel does not support)r� )r/ r� r rh r� r� r. Z to_pydatetime)r2 r� Ztz_aware_fixturer rU r r5 r r r � test_raise_when_saving_timezones! s z0TestExcelWriter.test_raise_when_saving_timezonesc C sh t ddgddgd��}|j|dddgdd � t�|�}t dddgdddggddd gd �}t�||� d S ) Nr r& r� rF )r> r? r> r? F)r= r* zA.1)r= )r r. r/ r0 r r1 )r2 r r5 r6 r3 r r r �'test_excel_duplicate_columns_with_names/ s "z7TestExcelWriter.test_excel_duplicate_columns_with_namesc C sJ d}t �|��2}tjtt�|�d�� t|dd� W d Q R X W d Q R X d S )Nz7if_sheet_exists is only valid in append mode (mode='a'))r� rK )Zif_sheet_exists)r r rh r� r� �re�escaper )r2 r r� r r r r �test_if_sheet_exists_raises8 s z+TestExcelWriter.test_if_sheet_exists_raises)Cr� r� r� r� r� r� r� r� r� r� rh r� r� rm Zint8Zint16Zint32r� r� Zfloat16Zfloat32rR r� Zbool8Zbool_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 r$ r% r& r* r= r? rF rH rL rM rO rP rT �objectrV rW rZ r r r r r� 4 sn 1 & " 9 , 5' r� c @ s� e Zd Zej�dejede� d�d�eje de� d�d�eje de� d�d�g�dd � �Z d d � Z d d � Zej�dejde� d�d�ejde� d�d�ejde� d�d�g�dd� �ZdS )�TestExcelWriterEngineTestsz klass,extz.xlsxr )r r z.xlsr c C s^ t �|��J}t|��6}|dkr8t�d�r8t|t�sFt�nt||�sFt�W d Q R X W d Q R X d S )Nz.xlsxr )r r r �tdZ safe_import� isinstancer r� )r2 �klassr r rC r r r �test_ExcelWriter_dispatchB s z4TestExcelWriterEngineTests.test_ExcelWriter_dispatchc C s& t jtdd�� td� W d Q R X d S )Nz No engine)r� Znothing)rh r� r� r )r2 r r r � test_ExcelWriter_dispatch_raisesS s z;TestExcelWriterEngineTests.test_ExcelWriter_dispatch_raisesc s� g � g �G � �fdd�dt �}� �fdd�}t�dd��� d}t�|��N�t|� t ���}t||�sjt�W d Q R X t�dd��|��fd d �� W d Q R X t�d ���|��fd d �� W d Q R X W d Q R X d S ) Nc s8 e Zd ZdZdZddgZdZ� fdd�Z�fdd�Zd S ) zCTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClassFr; Zxlsr� c s � � d� d S )NT)�append)r2 )� called_saver r �saveb s zHTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass.savec s � � d� d S )NT)rb )r2 �args�kwargs)�called_write_cellsr r � write_cellse s zOTestExcelWriterEngineTests.test_register_writer.<locals>.DummyClass.write_cellsN) r� r� r� rc rg Zsupported_extensionsr rd rh r )rc rg r r � DummyClass\ s ri c s> | � t � �dkst�t ��dks&t�� d d �= �d d �= d S )Nr& )r� r� )�func)rc rg r r � check_calledh s zETestExcelWriterEngineTests.test_register_writer.<locals>.check_calledzio.excel.xlsx.writerr� zsomething.xlsxr& c s � � ��S )N)r. r )r5 �filepathr r rv v rw zATestExcelWriterEngineTests.test_register_writer.<locals>.<lambda>z something.xlsc s � j �dd�S )Nr� )r )r. r )r5 rl r r rv x rw ) r r/ Zoption_contextr r r r^ r� rj )r2 ri rk r rC r )rc rg r5 rl r �test_register_writerW s z/TestExcelWriterEngineTests.test_register_writerr z.odsr! c C sF t �d�}tjt|d��$ tdddidd�� W d Q R X W d Q R X d S )Nz*Cannot use both engine_kwargs and **kwargs)r� r% r; r& r( )Z engine_kwargsr< )rX rY rh r� r� r )r2 r r� r r r �$test_engine_kwargs_and_kwargs_raisesz s z?TestExcelWriterEngineTests.test_engine_kwargs_and_kwargs_raisesN)r� r� r� rh r� r� �paramr r] � skip_if_nor r r` ra rm rn r r r r r\ A s " #r\ c @ s e Zd Zdd� Zdd� ZdS )� TestFSPathc C s^ t �d��J}tdddgi�}|�|� t|��}t�|�}W d Q R X ||ksPt�W d Q R X d S )Nzfoo.xlsxr> r& r( )r r r r. r �os�fspathr� )r2 r r5 r� r6 r r r �test_excelfile_fspath� s z TestFSPath.test_excelfile_fspathc C sD t �d��0}t|�� }t�|�t|�ks,t�W d Q R X W d Q R X d S )Nzfoo.xlsx)r r r rr rs r� r� )r2 r rC r r r �test_excelwriter_fspath� s z"TestFSPath.test_excelwriter_fspathN)r� r� r� rt ru r r r r rq � s rq )-r r r � functoolsr �ior rr rX �numpyrm rh Z pandas.util._test_decorators�utilZ_test_decoratorsr] Zpandasr/ r r r r r Zpandas._testingZ_testingr Zpandas.io.excelr r r r r r Zfixturer r r� r� ro rp r" Z usefixturesr� r\ rq r r r r �<module> sd & n |I
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка