U
    ÎZjW/  ã                   @   s6  d dl mZ d dlZd dlZd dlmZmZ d dlm	Z	m
Z
mZ d dlmZ d dlmZmZmZmZ dd„ Zejdd	d
dddddddg
d�dd„ ƒZejdd„ ƒZej d¡dd„ ƒZej d¡dd„ ƒZej dddddd g¡d!d"„ ƒZejd#d$„ ƒZejjd%d&„ ƒZ ejj!ejjej d$¡G d'd(„ d(ƒƒƒƒZ"dS ))é    )ÚdedentN)ÚPyperclipExceptionÚPyperclipWindowsException)Ú	DataFrameÚ
get_optionÚread_clipboard)ÚCheckedCallÚ_stringifyTextÚclipboard_getÚclipboard_setc                 C   s(   i }|dkr||d< | dkr$| |d< |S )NÚdefaultÚexcelÚsep© )r   r   Úkwargsr   r   úa/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/pandas/tests/io/test_clipboard.pyÚbuild_kwargs   s    r   ÚdelimsÚutf8Úutf16ÚstringÚlongÚnonasciiÚcolwidthÚmixedÚfloatÚint)Úparamsc              	      s¤  | j }|dkr$tddgddgdœƒS |dkrBtdd	gd
dgdœƒS |dkr`tddgddgdœƒS |dkr‚tjddddd gd gd�S |dkr¶tdƒ}tj|d ddd„ ddd gd gd�S |dkrÔtd ¡ d ¡ dœƒS |d k�rtd!ƒd ‰ tjdd‡ fd"d„ddd gd gd�S |d#k�rHtt d$d%¡d& t dd'¡ tj	¡t
d(ƒd)œƒS |d*k�rrtjddd+d„ ddd gd gd�S |d,k�rœtjddd-d„ ddd gd gd�S t‚d S ).Nr   z"a,	"b|cu   d	efÂ´zhi'jzk''lm)ÚaÚbr   u   Âµasdu	   Î©Å“âˆ‘Â´u   Ã¸Ï€âˆ†ËšÂ¬u	   Å“âˆ‘Â´Â®r   u   ðŸ‘�ðŸ‘�ÚabcÚdefr   é   é   ÚsÚi)Ú
c_idx_typeÚ
r_idx_typeÚc_idx_namesÚr_idx_namesr   zdisplay.max_rowsé   c                  W   s   t j d¡S ©Né   ©ÚnpÚrandomÚrandint©Úargsr   r   r   Ú<lambda>E   ó    zdf.<locals>.<lambda>)Z
data_gen_fr&   r'   r(   r)   r   z
in Englishu   en espaÃ±ol)ÚenÚesr   zdisplay.max_colwidthc                     s   dˆ  S )NÚxr   r1   ©Z_cwr   r   r3   R   r4   r   ç      ð?g      @ç{®Gáz„?é   Zabcde)r   r   Úcr   c                 S   s   t | ƒd S )Nr:   )r   )Úrr<   r   r   r   r3   d   r4   r   c                  W   s   t j d¡S r+   r-   r1   r   r   r   r3   n   r4   )Úparamr   ÚtmZmakeCustomDataframer   Úsplitr.   ZarangeZastypeZint64ÚlistÚ
ValueError)ÚrequestZ	data_typeZmax_rowsr   r8   r   Údf#   sŒ    ÿ     ÿù	

ù	
ýÿ
ù	
ù
rD   c              	   c   s6   dd„ }|   ¡ �}|jd|dd� dV  W 5 Q R X dS )z<
    Mocks WinError to help with testing the clipboard.
    c                   S   s   dS )NzWindow Errorr   r   r   r   r   Ú_mock_win_error~   s    z$mock_ctypes.<locals>._mock_win_errorzctypes.WinErrorF)ZraisingN)ÚcontextÚsetattr)ÚmonkeypatchrE   Úmr   r   r   Úmock_ctypesx   s    
rJ   c              	   C   sN   dd„ }|   ddd„ ¡ d|j› d�}tjt|d�� t|ƒƒ  W 5 Q R X d	S )
z…
    Give CheckCall a function that returns a falsey value and
    mock get_errno so it returns false so an exception is raised.
    c                   S   s   dS ©NFr   r   r   r   r   Ú_return_falseŽ   s    z6test_checked_call_with_bad_call.<locals>._return_falseúpandas.io.clipboard.get_errnoc                   S   s   dS ©NTr   r   r   r   r   r3   ‘   r4   z1test_checked_call_with_bad_call.<locals>.<lambda>zError calling z \(Window Error\)©ÚmatchN)rG   Ú__name__ÚpytestÚraisesr   r   )rH   rL   Úmsgr   r   r   Útest_checked_call_with_bad_call‡   s
    rU   c                 C   s2   dd„ }|   ddd„ ¡ t|ƒ}|ƒ dks.t‚dS )zÆ
    Give CheckCall a function that returns a truthy value and
    mock get_errno so it returns true so an exception is not raised.
    The function should return the results from _return_true.
    c                   S   s   dS rN   r   r   r   r   r   Ú_return_true    s    z7test_checked_call_with_valid_call.<locals>._return_truerM   c                   S   s   dS rK   r   r   r   r   r   r3   £   r4   z3test_checked_call_with_valid_call.<locals>.<lambda>TN)rG   r   ÚAssertionError)rH   rV   Zchecked_callr   r   r   Ú!test_checked_call_with_valid_call˜   s    rX   ÚtextZString_testTr*   r9   y              ð?c              	   C   sf   t tttf}t| |ƒr0t| ƒ}|t | ƒksbt‚n2dt| ƒj› �}t	j
t|d�� t| ƒ W 5 Q R X d S )NzJonly str, int, float, and bool values can be copied to the clipboard, not rO   )Ústrr   r   ÚboolÚ
isinstancer	   rW   ÚtyperQ   rR   rS   r   )rY   Zvalid_typesÚresultrT   r   r   r   Útest_stringify_textª   s    
ÿr_   c                 #   sB   i ‰ ‡ ‡fdd„}‡ ‡fdd„}|   d|¡ |   d|¡ ˆ V  dS )aM  Fixture mocking clipboard IO.

    This mocks pandas.io.clipboard.clipboard_get and
    pandas.io.clipboard.clipboard_set.

    This uses a local dict for storing data. The dictionary
    key used is the test ID, available with ``request.node.name``.

    This returns the local dictionary, for direct manipulation by
    tests.
    c                    s   | ˆ ˆj j< d S ©N©ÚnodeÚname©Údata©Z
_mock_datarC   r   r   Ú	_mock_setÓ   s    z!mock_clipboard.<locals>._mock_setc                      s   ˆ ˆj j S r`   ra   r   rf   r   r   Ú	_mock_getÖ   s    z!mock_clipboard.<locals>._mock_getz!pandas.io.clipboard.clipboard_setz!pandas.io.clipboard.clipboard_getN)rG   )rH   rC   rg   rh   r   rf   r   Úmock_clipboardÃ   s    ri   c                 C   sF   dd l }|jj d¡ dt|  ¡ ƒks*t‚|jj ¡ }|dksBt‚d S )Nr   r    )Úpandas.io.clipboardÚioÚ	clipboardr   ÚsetÚvaluesrW   r
   )ri   Úpandasr^   r   r   r   Útest_mock_clipboardß   s
    rp   c                
   @   st  e Zd Zd>dd„Zdd„ Zej dddd	g¡d
d„ ƒZdd„ Z	dd„ Z
dd„ Zej ddddg¡ej ddddg¡dd„ ƒƒZej dddg¡ej ddg¡dd„ ƒƒZdd„ Zdd„ Zej dd d d!d"d#g¡d$d$d%gd&d'd&gdddggfd d(d)d*d+g¡d$d$d%gd&d'd&ggfg¡d,d-„ ƒZd.d/„ Zej d0d1d2d3g¡d4d5„ ƒZej d6d7d8d9g¡ejjd:edd;�d<d=„ ƒƒZdS )?ÚTestClipboardNc                 C   s2   |j |||d� t|pdd|d�}t ||¡ d S )N)r   r   Úencodingú	r   )r   Z	index_colrr   )Úto_clipboardr   r?   Úassert_frame_equal)Úselfre   r   r   rr   r^   r   r   r   Úcheck_round_trip_frameí   s    z$TestClipboard.check_round_trip_framec                 C   s   |   |¡ d S r`   ©rw   ©rv   rD   r   r   r   Útest_round_trip_frameó   s    z#TestClipboard.test_round_trip_framer   rs   ú,ú|c                 C   s   | j ||d� d S )N©r   rx   )rv   rD   r   r   r   r   Útest_round_trip_frame_sep÷   s    z'TestClipboard.test_round_trip_frame_sepc                 C   s<   |j dd d� tƒ }| ¡ | ¡ ks(t‚|j|jks8t‚d S )NF©r   r   )rt   r   Ú	to_stringrW   Úshape)rv   rD   r^   r   r   r   Útest_round_trip_frame_stringü   s    z*TestClipboard.test_round_trip_frame_stringc              	   C   s.   t jtddd�� |jddd� W 5 Q R X d S )NzAto_clipboard in excel mode requires a single character separator.F)rP   Zcheck_stacklevelTz\tr   )r?   Úassert_produces_warningÚUserWarningrt   ry   r   r   r   Útest_excel_sep_warning  s    ýz$TestClipboard.test_excel_sep_warningc              	   C   s&   t  ¡ � |jddd� W 5 Q R X d S )NFrs   r   )r?   rƒ   rt   ry   r   r   r   Útest_copy_delim_warning  s    
z%TestClipboard.test_copy_delim_warningr   r   Tc                 C   s6   t ||ƒ}|jf |Ž ||jj |jdd�ks2t‚d S )Nrs   r}   )r   rt   rb   rc   Zto_csvrW   )rv   r   r   rD   rC   ri   r   r   r   r   Ú test_clipboard_copy_tabs_default  s    
z.TestClipboard.test_clipboard_copy_tabs_defaultFc                 C   sH   t ||ƒ}|jf |Ž tdd�}| ¡ | ¡ ks4t‚|j|jksDt‚d S )Nz\s+r}   )r   rt   r   r€   rW   r�   )rv   r   r   rD   r   r^   r   r   r   Útest_clipboard_copy_strings  s
    

z)TestClipboard.test_clipboard_copy_stringsc                 C   s”   ddi}t d ¡ ƒ}|||jj< tf |Ž}|jd d dks@t‚t d ¡ ƒ}|||jj< tf |Ž}t d ¡ ƒ}|||jj< tf |Ž}t ||¡ d S )NZengineÚpythonz^
            John James	Charlie Mingus
            1	2
            4	Harry Carney
            r*   zHarry Carneyz@
            a	 b
            1  2
            3  4
            z@
            a  b
            1  2
            3  4
            )	r   Ústriprb   rc   r   ZilocrW   r?   ru   )rv   rC   ri   Zclip_kwargsrY   rD   ÚresÚexpr   r   r   Útest_read_clipboard_infer_excel$  s$    ÿ
ÿ
ÿ
z-TestClipboard.test_read_clipboard_infer_excelc                 C   sB   d}|||j j< tƒ }tdd dgdddgdœd�}t ||¡ d S )	Nzcol1	col2
1	red
	blue
2	greenr*   r,   ÚredÚblueÚgreen©Zcol1Zcol2rd   ©rb   rc   r   r   r?   ru   )rv   rC   ri   rY   rD   Údf_expectedr   r   r   Útest_infer_excel_with_nullsL  s    ÿz)TestClipboard.test_infer_excel_with_nullsÚ
multiindexÚ
z			col1	col2zA	0	True	1	redzA	1	True		bluezB	0	False	2	greenÚAÚBr   r*   z		col1	col2z	A	0	1	redz	A	1		bluezB	0	2	greenc                 C   sH   |d ||j j< tƒ }tdd dgdddgdœ|d d�}t ||¡ d S )	Nr   r*   r,   rŽ   r�   r�   r‘   )re   Úindexr’   )rv   rC   ri   r•   rD   r“   r   r   r   Ú test_infer_excel_with_multiindexY  s    þz.TestClipboard.test_infer_excel_with_multiindexc              	   C   sR   d}t jt|d�� |jdd� W 5 Q R X t jt|d�� tdd� W 5 Q R X d S )Nz&clipboard only supports utf-8 encodingrO   Úascii©rr   )rR   rS   rB   rt   ÚNotImplementedErrorr   )rv   rD   rT   r   r   r   Útest_invalid_encoding|  s
    z#TestClipboard.test_invalid_encodingÚenczUTF-8zutf-8r   c                 C   s   | j ||d� d S )Nrœ   rx   )rv   rŸ   rD   r   r   r   Útest_round_trip_valid_encodings„  s    z-TestClipboard.test_round_trip_valid_encodingsre   u   ðŸ‘�...u   Î©Å“âˆ‘Â´...zabcd...z4Flaky test in multi-process CI environment: GH 44584)ÚreasonrS   Ústrictc                 C   s   t |ƒ |tƒ kst‚d S r`   )r   r
   rW   )rv   re   r   r   r   Útest_raw_roundtripˆ  s    z TestClipboard.test_raw_roundtrip)NNN)rQ   Ú
__module__Ú__qualname__rw   rz   rR   ÚmarkÚparametrizer~   r‚   r…   r†   r‡   rˆ   r�   r”   Újoinrš   rž   r    ZxfailrW   r£   r   r   r   r   rq   é   sX   

	(üÿ÷
ÿüôþ

ýrq   )#Útextwrapr   Únumpyr.   rR   Zpandas.errorsr   r   ro   r   r   r   Zpandas._testingZ_testingr?   rj   r   r	   r
   r   r   ZfixturerD   rJ   r¦   ZusefixturesrU   rX   r§   r_   ri   rl   rp   Z
single_cpurq   r   r   r   r   Ú<module>   sZ   	öÿ
G




ûþ



	
