a
    Pf
                     @   s   d dl Z d dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
 edZe rXejjZejdd Zdd	 Zd
d Zejdddgdd Zdd ZdS )    N)is_platform_windows)	DataFrame)ExcelWriter
xlsxwriterc                   C   s   dS )Nz.xlsx r   r   r   ^/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/io/excel/test_xlsxwriter.pyext   s    r   c              
   C   sv  t d}t| J}tddgddgd}t|L}|| d}|j}| d }|	d|i}|
dd | W d    n1 s0    Y  t||<}	z|	d }
W n ty   |	jdd	}
Y n0 W d    n1 s0    Y  z|
d
 }W n ty   |
d
}Y n0 z
|j}W n tyB   |jjj}Y n0 ||ksRJ W d    n1 sh0    Y  d S )Nopenpyxli@ )ABz#,##0r   
num_formatzB:BZSheet1)nameZB2)pytestimportorskiptmensure_cleanr   r   Zto_excelbookZ
worksheetsZ
add_formatZ
set_column
contextlibclosingZload_workbook	TypeErrorZget_sheet_by_namecellZnumber_formatAttributeErrorstyleZ_format_code)r   r	   pathframewriterr   Zwrite_workbookZwrite_worksheetZ
col_formatZread_workbookZread_worksheetr   Zread_num_formatr   r   r   test_column_format   s0    


,0
r   c              	   C   sn   d}t | L}tjt|d t|ddd W d    n1 sB0    Y  W d    n1 s`0    Y  d S )Nz-Append mode is not supported with xlsxwriter!)matchr   a)enginemode)r   r   r   Zraises
ValueErrorr   )r   msgfr   r   r   test_write_append_mode_raises?   s    r$   nan_inf_to_errorsTFc              	   C   sx   dd|ii}t | N}t|d|d }|jj|ks8J W d    n1 sL0    Y  W d    n1 sj0    Y  d S )Noptionsr%   r   )r   engine_kwargs)r   r   r   r   r%   )r   r%   r'   r#   r   r   r   r   test_engine_kwargsG   s    r(   c              	   C   s   t | h}t|dd<}|ji ks(J |jd}|jd|iksFJ W d    n1 sZ0    Y  W d    n1 sx0    Y  d S )Nr   )r   Z	test_name)r   r   r   Zsheetsr   Zadd_worksheet)r   r#   r   Zsheetr   r   r   test_book_and_sheets_consistentP   s
    r)   )r   r   Zpandas.compatr   Zpandasr   Zpandas._testingZ_testingr   Zpandas.io.excelr   r   r   markZ
single_cpuZ
pytestmarkZfixturer   r   r$   Zparametrizer(   r)   r   r   r   r   <module>   s   

(
