U
|
ý°dú; ã @ sz d dl Z d dlZd dlmZ d dlZd dlZd dlmZm Z m
|
Z
|
d dl m Z m Z d dlZd dlmZmZmZmZmZ d dlmZ d dlmZmZ d dlm Z m Z m Z m Z dd Z ej!d
|
d d d ddddddg
|
ddd Z"ej!dd Z#ej$ %d¡dd Z&ej$ %d¡dd Z'ej$ (d d d d d!d"g¡d#d$ Z)ej!d%d& Z*ej$j+d'd( Z,ej$j-ej$j+ej$ %d&¡G d)d* d*Z.dS )+é N)Údedent)Úis_ci_environmentÚis_platform_linuxÚis_platform_mac)ÚPyperclipExceptionÚPyperclipWindowsException)ÚNAÚ DataFrameÚSeriesÚ
|
get_optionÚread_clipboard)ÚArrowStringArrayÚ StringArray)Ú CheckedCallÚ_stringifyTextÚ clipboard_getÚ clipboard_setc C s( i }|dkr||d< | dkr$| |d< |S )NÚdefaultÚexcelÚsep© )r r Úkwargsr r úUd:\z\workplace\vscode\pyvenv\venv\Lib\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 }|dkr$tddgddgdS |dkrBtdd gd
|
d gdS |d kr`td d gddgdS |dkrtjddddd gd gdS |dkr¶td}tj|d ddd ddd gd gdS |d krÔtd ¡ d ¡ d S |d krtd!d tjdd fd"dddd gd gdS |d#krHtt d$d%¡d& t dd'¡ tj ¡t
|
d(d)S |d*krrtjddd+d ddd gd gdS |d,krtjddd-d ddd gd gdS t d S ).Nr z"a, "b|cu d ef´zhi'jzk''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 zdisplay.max_rowsé c W s t j d¡S ©Né ©ÚnpÚrandomÚrandint©Úargsr r r Ú<lambda>S ó zdf.<locals>.<lambda>)Z
|
data_gen_fr- r. r/ r0 r z
|
in Englishu en español)ÚenÚesr zdisplay.max_colwidthc s d S )NÚxr r8 ©Z_cwr r r: ` r; r! ç ð?g @ç{®Gáz?é Zabcde)r% r& Úcr" c S s t | d S )NrA )r" )ÚrrC r r r r: r r; r# c W s t j d¡S r2 r4 r8 r r r r: | r; ) Úparamr ÚtmZmakeCustomDataframer Úsplitr5 ZarangeZastypeÚint64ÚlistÚ
|
ValueError)ÚrequestZ data_typeZmax_rowsr r? r Údf1 s ÿ ÿù
|
|
ù
|
ýÿ
|
ù
|
ù
|
rL c c s6 dd }| ¡ }|jd|dd dV W 5 Q R X dS )z<
|
Mocks WinError to help with testing the clipboard.
|
c S s dS )Nz Window Errorr r r r r Ú_mock_win_error s z$mock_ctypes.<locals>._mock_win_errorzctypes.WinErrorF)ZraisingN)ÚcontextÚsetattr)Ú monkeypatchrM Úmr r r Ú mock_ctypes s
|
rR c C sN dd }| ddd ¡ d|j d}tjt|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 dS ©NFr r r r r Ú _return_false s z6test_checked_call_with_bad_call.<locals>._return_falseú pandas.io.clipboard.get_errnoc S s dS ©NTr r r r r r: r; z1test_checked_call_with_bad_call.<locals>.<lambda>zError calling z \(Window Error\)©ÚmatchN)rO Ú__name__ÚpytestÚraisesr r )rP rT Úmsgr r r Ú test_checked_call_with_bad_call s
|
r] c C s2 dd }| ddd ¡ t|}| dks.tdS )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 dS rV r r r r r Ú _return_true® s z7test_checked_call_with_valid_call.<locals>._return_truerU c S s dS rS r r r r r r: ± r; z3test_checked_call_with_valid_call.<locals>.<lambda>TN)rO r ÚAssertionError)rP r^ Z checked_callr r r Ú!test_checked_call_with_valid_call¦ s r` ÚtextZ String_testTr1 r@ y ð?c C sf t tttf}t| |r0t| }|t | ksbtn2dt| 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 rW ) Ústrr# r" ÚboolÚ
|
isinstancer r_ ÚtyperY rZ r[ r )ra Z valid_typesÚresultr\ r r r Útest_stringify_text¸ s
|
ÿrg c # sB i fdd} fdd}| d|¡ | d|¡ V dS )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_datarK r r Ú _mock_setá s z!mock_clipboard.<locals>._mock_setc s j j S rh ri r rn r r Ú _mock_getä s z!mock_clipboard.<locals>._mock_getz!pandas.io.clipboard.clipboard_setz!pandas.io.clipboard.clipboard_getN)rO )rP rK ro rp r rn r Úmock_clipboardÑ s rq c C sF dd l }|jj d¡ dt| ¡ ks*t|jj ¡ }|dksBtd S )Nr r' )Úpandas.io.clipboardÚioÚ clipboardr ÚsetÚvaluesr_ r )rq Úpandasrf r r r Útest_mock_clipboardí s
|
rx c
|
@ s´ e Zd ZdFddZdd Zej dddd g¡d
|
d Zd d Z dd Z
|
dd Z ej ddddg¡ej ddddg¡dd Z ej dddg¡ej ddg¡dd Z dd Zd d Zej d d d d!d"d#g¡d$d$d%gd&d'd&gdddggfd d(d)d*d+g¡d$d$d%gd&d'd&ggfg¡d,d- Zd.d/ Zej d0d1d2d3g¡d4d5 Zejjej d6d7d8d9g¡ejjej d:¡dkoxe d;dd<d=d> Zej d?d@dAg¡dBdC ZdDdE Z dS )GÚ TestClipboardNc C s2 |j |||d t|pdd|d}t ||¡ d S )N)r r Úencodingú r )r Z index_colrz )Ú to_clipboardr rF Úassert_frame_equal)Úselfrm r r rz rf r r r Úcheck_round_trip_frameû s z$TestClipboard.check_round_trip_framec C s | |¡ d S rh ©r ©r~ rL r r r Útest_round_trip_frame s z#TestClipboard.test_round_trip_framer r{ ú,ú|c C s | j ||d d S )N©r r )r~ rL r r r r Útest_round_trip_frame_sep s z'TestClipboard.test_round_trip_frame_sepc C s< |j dd d t }| ¡ | ¡ ks(t|j|jks8td S )NF©r r )r| r Ú to_stringr_ Úshape)r~ rL rf r r r Ú test_round_trip_frame_string
|
s z*TestClipboard.test_round_trip_frame_stringc C s. t jtddd |jddd W 5 Q R X d S )NzAto_clipboard in excel mode requires a single character separator.F)rX Zcheck_stacklevelTz\tr )rF Úassert_produces_warningÚ UserWarningr| r r r r Útest_excel_sep_warning s ýz$TestClipboard.test_excel_sep_warningc C s& t ¡ |jddd W 5 Q R X d S )NFr{ r )rF r r| r r r r Útest_copy_delim_warning s
|
z%TestClipboard.test_copy_delim_warningr r Tc C s6 t ||}|jf | ||jj |jddks2td S )Nr{ r
)r r| rj rk Zto_csvr_ )r~ r r rL rK rq r r r r Ú test_clipboard_copy_tabs_default! s
|
z.TestClipboard.test_clipboard_copy_tabs_defaultFc C sH t ||}|jf | tdd}| ¡ | ¡ ks4t|j|jksDtd S )Nz\s+r
)r r| r r r_ r )r~ r r rL r rf r r r Útest_clipboard_copy_strings) s
|
|
|
z)TestClipboard.test_clipboard_copy_stringsc C s ddi}t d ¡ }|||jj< tf |}|jd d dks@tt d ¡ }|||jj< tf |}t d ¡ }|||jj< tf |}t ||¡ d S )NÚengineÚpythonz^
|
John James Charlie Mingus
|
1 2
|
4 Harry Carney
|
r1 z Harry Carneyz@
|
a b
|
1 2
|
3 4
|
z@
|
a b
|
1 2
|
3 4
|
) r Ústriprj rk r Zilocr_ rF r} )r~ rK rq Z clip_kwargsra rL ÚresÚexpr r r Ú test_read_clipboard_infer_excel2 s$ ÿ
|
ÿ
|
ÿ
|
z-TestClipboard.test_read_clipboard_infer_excelc C sB d}|||j j< t }tdd dgdddgdd}t ||¡ d S ) Nz col1 col2
|
1 red
|
blue
|
2 greenr1 r3 ÚredÚblueÚgreen©Zcol1Zcol2rl ©rj rk r r rF r} )r~ rK rq ra rL Ú df_expectedr r r Útest_infer_excel_with_nullsZ s ÿz)TestClipboard.test_infer_excel_with_nullsÚ
|
multiindexÚ
|
z col1 col2zA 0 True 1 redzA 1 True bluezB 0 False 2 greenÚAÚBr r1 z col1 col2z A 0 1 redz A 1 bluez B 0 2 greenc C sH |d ||j j< t }tdd dgdddgd|d d}t ||¡ d S ) Nr r1 r3 r r r r )rm Úindexr )r~ rK rq r rL r r r r Ú test_infer_excel_with_multiindexg s þz.TestClipboard.test_infer_excel_with_multiindexc C sR d}t jt|d |jdd W 5 Q R X t jt|d tdd W 5 Q R X d S )Nz&clipboard only supports utf-8 encodingrW Úascii©rz )rZ r[ rJ r| ÚNotImplementedErrorr )r~ rL r\ r r r Útest_invalid_encoding s
|
z#TestClipboard.test_invalid_encodingÚenczUTF-8zutf-8r c C s | j ||d d S )Nr¥ r )r~ r¨ rL r r r Ú test_round_trip_valid_encodings s z-TestClipboard.test_round_trip_valid_encodingsrm u ð...u ΩÅâ´...zabcd...ZDISPLAYzBCannot be runed if a headless system is not put in place with Xvfb)ÚreasonÚstrictc C s: t | |t kstt r6t r6tjdddgdd d S )NZxselz--deletez --clipboardT)Úcheck)r r r_ r r Ú
|
subprocessÚrun)r~ rm r r r Útest_raw_roundtrip s z TestClipboard.test_raw_roundtripr rC r c
|
sv |dks|dkrt d¡|dkrTttjddgtjd}ttjdtgtjd}n$t ddg¡}t dd g¡}d}|||jj < t
|
d|¡ t d||d } W 5 Q R X t |td
|
d gd dtd dgdd|td tgd dtd tgddtttgd dtddgddtdtgddd |dkrfddlm t fddjD d d g¡d< t | ¡ d S )NZpyarrowr r> Úy)Zdtypez@a,b,c,d,e,f,g,h,i
|
x,1,4.0,x,2,4.0,,True,False
|
y,2,5.0,,,,,False,zmode.string_storager )r Ú dtype_backendr r1 r3 ZInt64g @g @ZFloat64TFÚboolean) r% r& rC ÚdÚeÚfÚgÚhr, r )ÚArrowExtensionArrayc s$ i | ] }| j | d dqS )T)Z from_pandas)Úarray)Ú.0Úcol©r¸ ÚexpectedÚpar r Ú
|
<dictcomp>Î s ÿ zCTestClipboard.test_read_clipboard_dtype_backend.<locals>.<dictcomp>r¶ )rZ Z importorskipr r5 r¹ Zobject_r r rj rk ÚpdZoption_contextr r r
|
Z pandas.arraysr¸ ÚcolumnsrF r} )
|
r~ rK rq Zstring_storager± r Z string_arrayZstring_array_nara rf r r¼ r Ú!test_read_clipboard_dtype_backend¥ s@
|
÷ÿ
|
þÿz/TestClipboard.test_read_clipboard_dtype_backendc C s, d}t jt|d tdd W 5 Q R X d S )NzPdtype_backend numpy is invalid, only 'numpy_nullable' and 'pyarrow' are allowed.rW Únumpy)r± )rZ r[ rJ r )r~ r\ r r r Útest_invalid_dtype_backend× s ÿz(TestClipboard.test_invalid_dtype_backend)NNN) rY Ú
|
__module__Ú __qualname__r r rZ ÚmarkÚ parametrizer r r r r r r r Újoinr£ r§ r© Ú
|
single_cpuZxfailÚosÚenvironÚgetr r¯ r rÄ r r r r ry ÷ s`
|
|
( üÿ÷
|
ÿüôþ
|
|
ý
|
1ry )/rË r Útextwrapr rà r5 rZ Z pandas.compatr r r Z pandas.errorsr r rw rÀ r r r
|
r r Zpandas._testingZ_testingrF Zpandas.core.arraysr r rr r r r r r ZfixturerL rR rÇ Z usefixturesr] r` rÈ rg rq rt rx rÊ ry r r r r Ú<module> sd öÿ
|
G
|
|
|
|
|
ûþ
|
|
|
|
|
|
