U
    ¾mœd)  ã                   @   s’   d dl Z d dlZd dlZd dlmZ d dlmZ d dl	m
Z
 e d¡ZdgZejed�dd„ ƒZd	d
„ Zdd„ Zej dddddg¡dd„ ƒZdS )é    N)Ú	ExcelFile)Úinspect_excel_formatÚxlrdz.xls)Úparamsc                 C   s   | j S )z�
    Valid extensions for reading Excel files with xlrd.

    Similar to read_ext, but excludes .ods, .xlsb, and for xlrd>2 .xlsx, .xlsm
    )Úparam)Úrequest© r   úX/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/io/excel/test_xlrd.pyÚread_ext_xlrd   s    r
   c           	   
   C   sv   d}d}|ddddƒ}t  |¡�@}t||d��}tj||dd	�}W 5 Q R X tj|||dd
�}W 5 Q R X t ||¡ d S )Nr   ZSheet1ÚioÚdataÚexcelz	test1.xls©Úenginer   )Ú
sheet_nameÚ	index_col)r   r   r   )r   Zopen_workbookr   ÚpdÚ
read_excelÚtmZassert_frame_equal)	r
   Údatapathr   r   ÚpthÚbookZxlÚresultÚexpectedr   r   r	   Útest_read_xlrd_book   s       ÿr   c              	   C   sF   ddl m} | ddddƒ}tj|dd�� tj|d	d
� W 5 Q R X d S )Nr   )Ú	XLRDErrorr   r   r   z
test1.xlsxzExcel xlsx file; not supported)Úmatchr   r   )Z
xlrd.biffhr   ÚpytestZraisesr   r   )r   r   Úpathr   r   r	   Útest_read_xlsx_fails(   s    r   Úfile_headers   	    s   	    s   	    s   ÐÏà¡±ác                 C   s   t  | ¡}t|ƒdkst‚d S )NZxls)r   ÚBytesIOr   ÚAssertionError)r    Úfr   r   r	   Útest_read_old_xls_files1   s    
r$   )r   r   Zpandasr   Zpandas._testingZ_testingr   Zpandas.io.excelr   Zpandas.io.excel._baser   Zimportorskipr   ÚextsZfixturer
   r   r   ÚmarkZparametrizer$   r   r   r   r	   Ú<module>   s(   


		üþ	