a
    j=icW/                     @   s   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g dd	d
d Zejdd Zejddd Zejddd Zejdg d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   o/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/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rtjddddd gd gdS |dkrtd}tj|d ddd ddd gd gdS |dkrt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rt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      @{Gz?   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astypeint64list
ValueError)request	data_typeZmax_rowsr   r8   r   df#   s    	

	

	

rF   c                 c   sJ   dd }|   &}|jd|dd dV  W d   n1 s<0    Y  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)raisingN)contextsetattr)monkeypatchrG   mr   r   r   mock_ctypesx   s    
rM   c                 C   sb   dd }|  ddd  d|j d}tjt|d t|  W d	   n1 sT0    Y  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)rJ   __name__pytestraisesr   r   )rK   rO   msgr   r   r   test_checked_call_with_bad_call   s
    rX   c                 C   s2   dd }|  ddd  t|}| du s.J 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 rQ   r   r   r   r   r   _return_true   s    z7test_checked_call_with_valid_call.<locals>._return_truerP   c                   S   s   dS rN   r   r   r   r   r   r3      r4   z3test_checked_call_with_valid_call.<locals>.<lambda>TN)rJ   r   )rK   rY   Zchecked_callr   r   r   !test_checked_call_with_valid_call   s    rZ   text)ZString_testTr*   r9   y              ?c                 C   sz   t tttf}t| |r0t| }|t | ksvJ nFdt| j }tj	t
|d t|  W d    n1 sl0    Y  d S )NzJonly str, int, float, and bool values can be copied to the clipboard, not rR   )strr   r   bool
isinstancer	   typerT   rU   rV   r   )r[   Zvalid_typesresultrW   r   r   r   test_stringify_text   s    
ra   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_datarD   r   r   	_mock_set   s    z!mock_clipboard.<locals>._mock_setc                      s    j j S rb   rc   r   rh   r   r   	_mock_get   s    z!mock_clipboard.<locals>._mock_getz!pandas.io.clipboard.clipboard_setz!pandas.io.clipboard.clipboard_getN)rJ   )rK   rD   ri   rj   r   rh   r   mock_clipboard   s    rk   c                 C   sF   dd l }|jjd dt|  v s*J |jj }|dksBJ d S )Nr   r    )pandas.io.clipboardio	clipboardr   setvaluesr
   )rk   pandasr`   r   r   r   test_mock_clipboard   s
    rr   c                   @   sX  e Zd Zd2ddZdd Zejdg ddd	 Zd
d Z	dd Z
dd Zejdg dejdg d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g dg d g d!g d"gfdg d#g d g d!gfgd$d% Zd&d' Zejd(g d)d*d+ Zejd,g d-ejjd.edd/d0d1 ZdS )3TestClipboardNc                 C   s2   |j |||d t|pdd|d}t|| d S )N)r   r   encoding	r   )r   Z	index_colrt   )to_clipboardr   r?   assert_frame_equal)selfrg   r   r   rt   r`   r   r   r   check_round_trip_frame   s    z$TestClipboard.check_round_trip_framec                 C   s   |  | d S rb   ry   rx   rF   r   r   r   test_round_trip_frame   s    z#TestClipboard.test_round_trip_framer   )ru   ,|c                 C   s   | j ||d d S )Nr   rz   )rx   rF   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(J |j|jks8J d S )NFr   r   )rv   r   	to_stringshape)rx   rF   r`   r   r   r   test_round_trip_frame_string   s    z*TestClipboard.test_round_trip_frame_stringc                 C   sB   t jtddd |jddd W d    n1 s40    Y  d S )NzAto_clipboard in excel mode requires a single character separator.F)rS   Zcheck_stacklevelTz\tr   )r?   assert_produces_warningUserWarningrv   r{   r   r   r   test_excel_sep_warning  s    z$TestClipboard.test_excel_sep_warningc                 C   s:   t   |jddd W d    n1 s,0    Y  d S )NFru   r   )r?   r   rv   r{   r   r   r   test_copy_delim_warning  s    
z%TestClipboard.test_copy_delim_warning)ru   Nr   r   )TNr   c                 C   s:   t ||}|jf i | ||jj |jddks6J d S )Nru   r   )r   rv   rd   re   Zto_csv)rx   r   r   rF   rD   rk   r   r   r   r    test_clipboard_copy_tabs_default  s    
z.TestClipboard.test_clipboard_copy_tabs_defaultr   Fc                 C   sL   t ||}|jf i | tdd}| | ks8J |j|jksHJ d S )Nz\s+r   )r   rv   r   r   r   )rx   r   r   rF   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 i |}|jd d dksDJ t d }|||jj< tf i |}t d }|||jj< tf i |}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   striprd   re   r   Zilocr?   rw   )rx   rD   rk   Zclip_kwargsr[   rF   resexpr   r   r   test_read_clipboard_infer_excel$  s$    z-TestClipboard.test_read_clipboard_infer_excelc                 C   s>   d}|||j j< t }tg dg ddd}t|| d S )Nzcol1	col2
1	red
	blue
2	greenr*   Nr,   redbluegreenZcol1Zcol2rf   rd   re   r   r   r?   rw   )rx   rD   rk   r[   rF   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)Ar   B)r   r*   r   )TTF)z		col1	col2z	A	0	1	redz	A	1		bluezB	0	2	greenc                 C   sD   |d ||j j< t }tg dg dd|d d}t|| d S )Nr   r   r   r   r*   )rg   indexr   )rx   rD   rk   r   rF   r   r   r   r    test_infer_excel_with_multiindexY  s    z.TestClipboard.test_infer_excel_with_multiindexc                 C   sz   d}t jt|d |jdd W d    n1 s40    Y  t jt|d tdd W d    n1 sl0    Y  d S )Nz&clipboard only supports utf-8 encodingrR   asciirt   )rU   rV   rC   rv   NotImplementedErrorr   )rx   rF   rW   r   r   r   test_invalid_encoding|  s
    *z#TestClipboard.test_invalid_encodingenc)zUTF-8zutf-8r   c                 C   s   | j ||d d S )Nr   rz   )rx   r   rF   r   r   r   test_round_trip_valid_encodings  s    z-TestClipboard.test_round_trip_valid_encodingsrg   )u   👍...u   Ωœ∑´...zabcd...z4Flaky test in multi-process CI environment: GH 44584)reasonrV   strictc                 C   s   t | |t ksJ d S rb   )r   r
   )rx   rg   r   r   r   test_raw_roundtrip  s    z TestClipboard.test_raw_roundtrip)NNN)rT   
__module____qualname__ry   r|   rU   markparametrizer   r   r   r   r   r   r   r   joinr   r   r   ZxfailAssertionErrorr   r   r   r   r   rs      sP   

	(

rs   )#textwrapr   numpyr.   rU   Zpandas.errorsr   r   rq   r   r   r   Zpandas._testingZ_testingr?   rl   r   r	   r
   r   r   ZfixturerF   rM   r   ZusefixturesrX   rZ   r   ra   rk   rn   rr   Z
single_cpurs   r   r   r   r   <module>   s<   	
G








	
