3
���hK" � @ s. d dl m Z d dlZd dlZd dlZd dlmZmZmZ d dl j
Z ej dd� �Z
ej dd� �Zej dd � �Zejjd
d d ed
dgddgd�ddgd�fd dedd
gddgd�ddgd�fdd ed
dd gddejgd�ddd gd�fdded dd
gejddgd�d ddgd�fd d eejdd
gdddgd�dddgd�fd dedd
ejgdddgd�dddgd�fdd ed dd
ejgejdddgd�d dddgd�fdded dd
ejgejdddgd�d dddgd�fg�dd� �Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zejjd'dd g�d(d)� �ZG d*d+� d+�ZdS ),� )�datetimeN)� DataFrame�Index� period_rangec C s* t tjd�jdd�td�tdddd�d�S ) N� � � Zabcde�2000�A)�start�freqZperiods)�data�columns�index)r �npZarangeZreshape�listr � r r �>/tmp/pip-build-5_djhm0z/pandas/pandas/tests/frame/test_join.py�frame_with_period_index s r c C s t ddddgidddgd�S )N�ar �
r � � )r )r r r r r �left s r c C s t ddddgidddgd�S ) N�bi, �d �� � r r )r )r r r r r �right s r zhow, sort, expected�innerFr r r r )r r r r )r Ti, r �outerc C s | j |||d�}tj||� d S )N)�how�sort)�join�tm�assert_frame_equal)r r r! r" �expected�resultr r r � test_join s -r( c C s� | j | jd d� ddgf }| j | jdd � ddgf jd d d� }|j|�}tj|j|j� tddddg�}tj|j|� |j|dd �}tj|j|j� tj|j|� |j|d
d �}tj|j|j� tj|j|� |j|d d �}tj|j|jdd� � tj|j|� |j|d d �}tj|j| jj� � tj|j|� t j
t d
d�� |j|dd � W d Q R X d}x6dD ].}t j
t |d�� | j| |d � W d Q R X �qjW d S )Nr r
�Br �C�Dr r )r! r r r z join method)�matchZfooz columns overlap but no suffix���)r r r ) �locr Zilocr# r$ Zassert_index_equalr r Z sort_values�pytest�raises�
ValueError)� float_frame�f�f2�joinedZexpected_columns�msgr! r r r �test_join_indexO s0 (
r7 c C s� | j d d �ddgf }| j d d d�ddgf }|j� }| d d d d� |d< | d d d d� |d< |j|�}tj||� |j|dd�}tj||d d d� � |j|dd�}tj||j d d �|jf � d S )Nr
r) r r* r+ r )r! )r. �copyr# r$ r% r )r2 �afZbfr& r' r r r �test_join_index_morex s
r: c C s` | j � }|j| jd �}|j|�}tj|| dd� d |_tjt dd�� |j|� W d Q R X d S )Nr F)Z check_nameszmust have a name)r, r- )
r8 �popr r# r$ r% �namer/ r0 r1 )r2 Zdf�sr5 r r r �test_join_index_series� s
r> c C s� | j d d �dddgf }| j d d �dddgf }|j|ddd�}|j d d �ddgf jd�}|j d d �ddgf jd�}| j d d �ddgf }|j|�j|�}tj||j d d �|jf � d S )Nr
r) r* r+ Z_df1Z_df2)ZlsuffixZrsuffix)r. r# Z
add_suffixr$ r% r )r2 �df1�df2r5 Zdf1_sufZdf2_sufZ
no_overlapr& r r r �test_join_overlap� s rA c C s^ | j dd� d�}tj| jgd dd�}| jj|j�}| j|�}t||| jd�}t j
||� d S )Nc S s | � | � �S )Nr )�keyr r r �<lambda>� s z(test_join_period_index.<locals>.<lambda>)r r r )Zaxis)r
r r ) �renamer Z concatenate�valuesr �appendr# r r r$ r% )r �otherZ
joined_valuesZ joined_colsr5 r& r r r �test_join_period_index� s
rH c C s� t ddddgidddgd�} t d d
d d gid
ddgd�}t ddddgiddd
gd�}| j||gdd�}t ddddgtjd d d gtjddtjgd�ddddgd�}tj||� d S )Nr r r r r r r )r r r r i, r �ci� i� iX r )r! )r r rI )r r# r �nanr$ r% )r? r@ Zdf3r5 r& r r r �(test_join_left_sequence_non_unique_index� s
rK �sort_kwc C s� t dddgiddgd�}t ddd gid
dgd�}t d d d
gidd
gd�}t dtd�dd�ddtd�d�ddtd�d�d��}| dkr�|jddd
gd�}tjd dd�� |j||gd| d�}W d Q R X tj||� d S )N�col1r r rI r )r �col2r r r �col3� � g @rJ g �?)r r rI g @g @g @g @)rM rN rO F)Zcheck_stacklevelr )r! r" )r �floatZreindexr$ Zassert_produces_warningr# r% )rL r r rI r& r' r r r �)test_suppress_future_warning_with_sort_kw� s rS c @ s e Zd Zdd� Zdd� ZdS )�TestDataFrameJoinc C sr ddg}t ddd�t ddd�g}t|td�dgd�}td dgd
d gg||d�}|j|dd �}t|j�d
ksnt�d S )
NZ20120209Z20120222i� r � � Zaa)r r r r r )�on)r r �ranger# �lenr �AssertionError)�selfZ str_datesZdt_datesr
r* Ztstr r r �test_join_str_datetime� s z(TestDataFrameJoin.test_join_str_datetimec C s� t jdddgdddgdddgddd gddd
gddd gd dd
gd ddgd ddgg dddgd�jddg�}t jddgddggddgd�jdg�}t jddgddgddgd!dgd
dgd"dgdtjgdtjgdtjgg |jddgd�}tj|j|dd �|� tj|j|d d �|ddg � t j j
ddgdddggddgd �}t jddgddgddgd#dgd
dgd$dgg|ddgd�}tj|j|d d �|� tj|j|dd �|ddg � d S )%Nr �xgR
���1�?�yg��� ��?�zg�<�+J �?r g��4��?g^�9�S��?g? ����?rI r r r �first�secondZvalue1)r r r Zvalue2g �?g @g @)r r r )r! r )�namesg��4�ֿg? �����g��4�ֿg? �����g��4�ֿg? �����) �pdr Z set_indexr rJ r r$ r% r# Z
MultiIndexZ from_product)r[ r? r@ �expZexp_idxr r r � test_join_multiindex_leftright� sT
z0TestDataFrameJoin.test_join_multiindex_leftrightN)�__name__�
__module__� __qualname__r\ re r r r r rT � s rT ) r Znumpyr r/ Zpandasrc r r r Zpandas._testingZ_testingr$ Zfixturer r r �markZ parametrizerJ r( r7 r: r> rA rH rK rS rT r r r r �<module> sT
""&&&& )
|