3
���h\ � @ s d dl mZ d dlZd dlmZ d dlZd dlZd dlm Z m
Z
m Z d dl j
Zd dlmZmZ dd� Zejdd d
d d d
ddddg
d�dd� �Zejdd� �Zejjdd� �Zejjejjejjd�G dd� d����Zejjejjejj dd d d g�d d � ���Z dS )!� )�dedentN)�randint)� DataFrame�
get_option�read_clipboard)�
clipboard_get�
clipboard_setc C s( i }|dkr||d< | dkr$| |d< |S )N�default�excel�sep� )r r
�kwargsr r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/io/test_clipboard.py� build_kwargs s r �delims�utf8�utf16�string�long�nonascii�colwidth�mixed�float�int)�paramsc s� | j }|dkr&tjddgddgd��S |dkrFtjdd gd
d gd��S |d krftjd
d
gddgd��S |dkr�tjddddd gd gd�S |dkr�td�}tj|d ddd� ddd gd gd�S |d kr�tjd j� d j� d ��S |d k�rtd!�d � tjdd� fd"d�ddd gd gd�S |d#k�rPttjd$d%�d& tjdd'�j tj
�t d(�d)��S |d*k�rztjddd+d� ddd gd gd�S |d,k�r�tjddd-d� ddd gd gd�S 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 d�S )N� )r )�argsr r r �<lambda>9 s zdf.<locals>.<lambda>)Z
data_gen_fr# r$ r% r& r z
in Englishu en español)�en�esr zdisplay.max_colwidthc s d� S )N�xr )r) )�_cwr r r* F s r g �?g @g{�G�z�?� Zabcde)r r �cr c S s t | �d S )Ng{�G�z�?)r )�rr0 r r r r* X s r c W s t d�S )Nr( )r )r) r r r r* b s )
�param�pdr �tmZmakeCustomDataframer �split�npZarangeZastypeZint64�list�
ValueError)�requestZ data_typeZmax_rowsr )r. r �df sr
r: c # sB i � � �fdd�}� �fdd�}| j d|� | j 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)�
_mock_datar9 r r � _mock_set| s z!mock_clipboard.<locals>._mock_setc s � �j j S )N)r; r< r )r> r9 r r � _mock_get s z!mock_clipboard.<locals>._mock_getz!pandas.io.clipboard.clipboard_setz!pandas.io.clipboard.clipboard_getN)�setattr)Z monkeypatchr9 r? r@ r )r> r9 r �mock_clipboardl s rB c C sF dd l }|jjjd� dt| j� �ks*t�|jjj� }|dksBt�d S )Nr r )�pandas.io.clipboard�io� clipboardr �set�values�AssertionErrorr )rB �pandas�resultr r r �test_mock_clipboard� s
rK c @ s� e Zd Zd$dd�Zdd� Zejjdddd g�d
d � �Zd d
� Z dd� Z
dd� Z ejjddddg�ejjddddg�dd� ��Z ejjdddg�ejjddg�dd� ��Z
dd� Zd d � Zejjd d d d!g�d"d#� �ZdS )%�
TestClipboardNc C s2 |j |||d� t|pdd|d�}tj||� d S )N)r
r �encoding� r )r Z index_colrM )� to_clipboardr r4 �assert_frame_equal)�selfr= r
r rM rJ r r r �check_round_trip_frame� s z$TestClipboard.check_round_trip_framec C s | j |� d S )N)rR )rQ r: r r r �test_round_trip_frame� s z#TestClipboard.test_round_trip_framer rN �,�|c C s | j ||d� d S )N)r )rR )rQ r: r r r r �test_round_trip_frame_sep� s z'TestClipboard.test_round_trip_frame_sepc C s< |j dd d� t� }|j� |j� ks(t�|j|jks8t�d S )NF)r
r )rO r � to_stringrH �shape)rQ r: rJ r r r � test_round_trip_frame_string� s z*TestClipboard.test_round_trip_frame_stringc C s& t j� � |jddd� W d Q R X d S )NTz\t)r
r )r4 �assert_produces_warningrO )rQ r: r r r �test_excel_sep_warning� s
z$TestClipboard.test_excel_sep_warningc C s& t j� � |jddd� W d Q R X d S )NFrN )r
r )r4 rZ rO )rQ r: r r r �test_copy_delim_warning� s
z%TestClipboard.test_copy_delim_warningr r
Tc C s6 t ||�}|jf |� ||jj |jdd�ks2t�d S )NrN )r )r rO r; r< Zto_csvrH )rQ r r
r: r9 rB r
r r r � test_clipboard_copy_tabs_default� s
z.TestClipboard.test_clipboard_copy_tabs_defaultFc C sH t ||�}|jf |� tdd�}|j� |j� ks4t�|j|jksDt�d S )Nz\s+)r )r rO r rW rH rX )rQ r r
r: r
rJ r r r �test_clipboard_copy_strings� s
z)TestClipboard.test_clipboard_copy_stringsc C s� t dd�}tdj� �}|||jj< tjf |�}|jd d dksDt�tdj� �}|||jj< tjf |�}tdj� �}|||jj< tjf |�}t j
||� d S )N�python)Zenginez^
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
) �dictr �stripr; r< r3 r ZilocrH r4 rP )rQ r9 rB Z clip_kwargs�textr: �res�expr r r � test_read_clipboard_infer_excel� s
z-TestClipboard.test_read_clipboard_infer_excelc C sH t jt�� |jdd� W d Q R X t jt�� tjdd� W d Q R X d S )N�ascii)rM )�pytestZraisesr8 rO �NotImplementedErrorr3 r )rQ r: r r r �test_invalid_encoding� s z#TestClipboard.test_invalid_encoding�enczUTF-8zutf-8r c C s | j ||d� d S )N)rM )rR )rQ rj r: r r r � test_round_trip_valid_encodings� s z-TestClipboard.test_round_trip_valid_encodings)NNN)�__name__�
__module__� __qualname__rR rS rg �mark� parametrizerV rY r[ r\ r] r^ re ri rk r r r r rL � s
(rL r= u 👍...u Ωœ∑´...zabcd...c C s t | � | t� kst�d S )N)r r rH )r= r r r �test_raw_roundtrip� s rq ) �textwrapr Znumpyr6 Z numpy.randomr rg rI r3 r r r Zpandas._testingZ_testingr4 rC r r r Zfixturer: rB ro rE rK ZsingleZ usefixturesrL rp rq r r r r �<module> s6
J
i |