Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/excel/__pycache__/test_readers.cpython-37.pyc
Назад
B �5�g�� � @ s� d dl m Z mZ d dlmZ d dlZd dlmZ d dlmZ d dl m Z d dl Z d dl Z d dlm mZ d dlZd dlmZmZmZmZ d dlmZ d dlm Z d d l m Z d d d d dgZ e j de�!d�gd�e j de�!d�e j"�#d�gd�e j de�!d�gd�e j de�!d�d�e j de�!d�d�gZ$e%e&d�dd�Z'dd� Z(e j)ddd � e$D �d �d d � �Z*e j)d d!� �Z+e j)d"d#� �Z,G d$d%� d%�Z-G d&d'� d'�Z.dS )(� )�datetime�time)�partialN)�Path)�URLError)� BadZipFile)� DataFrame�Index� MultiIndex�Series)� xlrd_version)�Versionz.xlsz.xlsxz.xlsmz.xlsbz.ods�xlrd)�marks�openpyxlzignore:.*html argument�pyxlsb�odf)�read_ext�returnc C s� | j d } | dkr |dkr dS | dkr2|dkr2dS |dkrF| dkrFdS | dkrZ|dkrZdS |dkrn| dkrndS | d kr�td k r�ttd �kr�|dkr�dS d S ) zm Filter out invalid (engine, ext) pairs instead of skipping, as that produces 500+ pytest.skips. r r z.xlsFr z.odsr z.xlsbr N�2T)�valuesr r )�enginer � r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/excel/test_readers.py�_is_valid_engine_ext_pair8 s" r c C s | j |f }tj|| jd�}|S )z� engine gives us a pytest.param object with some marks, read_ext is just a string. We need to generate a new pytest.param inheriting the marks. )r )r �pytest�paramr )r r r � new_paramr r r �_transfer_marksR s r Tc C s* g | ]"}t D ]}t||�r t||��q qS r )�read_ext_paramsr r )�.0Zeng�extr r r � <listcomp>_ s r" )�autouse�paramsc C s | j S )zS Fixture for Excel reader engine and read_ext, only including valid pairs. )r )�requestr r r �engine_and_read_ext\ s r& c C s | \}}|S )Nr )r& r r r r r r l s r c C s | \}}|S )Nr )r& r r r r r r r s r c @ s� e Zd Zejdd�dd� �Zdd� Zdd� Zd d � Zd d � Z ej � d dddgdddgdddgdddgdddgdddgg�dd� �Z ej � d ddgddgg�dd� �Z dd� Zdd� Zdd � Zd d � Zd d � Zej � d!d"d#g�d$d%� �Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zej � d6d"e dd#dd7gd8d9d:d;gdd#dd7gd<d=e j d>gd?��fd@dAe e d?�e e dd#dd7gd@dB�e d8d9d:d;gdAdB�dCdDdEdFgdGdHe j dIgd?��fg�dJdK� �Z!ej � dLi dGfdMdNidfg�dOdP� �Z"dQdR� Z#ej � dSdTe dUdVgi�fdWe dXgdY�fg�dZd[� �Z$d\d]� Z%d^d_� Z&d`da� Z'dbdc� Z(ddde� Z)dfdg� Z*dhdi� Z+djdk� Z,dldm� Z-ej � dndddgddgdodpdogdodpgg�dqdr� �Z.dsdt� Z/dudv� Z0e1j2dwdx� �Z3e4j5dydz� �Z6d{d|� Z7ej j8d}d~� �Z9dd�� Z:e4�;d��e4j<d�d�� ��Z=e4j<d�d�� �Z>d�d�� Z?d�d�� Z@ej � d�d�e j d�d�d�gfd�e j gd7 fg�d�d�� �ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd"S )�� TestReadersT)r# c C s2 t tj|d�}|�|ddd�� |�td|� dS )zG Change directory and set engine for read_excel calls. )r �io�data�excel� read_excelN)r �pdr+ �chdir�setattr)�selfr �datapath� monkeypatch�funcr r r �cd_and_set_enginey s z TestReaders.cd_and_set_enginec C sz dd� }|� tjd|� dddddd�}td | d ��}t�|�}W d Q R X |d k rZ|}n||d d � }||ksvt�d S ) Nc _ s | j S )N)r )r/ �args�kwargsr r r �parser� s z,TestReaders.test_engine_used.<locals>.parser�parser r r r )�xlsx�xlsm�xlsb�xls�ods�test1�rb� )r. r, � ExcelFile�openr+ �AssertionError) r/ r r r1 r6 �expected_defaults�f�result�expectedr r r �test_engine_used� s z TestReaders.test_engine_usedc C s~ |j dddgd�}d}tjt|d�� tjd| dd d d � W d Q R X tjt|d�� tjd| d d gd d d� W d Q R X d S )N�A�B�C)�columnsz Passing an integer for `usecols`)�matchr= �Sheet1r � )� sheet_name� index_col�usecols�Sheet2r? )rO �skiprowsrP rQ )�reindexr �raises� ValueErrorr, r+ )r/ r �df_ref�msgr r r �test_usecols_int� s z TestReaders.test_usecols_intc C s� t jjd dkr&|j�tjjdd�� |jddgd�}t jd| d d d d d gd �}t jd| ddgd d d d gd�}t j ||dd� t j ||dd� d S )Nr r z3Sheets containing datetimes not supported by pyxlsb)�reasonrI rJ )rK r= rM r � rN )rO rP rQ rR r? )rO rS rP rQ F)� check_names) r, r+ �keywords�node� add_markerr �mark�xfailrT �tm�assert_frame_equal)r/ r% r rW �df1�df2r r r �test_usecols_list� s z TestReaders.test_usecols_listc C sL t jjd dkr&|j�tjjdd�� |jdddgd�}t jd | d d d d �}t jd | ddgd d d�}t j ||dd� t j ||dd� |jddgd�}t jd | d d dd �}t jd | ddgd dd�}t j ||dd� t j ||dd� |jddgd�}t jd | d d dd �}t jd | ddgd dd�}t j ||dd� t j ||dd� d S )Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ rH rI rJ )rK r= rM r zA:D)rO rP rQ rR r? )rO rS rP rQ F)r\ zA,C,DzA,C:D) r, r+ r] r^ r_ r r` ra rT rb rc )r/ r% r rW rd re �df3r r r �test_usecols_str� sJ z TestReaders.test_usecols_strrQ r r? rN c C s\ t jjd dkr&|j�tjjdd�� |ddg }t jd| dd |d �}tj ||d d � d S ) Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ rH rJ r= rM r )rO rP rQ F)r\ ) r, r+ r] r^ r_ r r` ra rb rc )r/ r% r rQ rW rF rE r r r �.test_usecols_diff_positional_int_columns_order� s z:TestReaders.test_usecols_diff_positional_int_columns_orderrI �Dc C sB |ddg }t t|��|_tjd| d|d�}tj||dd� d S )NrI rj r= rM )rO rQ F)r\ )�range�len�indexr, r+ rb rc )r/ r rQ rW rF rE r r r �.test_usecols_diff_positional_str_columns_order s z:TestReaders.test_usecols_diff_positional_str_columns_orderc C sR t jjd dkr&|j�tjjdd�� |}t jd| ddd�}tj ||d d � d S ) Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ r= rM r )rO rP F)r\ ) r, r+ r] r^ r_ r r` ra rb rc )r/ r% r rW rF rE r r r � test_read_excel_without_slicing s z+TestReaders.test_read_excel_without_slicingc C s\ t jjd dkr&|j�tjjdd�� |ddg }t jd| dd d d �}tj ||d d � d S )Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ rJ rj r= rM r zA,D:E)rO rP rQ F)r\ ) r, r+ r] r^ r_ r r` ra rb rc )r/ r% r rW rF rE r r r � test_usecols_excel_range_str" s z(TestReaders.test_usecols_excel_range_strc C s6 d}t jt|d�� tjd| ddd� W d Q R X d S )NzInvalid column name: E1)rL r= rM zD:E1)rO rQ )r rU rV r, r+ )r/ r rX r r r �$test_usecols_excel_range_str_invalid0 s z0TestReaders.test_usecols_excel_range_str_invalidc C s> d}t jt|d��" tjd| ddgddgd� W d Q R X d S )Nz(list indices must be integers.*, not str)rL r= rM rH rJ )rO rP rQ )r rU � TypeErrorr, r+ )r/ r rX r r r �test_index_col_label_error6 s z&TestReaders.test_index_col_label_errorc C sX t jd| ddddgd�}tddd gtg gd g gd dddgd �d �}t�||� d S ) Nr= �Sheet3rH rI rJ )rO rP rj �E�FrN )�levels�codes�names)rK rm )r, r+ r r rb rc )r/ r rE rF r r r �test_index_col_emptyA s $z TestReaders.test_index_col_emptyrP Nr[ c C sZ t jd| d|d�}tdddgddd ggd d d gd �}|rJ|�|j| �}t�||� d S )Nr= �Sheet4)rO rP �i1�a�x�i2�b�yz Unnamed: 0Zcol1Zcol2)rK )r, r+ r � set_indexrK rb rc )r/ r rP rE rF r r r �test_index_col_with_unnamedL s z'TestReaders.test_index_col_with_unnamedc C s6 d}t jt|d�� tjd| dgd� W d Q R X d S )NzEUsecols do not match columns, columns expected but not found: \['E'\])rL r= ru )rQ )r rU rV r, r+ )r/ r rX r r r �%test_usecols_pass_non_existent_columnZ s z1TestReaders.test_usecols_pass_non_existent_columnc C s8 d}t jt|d�� tjd| ddgd� W d Q R X d S )Nz['usecols' must either be list-like of all strings, all unicode, all integers or a callable.)rL r= ZE1r )rQ )r rU rV r, r+ )r/ r rX r r r �test_usecols_wrong_typec s z#TestReaders.test_usecols_wrong_typec C s8 t jd| dd�}tddggddgd�}t�||� d S ) NZtest2rM )rO ZaaaaZbbbbb�TestZTest1)rK )r, r+ r rb rc )r/ r �parsedrF r r r �test_excel_stop_iteratorl s z$TestReaders.test_excel_stop_iteratorc C s\ t jjd dkr&|j�tjjdd�� t jd| dd�}tt j ggdgd �}t � ||� d S ) Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ Ztest3rM )rO r� )rK ) r, r+ r] r^ r_ r r` ra r �np�nanrb rc )r/ r% r r� rF r r r �test_excel_cell_error_nar s z$TestReaders.test_excel_cell_error_nac C s� t jjd dkr&|j�tjjdd�� t jd| ddd�}t jd| d d gdd �}tj ||d d � tj ||d d � t jd| ddd d�}t� ||j d d� � d S )Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ r= rM r )rO rP rR r? )rO rS rP F)r\ )rO rP � skipfooter���) r, r+ r] r^ r_ r r` ra rb rc �iloc)r/ r% r rW rd re rg r r r �test_excel_table~ s z TestReaders.test_excel_tablec C s t jjd dkr&|j�tjjdd�� t� ddddd gd d d d dgdddddgdddddgdddddgt ddd�t ddd�t ddd�t dd d �t d dd �gd ��}d }t j|| d!d"�}t � ||� |� � }|d# �t�|d#< d$|j|jd d%f< t jtd&dd'�� t j|| d!dd(�}W d Q R X t � ||� xBt|j�D ]4\}}t j|| d!|d)�}|�|�} t � || � �q:W |d* �t�|d*< t j|| d!d*tid+�}t � ||� |� � } | d* �t�| d*< t jtd&dd'�� t j|| d!dd*tid,�}W d Q R X t � || � d S )-Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ r? r[ ���� r g �?g @gH�z�G�?g� ��Q��?g��&� .>TFrN � r} �c�d�ei� � � � iq � � i� )�IntCol�FloatCol�BoolCol�StrCol�Str2ColZDateColZ test_typesrM )rO r� g @r� zconvert_float is deprecated)rL �raise_on_extra_warnings)rO � convert_float)rO rP r� )rO � converters)rO r� r� )r, r+ r] r^ r_ r r` ra r � from_dictr rb rc �copy�astype�float�locrm �assert_produces_warning� FutureWarning� enumeraterK r� �apply�str) r/ r% r rF �basename�actualZfloat_expectedZicol�name�expZno_convert_floatr r r �test_reader_special_dtypes� sb z&TestReaders.test_reader_special_dtypesc C s� d}t �dddddgdtjdd d gd d d d d gd tjdddgd��}dd� dd� dd� dd� d�}tj|| d|d�}t�||� d S )NZtest_convertersr? r[ r� i���r g )@g�����L2@g3333333@g:�0�y5>�Foundz Not found�1�3�4�5)r� r� r� r� c S s | dkrt | �S dS )N� i���)�int)r~ r r r �<lambda>� � z4TestReaders.test_reader_converters.<locals>.<lambda>c S s | r d| S t jS )Nr� )r� r� )r~ r r r r� � r� c S s | dkr dS dS )Nr� r� z Not foundr )r~ r r r r� � r� c S s | r t | �S dS )Nr� )r� )r~ r r r r� � r� )r� r� r[ rN rM )rO r� )r r� r� r� r, r+ rb rc )r/ r r� rF r� r� r r r �test_reader_converters� s z"TestReaders.test_reader_convertersc C s� d}t �|| �}tddddgdddd gddddgd d tjd gd ��jddddgd�}t�||� t j|| ddtd�d�}|d � d�|d< |d � d�|d< ddddg|d< t�||� d}t j t |d �� t j|| dd id� W d Q R X d S ) N� testdtyper? r[ rN r� g @g @g @g @g �?g @g @)r} r� r� r� r} r� r� r� )rK �float64�float32)r} r� r� )�dtype�001�002�003�004z(Unable to convert column d to type int64)rL �int64) r, r+ r r� r� rT rb rc r� r� r rU rV )r/ r r� r� rF rX r r r �test_reader_dtype s$ z TestReaders.test_reader_dtypezdtype,expectedr� g @g @g @g @g �?g @g @)r} r� r� r� r� r� )r� r� r� r� r� r� r r� c C s& d}t j|| |d�}t�||� d S )Nr� )r� )r, r+ rb rc )r/ r r� rF r� r� r r r �test_reader_dtype_str s z!TestReaders.test_reader_dtype_strzdtypes, exp_valueza.1r� c C s\ d}dt i|�}|�� }tj|| |d�}tdg|gd��}||ksLtd��t�||� d S )NZdf_mangle_dup_col_dtypesr} )r� r� )r} za.1zdtype dict changed)r� r� r, r+ r rB rb rc ) r/ r ZdtypesZ exp_valuer� Z dtype_dictZdtype_dict_copyrE rF r r r �test_dtype_mangle_dup_cols? s z&TestReaders.test_dtype_mangle_dup_colsc C s8 d}t �|| �}tddddddgi�}t�||� d S )NZ test_spacesZtestcolz this is greatz 4 spacesz 1 trailing z 1 leadingz2 spaces multiple times)r, r+ r rb rc )r/ r r� r� rF r r r �test_reader_spacesK s z TestReaders.test_reader_spaceszbasename,expectedzgh-35802ZCOLUMNzTest (1)zgh-36122z got 2nd sa)rK c C sB t jjd }|dkr$t�d|� �� t �|| �}t�||� d S )Nr r zSkipped for engine: )r, r+ r] r �skiprb rc )r/ r r� rF r r� r r r � test_read_excel_ods_nested_xml^ s z*TestReaders.test_read_excel_ods_nested_xmlc C sH d}t j|| d d�}dddg}t�||�� � |t|�� �ksDt�d S )N�test_multisheet)rO �Charlie�Alpha�Beta)r, r+ rb �assert_contains_all�keys�listrB )r/ r r� �dfs� expected_keysr r r �test_reading_all_sheetsn s z#TestReaders.test_reading_all_sheetsc C sX d}dddg}t j|| |d�}tt|��}t�||�� � t|�t|�� �ksTt�d S )Nr� r[ r� )rO ) r, r+ r� �setrb r� r� rl rB )r/ r r� r� r� r r r �%test_reading_multiple_specific_sheets{ s z1TestReaders.test_reading_multiple_specific_sheetsc C s4 d}t j|| d d�}dddg}t�||�� � d S )N�blank_with_header)rO rM rR rt )r, r+ rb r� r� )r/ r r� r� r� r r r �"test_reading_all_sheets_with_blank� s z.TestReaders.test_reading_all_sheets_with_blankc C s$ t jd| dd�}t�|t� � d S )N�blankrM )rO )r, r+ rb rc r )r/ r r� r r r �test_read_excel_blank� s z!TestReaders.test_read_excel_blankc C s0 t ddgd�}tjd| dd�}t�||� d S )NZcol_1Zcol_2)rK r� rM )rO )r r, r+ rb rc )r/ r rF r� r r r �!test_read_excel_blank_with_header� s z-TestReaders.test_read_excel_blank_with_headerc C s� t jjd dkr&|j�tjjdd�� tt � d�dgt � d�dgd d ggd d gd �}t jjd dkrz|j�tjjdd�� t jjd d kr�|dkr�|j�tjjdd�� t �d| �}t � ||� d S )Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ z 2016-03-12z Marc Johnsonz 2016-03-16z Jack Blackg@��x �Dz Timothy BrownZDateColWithBigIntZ StringCol)rK r z Maybe not supported by openpyxl)z.xlsxz.xlsmz)Defaults to openpyxl, maybe not supportedZtestdateoverflow) r, r+ r] r^ r_ r r` ra r � Timestamprb rc )r/ r% r rF rE r r r � test_date_conversion_overflow� s" z)TestReaders.test_date_conversion_overflowc C sz t jjd dkr&|j�tjjdd�� d}d}t j|| |dd�}t j|| d|d �}tj ||d d � tj ||d d � d S ) Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ r= rM r )rO rP )rP rO F)r\ ) r, r+ r] r^ r_ r r` ra rb rc )r/ r% r rW �filenamerO rd re r r r �test_sheet_name� s zTestReaders.test_sheet_namec C sN d| }t j|ddd�}t|d��"}t j|ddd�}t�||� W d Q R X d S )Nr= rM r )rO rP r> )r, r+ rA rb rc )r/ r �pthrF rD r� r r r �test_excel_read_buffer� s z"TestReaders.test_excel_read_bufferc C s0 d}t jtdd�� tjd|d� W d Q R X d S )N�foozUnknown engine: foo)rL r� )r )r rU rV r, r+ )r/ r Z bad_enginer r r �test_bad_engine_raises� s z"TestReaders.test_bad_engine_raisesrO r{ rM c C s4 d}t jt|d�� tjd| |d� W d Q R X d S )Nz?Worksheet index 3 is invalid|Worksheet named 'Sheet4' not found)rL r� )rO )r rU rV r, r+ )r/ r rO rX r r r �test_bad_sheetname_raises� s z%TestReaders.test_bad_sheetname_raisesc C s2 d|� �}t jtdd�� t�|� W d Q R X d S )Nr� u7 (No such file or directory|没有那个文件或目录))rL )r rU �FileNotFoundErrorr, r+ )r/ r Zbad_filer r r �test_missing_file_raises� s z$TestReaders.test_missing_file_raisesc C sd d}|d krt }d}n&|dkr4ddlm} |}d}nt}d}tj||d�� t�|� W d Q R X d S ) Ns foozLExcel file format cannot be determined, you must specify an engine manually.r r )� XLRDErrorzFUnsupported format, or corrupt file: Expected BOF record; found b'foo'zFile is not a zip file)rL )rV r r� r r rU r, r+ )r/ r r Z bad_stream�errorrX r� r r r �test_corrupt_bytes_raises� s z%TestReaders.test_corrupt_bytes_raisesc C s0 d| }t �|�}t �d| �}t�||� d S )Nz[https://raw.githubusercontent.com/pandas-dev/pandas/master/pandas/tests/io/data/excel/test1r= )r, r+ rb rc )r/ r �url� url_table� local_tabler r r �test_read_from_http_url s z#TestReaders.test_read_from_http_urlc C sf t d| d�� }|�d�jd| |d� W d Q R X d| }tj||d�}t�d| �}t�||� d S )Nr= r> z pandas-test)�Key�Bodyzs3://pandas-test/test1)Zstorage_options)rA �Bucket� put_objectr, r+ rb rc )r/ r � s3_resource�s3sorD r� r� r� r r r �test_read_from_s3_url s "z!TestReaders.test_read_from_s3_urlc C s� t d| d�� }|�d�jd| |d� W d Q R X dd l}|jf |�}|� d| ��}t�|�}W d Q R X t�d| �}t�||� d S )Nr= r> z pandas-test)r� r� r zs3://pandas-test/test1) rA r� r� �s3fsZ S3FileSystemr, r+ rb rc ) r/ r r� r� rD r� Zs3r� r� r r r �test_read_from_s3_object s " z$TestReaders.test_read_from_s3_objectc C s� t j�|ddd�d| �}t�|�}yt�d| �}W n> tk rt dd l}d�|�� ��� }t � d|� �� Y nX t � ||� d S ) Nr( r) r* r= zfile://localhost/r � z failing on ) �os�path�joinr, r+ r �platform�uname�stripr r� rb rc )r/ r r0 Z localtabler� r� r� Z platform_infor r r �test_read_from_file_url) s z#TestReaders.test_read_from_file_urlc C sP ddl m} d| }tj|ddd�}|d| �}tj|ddd�}t�||� d S )Nr )r r= rM )rO rP )�pathlibr r, r+ rb rc )r/ r r �str_pathrF �path_objr� r r r �test_read_from_pathlib_path; s z'TestReaders.test_read_from_pathlib_pathzpy.pathc C s\ ddl m} tj�d| �}tj|ddd�}|� �d| �}tj|ddd�}t�||� d S )Nr )�localr= rM )rO rP ) Zpy.pathr r� r� r� r, r+ rb rc )r/ r Z LocalPathr rF r r� r r r �test_read_from_py_localpathH s z'TestReaders.test_read_from_py_localpathc C sD t j�d| �}t|d�� }tj|ddd�}~|�� W d Q R X d S )Nr= r> rM r )rO rP )r� r� r� rA r, r+ �read)r/ r r rD r~ r r r � test_close_from_py_localpathW s z(TestReaders.test_close_from_py_localpathc C s� t jjd dkr&|j�tjjdd�� t� dt ddd�t dd d d �t d d dd�t dddd�t dddd�t dddd�t d d d d �t ddd d!�t d d"dd#�t d$d%d&d'�t d(d)d*�g i�}t jd+| d,d-�}t � ||� t jd.| d,d-�}t � ||� d S )/Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ ZTimer? r[ rN �- �8 i�� r� � �1 i@ � � �* i� � �9 �# i� � �) � i � � � � i�' r� i`� �5 i 5 � �% r i�� � � �6 Z times_1900rM )rO Z times_1904) r, r+ r] r^ r_ r r` ra r r� r rb rc )r/ r% r rF r� r r r �test_reader_secondsb s* z TestReaders.test_reader_secondsc C sJ t jjd dkr&|j�tjjdd�� t� ddgddgg�}d | }t d d t � d �d gddt � d�dgddt � d�dgddt � d�d gg|d�}t j|ddd gdd�}t � ||� ||_ddd d g|_t j|d dd gd �}t j ||dd � ||_t j|d!dd gdd gd"�}t j ||dd � ddd d g|_|�d#d$g�|_t j|d%dd gd �}t � ||� ttd��|_|�d&d'g�|_t j|d(dd gdd�}t � ||� |jd dgd d)��d&d'g�|_t j|d*ddd gd"�}t � ||� |�d&d'g�|_|�d#d$g�|_t j|d+dd gdd gd"�}t � ||� t j|d,dd gdd gdd-�}t � ||� d S ).Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ r� �barr} r� �testmultiindexr? g @z 2015-01-01Tr[ g @z 2015-01-02FrN g @z 2015-01-03r� g @z 2015-01-04)rK Z mi_columnr )rO �headerrP r� r� Zmi_index)rO rP )r\ Zboth)rO rP r �ilvl1�ilvl2Z mi_index_name�c1�c2Zmi_column_name)�levelZ name_with_intZ both_nameZboth_name_skiprows)rO rP r rS )r, r+ r] r^ r_ r r` ra r � from_productr r� rb rc rm rK Z set_namesr� rk Z set_levels)r/ r% r �mi�mi_filerF r� r r r �test_read_excel_multiindex� s` z&TestReaders.test_read_excel_multiindexzsheet_name,idx_lvl2Z both_name_blank_after_mi_namer� r} Zboth_name_multiple_blanksc C s� t jjd dkr&|j�tjjdd�� d| }tj ddgdd ggd d gd �}t d dt � d�dgddt � d�dgddt � d�dgddt � d�dgg|tj ddddg|fdd gd �d �}t j||d d gd d gd �}t �||� d S ) Nr r z;Sheets containing datetimes not supported by pyxlsb (GH4679)rZ r r� r r} r� r# r$ )ry r? g @z 2015-01-01Tr[ g @z 2015-01-02FrN g @z 2015-01-03r� g @z 2015-01-04r! r" )rK rm r )rO rP r )r, r+ r] r^ r_ r r` ra r r& r r� Z from_arraysrb rc ) r/ r% r rO Zidx_lvl2r( r' rF rE r r r �+test_read_excel_multiindex_blank_after_name� s* z7TestReaders.test_read_excel_multiindex_blank_after_namec C sT d| }t j|dddgd�}t�ddg�}tddd d ggd |d �}t�||� d S ) Nr �index_col_noner r? )rO r )rH rI )�key�valr[ rN r� )rK )r, r+ r r& r rb rc )r/ r r( rE Z exp_columnsrF r r r �&test_read_excel_multiindex_header_only s z2TestReaders.test_read_excel_multiindex_header_onlyc C s d| }t �d d d d d gdddddgddd d d gd d dddgdddddgdddddgg�}dd d d d g}td d!d"d#d$d%gd&d'd(d)d*d+ggd,d-d.d/d0d1gd,d-d.d/d0d1ggd d gd2�}td d!d"d#d$d%gd d3�}t|||d4�}tj|d5d,d6�}t�||� ||_ tj|d7d,d-gd6�}t�||� t �dddddgddd d d gd d dddgdddddgdddddgg�}dd d d d g}td!d"d#d$d%gd'd(d)d*d+ggd,d-d.d/d0gd,d-d.d/d0ggd d gd2�}td!d"d#d$d%gd d3�}t|||d4�}tj|d8d,d6�}t�||� ||_ tj|d9d,d-gd6�}tj||d:d;� d S )<NZtest_index_name_pre17ZR0C0ZR0C1ZR0C2ZR0C3ZR0C4ZR1C0ZR1C1ZR1C2ZR1C3ZR1C4ZR2C0ZR2C1ZR2C2ZR2C3ZR2C4ZR3C0ZR3C1ZR3C2ZR3C3ZR3C4ZR4C0ZR4C1ZR4C2ZR4C3ZR4C4ZC_l0_g0ZC_l0_g1ZC_l0_g2ZC_l0_g3ZC_l0_g4ZR0ZR_l0_g0ZR_l0_g1ZR_l0_g2ZR_l0_g3ZR_l0_g4ZR1ZR_l1_g0ZR_l1_g1ZR_l1_g2ZR_l1_g3ZR_l1_g4r r? r[ rN r� r� )rw rx ry )r� )rm rK Z single_names)rO rP Z multi_namesZsingle_no_namesZmulti_no_namesF)r\ ) r� �arrayr r r r, r+ rb rc rm ) r/ r r� r) rK r' �sirF r� r r r �test_excel_old_index_format sR z'TestReaders.test_excel_old_index_formatc C sB d}x8dD ]0}t jt|d�� tjd| |d� W d Q R X q W d S )Nz#Passing a bool to header is invalid)TF)rL r= )r )r rU rr r, r+ )r/ r rX �argr r r � test_read_excel_bool_header_argU s z+TestReaders.test_read_excel_bool_header_argc C s^ t jjd dkr&|j�tjjdd�� t jd| dddgd �}td d t � d �d gddt � d�dgddt � d�dgddt � d�d ggddddgd�}t � ||� t jd| dt � ddg�d �}t � ||� t jd| dd d � d �}t � ||� t jd| ddddddgd �}tddt � d�dgddt � d�dgddt � d�d ggddddgd�}t � ||� d S ) Nr r z3Sheets containing datetimes not supported by pyxlsb)rZ Z testskiprowsZ skiprows_listr r[ )rO rS r? g @z 2015-01-01Tg @z 2015-01-02FrN g @z 2015-01-03r� g @z 2015-01-04r} r� r� r� )rK c S s | dkS )N)r r[ r )r~ r r r r� ~ r� z6TestReaders.test_read_excel_skiprows.<locals>.<lambda>)rO rS ry )r, r+ r] r^ r_ r r` ra r r� rb rc r� r/ )r/ r% r r� rF r r r �test_read_excel_skiprows\ sD z$TestReaders.test_read_excel_skiprowsc C s@ d}t jd| |d�}t �d| �}|d |� }t�||� d S )Nr� r= )�nrows)r, r+ rb rc )r/ r �num_rows_to_pullr� rF r r r �test_read_excel_nrows� s z!TestReaders.test_read_excel_nrowsc C s@ t �d| �}t|�}|d }t jd| |d�}t�||� d S )Nr= r� )r5 )r, r+ rl rb rc )r/ r rF Znum_records_in_filer6 r� r r r �0test_read_excel_nrows_greater_than_nrows_in_file� s z<TestReaders.test_read_excel_nrows_greater_than_nrows_in_filec C s4 d}t jt|d�� tjd| dd� W d Q R X d S )Nz 'nrows' must be an integer >=0)rL r= r� )r5 )r rU rV r, r+ )r/ r rX r r r �+test_read_excel_nrows_non_integer_parameter� s z7TestReaders.test_read_excel_nrows_non_integer_parameterc C s� d| }t j|dddd�}tdddgdd d gd d �}d |j_t�||� t j|ddd�}tdd d gdddgd��}t�||� t j|ddd�}tdddgd d �}t�||� d S )NZ test_squeezeZ two_columnsr T)rO rP �squeezer[ rN r� r� r r� )r� r} )rO r: )r} r� Z one_columnr? ) r, r+ r rm r� rb Zassert_series_equalr rc )r/ r rD r� rF r r r �test_read_excel_squeeze� s z#TestReaders.test_read_excel_squeezec C s> t jtdd�� t�d| dd� W d Q R X t�d| � d S )NF)r� r= rM r )rb r� r� r, r+ )r/ r r r r �test_deprecated_kwargs� s z"TestReaders.test_deprecated_kwargsc C s^ d| }ddddg}t jddddgdd�}t||d d �}tj|d d d gd d�}t�||� d S )Nr )rI rI )r, r- )rN r� )rH rH )r? r[ )r r? )ry )r[ rN )rm rK r+ r r? )rO rP r )r � from_tuplesr r, r+ rb rc )r/ r � file_namer) �idxrF rE r r r �"test_no_header_with_list_index_col� s z.TestReaders.test_no_header_with_list_index_colc C s> d| }dt jddg}t|dgd�}t�|�}t�||� d S )NZone_col_blank_lineg �?r? r[ �numbers)rK )r� r� r r, r+ rb rc )r/ r r> r) rF rE r r r � test_one_col_noskip_blank_line� s z*TestReaders.test_one_col_noskip_blank_linec C sj d| }t �ddg�}tjtjgtjtjgddgddgg}t||d�}tj|d d dgd �}t�||� d S ) Nr )r} rH )r� rI r? rN r[ r� )rK Zmi_column_empty_rowsr )rO r ) r r= r� r� r r, r+ rb rc )r/ r r> rK r) rF rE r r r � test_multiheader_two_blank_lines� s $ z,TestReaders.test_multiheader_two_blank_linesc C s$ d| }t �|�}|jdks t�dS )z� Sheets can contain blank cells with no data. Some of our readers were including those cells, creating many empty rows and columns Ztrailing_blanks)rN rN N)r, r+ �shaperB )r/ r r> rE r r r �test_trailing_blanks� s z TestReaders.test_trailing_blanks)O�__name__� __module__� __qualname__r �fixturer3 rG rY rf rh r` � parametrizeri rn ro rp rq rs rz 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� rb �networkr� �tdZskip_if_not_us_localer� r� Zslowr� r � skip_if_noZcheck_file_leaksr r r r) r* r. r1 r3 r4 r7 r8 r9 r; r< r@ rB rC rE r r r r r' x s� 5@" R " & * !V % H7 r' c @ s e Zd Zejdd�dd� �Zdd� Zdd� Zej� d d dd g�d d � �Z dd� Z dd� Z ej� ddddgddgdddgddgg�dd� �Z dd� Zdd � Zd d � Zd d � Zd!d"� Zejjed k o�eed#�kd$d%�d&d'� �Zej� d(d)d*g�d+d,� �Zd-d.� Zd/d0� Zd1d2� Zd S )3�TestExcelFileReadT)r# c C s2 t tj|d�}|�|ddd�� |�td|� dS )zH Change directory and set engine for ExcelFile objects. )r r( r) r* r@ N)r r, r@ r- r. )r/ r r0 r1 r2 r r r r3 � s z#TestExcelFileRead.cd_and_set_enginec C s^ dddddd�}t �d| �� }|j}W d Q R X |d k r>|}n||dd � }||ksZt�d S )Nr r r r )r8 r9 r: r; r< r= r? )r, r@ r rB )r/ r r r1 rC r* rE rF r r r rG � s z"TestExcelFileRead.test_engine_usedc C s� t �d| ��}t j|dddgd�}W d Q R X tdgdgdgtjgdggd gd �}t�||� t �d| ��}t j|dd dgd�}W d Q R X ttjgdgtjgtjgdggd gd �}t�||� t �d | ��}t j|dddgd�}W d Q R X td gdgdgtjgdggd gd �}t�||� t �d | ��}t j|dd dgd�}W d Q R X ttjgdgtjgtjgdggd gd �}t�||� d S )NZtest4rM F�apple)rO �keep_default_na� na_valuesZNAr? �rabbitr� )rK T�test5z1.#QNANr� )r, r@ r+ r r� r� rb rc )r/ r r* r� rF r r r �test_excel_passes_na s0 " & " &z&TestExcelFileRead.test_excel_passes_na� na_filterNFc C s� i }|d k r||d< t �d| ��"}t j|fdddgd�|��}W d Q R X |dkrldgd gd gdgd gg}n tjgd gtjgtjgd gg}t|d gd �}t�||� d S )NrU rS rM TrO )rO rP rQ Fz1.#QNANr? r� rR r� )rK )r, r@ r+ r� r� r rb rc )r/ r rU r5 r* r� rF r r r �test_excel_passes_na_filter, s z-TestExcelFileRead.test_excel_passes_na_filterc C s\ |dkr |j �tjjdd�� t�d| ��*}tj|ddd�}tj|ddgdd�}W d Q R X tj ||d d � tj ||d d � t�d| ��&}|j ddd �}|j ddgdd �}W d Q R X tj ||d d � tj ||d d � t�d| ��}tj|dddd �}W d Q R X t� ||j d d� � t�d| ��}|j dddd�}W d Q R X t� ||j d d� � d S )Nz.xlsbz3Sheets containing datetimes not supported by pyxlsb)rZ r= r )rO rP r? )rO rS rP F)r\ )rP )rS rP )rO rP r� r� )rP r� ) r^ r_ r r` ra r, r@ r+ rb rc r7 r� )r/ r% r rW r* rd re rg r r r � test_excel_table_sheet_by_indexE s( z1TestExcelFileRead.test_excel_table_sheet_by_indexc C s� |dkr |j �tjjdd�� d}d}t�|| ��}|j|dd�}W d Q R X t�|| ��}|jd|d�}W d Q R X tj ||d d � tj ||d d � d S ) Nz.xlsbz3Sheets containing datetimes not supported by pyxlsb)rZ r= rM r )rO rP )rP rO F)r\ ) r^ r_ r r` ra r, r@ r7 rb rc ) r/ r% r rW r� rO r* Z df1_parseZ df2_parser r r r� d s z!TestExcelFileRead.test_sheet_namerO rN r r{ rM c C sH d}t jt|d��, t�d| ��}|j|d� W d Q R X W d Q R X d S )Nz?Worksheet index 3 is invalid|Worksheet named 'Sheet4' not found)rL r� )rO )r rU rV r, r@ r7 )r/ r rO rX r* r r r r� z s z+TestExcelFileRead.test_bad_sheetname_raisesc C sf d| }t j|dd|d�}t|d��,}t �|��}t j|ddd�}W d Q R X W d Q R X t�||� d S )Nr= rM r )rO rP r r> )rO rP )r, r+ rA r@ rb rc )r/ r r r� rF rD r; r� r r r r� � s $z(TestExcelFileRead.test_excel_read_bufferc C sP t d| d��.}t�|��}tj|dd|d� W d Q R X W d Q R X |jsLt�d S )Nr= r> rM r )rO rP r )rA r, r@ r+ �closedrB )r/ r r rD r8 r r r �test_reader_closes_file� s &z)TestExcelFileRead.test_reader_closes_filec C sJ d}t �d| ��.}tjt|d�� t j|dd� W d Q R X W d Q R X d S )Nz8Engine should not be specified when passing an ExcelFiler= )rL r� )r )r, r@ r rU rV r+ )r/ r rX Zxlr r r � test_conflicting_excel_engines� s z0TestExcelFileRead.test_conflicting_excel_enginesc C sR t jd| |d�}td| d��}|�� }W d Q R X t j||d�}t�||� d S )Nr= )r r> )r, r+ rA r rb rc )r/ r r rF rD r) r� r r r �test_excel_read_binary� s z(TestExcelFileRead.test_excel_read_binaryc C sF t d| d��}t�|�}W d Q R X tjd| |d�}t�||� d S )Nr= r> )r )rA r, r+ rb rc )r/ r r rD rE rF r r r �%test_excel_read_binary_via_read_excel� s z7TestExcelFileRead.test_excel_read_binary_via_read_excelr z xlrd no longer supports xlsx)rZ c C s. t dgdgd�}tjddd�}t�||� d S )Nu �ZColumn1)rK zhigh_surrogate.xlsxr )r )r r, r+ rb rc )r/ r rF r� r r r �test_excel_high_surrogate� s z+TestExcelFileRead.test_excel_high_surrogater� z df_empty.xlsxzdf_equals.xlsxc C s` t dgdd�}tjddgddgd�}td d gg||d d �}tj|d ddd gd�}t�||� d S )N�ZZI2)r� )rH rI )rH zB.1ZI11ZI12)ry r? rN r� )rm rK r� rM r )rO rP r )r r r= r r, r+ rb rc )r/ r r� r? �colsrF rE r r r �test_header_with_index_col� s z,TestExcelFileRead.test_header_with_index_colc C s� |dkr |j �tjjdd�� d| }t�|�� }tj|ddgd|d�}W d Q R X tj t� d�t� d �fgt� d�� � t� d �� � gd �}t g |d �}t �||� d S ) Nr z3Sheets containing datetimes not supported by pyxlsb)rZ Ztest_datetime_mir r? )r rP r z 02/29/2020z 03/01/2020)ry )rK )r^ r_ r r` ra r, r@ r+ r r= � to_datetimeZ to_pydatetimer rb rc ) r/ r% r r rD r* r� Zexpected_column_indexrF r r r � test_read_datetime_multiindex� s z/TestExcelFileRead.test_read_datetime_multiindexc C s> t jtdd��& t�d|� d�d�� W d Q R X W d Q R X d S )NzValue must be one of *)rL zio.excelz.reader�abc)r rU rV r, Zoption_context)r/ r r r r �test_engine_invalid_option� s z,TestExcelFileRead.test_engine_invalid_optionc C s� t f}|d krt�� n |dkr4dd l}t |jjf}t�d|� ���R}t|�� d� t� d��. yt j ||d� W n |k r� Y nX W d Q R X W d Q R X d S )Nr r ZcorruptF)r ) r r r� r Zbiffhr� rb Z ensure_cleanr � write_textr� r, r@ )r/ r% r r �errorsr �filer r r �test_corrupt_files_closed� s z+TestExcelFileRead.test_corrupt_files_closed)rF rG rH r rI r3 rG rT r` rJ rV rW r� r� r� rY rZ r[ r\ Zskipifr r r] r` rb rd rh r r r r rN � s* & * rN )/r r � functoolsr r� r� r � urllib.errorr �zipfiler �numpyr� r Z pandas.util._test_decorators�utilZ_test_decoratorsrL Zpandasr, r r r r Zpandas._testingZ_testingrb Zpandas.tests.io.excelr Zpandas.util.versionr r r rM r` �filterwarningsZ engine_paramsr� �boolr r rI r& r r r' rN r r r r �<module> sX z
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка