U sVcW/ @s6ddlmZddlZddlZddlmZmZddlm Z m Z m Z ddl m ZddlmZmZmZmZddZejdd d d d d ddddg dddZejddZejdddZejdddZejdddddd gd!d"Zejd#d$Zejjd%d&Z ejj!ejjejd$Gd'd(d(Z"dS)))dedentN)PyperclipExceptionPyperclipWindowsException) DataFrame get_optionread_clipboard) CheckedCall_stringifyText clipboard_get clipboard_setcCs(i}|dkr||d<|dkr$||d<|S)Ndefaultexcelsep)rr kwargsrrB/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/io/test_clipboard.py build_kwargss rdelimsutf8utf16stringlongnonasciicolwidthmixedfloatint)paramsc s|j}|dkr$tddgddgdS|dkrBtdd gd d gdS|d kr`td d gddgdS|dkrtjdddddgdgdS|dkrtd}tj|dddddddgdgdS|dkrtdddS|d krtd!dtjddfd"ddddgdgdS|d#krHttd$d%d&tdd'tj t d(d)S|d*krrtjddd+ddddgdgdS|d,krtjddd-ddddgdgdSt dS).Nrz"a, "b|cud ef´zhi'jzk''lm)abruµasdu Ωœ∑´u øπ∆˚¬u œ∑´®ru👍👍abcdefrsi) c_idx_type r_idx_type c_idx_names r_idx_namesrzdisplay.max_rowscWs tjdSNnprandomrandintargsrrrEzdf..)Z data_gen_fr&r'r(r)rz in Englishu en español)enesrzdisplay.max_colwidthcsdS)Nxrr1Z_cwrrr3Rr4r?g@{Gz?Zabcde)rrcrcSs t|dS)Nr:)r)rr<rrrr3dr4rcWs tjdSr+r-r1rrrr3nr4) paramrtmZmakeCustomDataframersplitr.ZarangeZastypeZint64list ValueError)requestZ data_typeZmax_rowsrr8rdf#s        rDc cs6dd}|}|jd|dddVW5QRXdS)z< Mocks WinError to help with testing the clipboard. cSsdS)Nz Window Errorrrrrr_mock_win_error~sz$mock_ctypes.._mock_win_errorzctypes.WinErrorF)ZraisingN)contextsetattr) monkeypatchrEmrrr mock_ctypesxs rJc CsNdd}|dddd|jd}tjt|dt|W5QRXd S) z Give CheckCall a function that returns a falsey value and mock get_errno so it returns false so an exception is raised. cSsdSNFrrrrr _return_falsesz6test_checked_call_with_bad_call.._return_falsepandas.io.clipboard.get_errnocSsdSNTrrrrrr3r4z1test_checked_call_with_bad_call..zError calling z \(Window Error\)matchN)rG__name__pytestraisesrr)rHrLmsgrrrtest_checked_call_with_bad_calls rUcCs2dd}|dddt|}|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. cSsdSrNrrrrr _return_truesz7test_checked_call_with_valid_call.._return_truerMcSsdSrKrrrrrr3r4z3test_checked_call_with_valid_call..TN)rGrAssertionError)rHrVZ checked_callrrr!test_checked_call_with_valid_callsrXtextZ String_testTr*r9y?c Csfttttf}t||r0t|}|t|ksbtn2dt|j}t j t |dt|W5QRXdS)NzJonly str, int, float, and bool values can be copied to the clipboard, not rO) strrrbool isinstancer rWtyperQrRrSr)rYZ valid_typesresultrTrrrtest_stringify_texts  r_c#sBifdd}fdd}|d||d|VdS)aMFixture 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. cs|jj<dSNnodenamedataZ _mock_datarCrr _mock_setsz!mock_clipboard.._mock_setcs jjSr`rarrfrr _mock_getsz!mock_clipboard.._mock_getz!pandas.io.clipboard.clipboard_setz!pandas.io.clipboard.clipboard_getN)rG)rHrCrgrhrrfrmock_clipboards   ricCsFddl}|jjddt|ks*t|jj}|dksBtdS)Nrr )pandas.io.clipboardio clipboardr setvaluesrWr )ripandasr^rrrtest_mock_clipboards  rpc @steZdZd>ddZddZejdddd gd d Zd d Z ddZ ddZ ejddddgejddddgddZ ejdddgejddgddZ ddZddZejddd d!d"d#gd$d$d%gd&d'd&gdddggfdd(d)d*d+gd$d$d%gd&d'd&ggfgd,d-Zd.d/Zejd0d1d2d3gd4d5Zejd6d7d8d9gejjd:edd;dsZ    G