3
���hP$ � @ s� d dl Zd dlZd dlmZ d dlmZmZm Z m
Z
d dl j Z
G dd� d�Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )� N)�join)� Categorical� DataFrame�Index�mergec @ s, e Zd Zejjddddddg�dd� �Zd S )
� TestIndexer�dtypeZint32�int64Zfloat32Zfloat64�objectc
C s� t j}tjd|d�}tjdd|d�}tjg |d�}|||�\}}}t|tj�sRt�t|tj�sbt�t|tj�srt�tj |tjd|d�� tjdddddgtj
d�} tj || � tjd d
dddgtj
d�} tj || � |||�\}}}tj ||� tjd d d
gtj
d�} tj || � tjdddgtj
d�} tj || � |||�\}}}tj ||� tjdddgtj
d�} tj || � tjdddgtj
d�} tj || � d S )N� )r � � r � ���r r r r r r r r r ) �_join�outer_join_indexer�np�arange�array�
isinstanceZndarray�AssertionError�tm�assert_numpy_array_equalr )
�selfr Zindexer�left�right�empty�resultZlindexerZrindexer�exp� r �8/tmp/pip-build-5_djhm0z/pandas/pandas/tests/test_join.py�test_outer_join_indexer s2 z#TestIndexer.test_outer_join_indexerN)�__name__�
__module__� __qualname__�pytest�markZ parametrizer! r r r r r
s r c C sj t jdddddgt jd�} t jdddddgt jd�}tj|| �}t jdddddgt jd�}tj||� d S )Nr r r � r
)r )r r r r Zleft_join_indexer_uniquer r )�a�br �expectedr r r � test_left_join_indexer_unique. s
r+ c e C sd t jddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddgdt jd�} t jddgt jd�}d}tj| ||dd�\}}t jt| �t jd�}t jt| �t jd� }d|| dk< d|| dk< tj ||� tj ||� d S ) Nr r r r )r r' F)�sort)
r r r r Zleft_outer_joinr �lenZonesr r )r r Z
max_groups�lidx�ridx�exp_lidx�exp_ridxr r r �test_left_outer_join_bug7 s�
r2 c C s. t jdddddgt jd�} t jddddd gt jd�}tj| |�\}}}t jddgt jd�}tj||� t jddgt jd�}t jddgt jd�}tj||� tj||� t jdgt jd�} t jdgt jd�}tj| |�\}}}tj|t jdgt jd�� tj|t jdgt jd�� tj|t jdgt jd�� d S )
Nr r r r' r
)r r � � )r r r r �inner_join_indexerr �assert_almost_equalr )r( r) �index�ares�bres� index_exp�aexp�bexpr r r �test_inner_join_indexer� s r= c C sR t jdddddgt jd�} t jddddd gt jd�}tj| |�\}}}t jdddddddd gt jd�}tj||� t jd
dddddd d gt jd�}t jdd
ddddddgt jd�}tj||� tj||� t jdgt jd�} t jdgt jd�}tj| |�\}}}tj|t jdgt jd�� tj|t jdgt jd�� tj|t jdgt jd�� d S )Nr r r r' r
)r r r3 r4 r r r r r r )r r r r r r r6 r )r( r) r7 r8 r9 r: r; r< r r r r! � s r! c C s& t jdddddgt jd�} t jddddd gt jd�}tj| |�\}}}tj|| � t jdddddgt jd�}t jd
d dd dgt jd�}tj||� tj||� t jdgt jd�} t jdgt jd�}tj| |�\}}}tj|t jdgt jd�� tj|t jdgt jd�� tj|t jdgt jd�� d S )
Nr r r r' r
)r r r3 r4 r r r )r r r r �left_join_indexerr r6 r )r( r) r7 r8 r9 r; r< r r r �test_left_join_indexer� s r? c C s� t ddddg�} t dddddg�}tj|j| j�\}}}tjddddddgtjd�}tj||� tjdddddd gtjd�}tj||� tjddddd
d gtjd�}tj||� d S ) Nr r r
r3 r4 )r r r r' r r ) r r r> �valuesr r r r r6 )�idx�idx2�resr. r/ �exp_resr0 r1 r r r �test_left_join_indexer2� s rE c C s� t ddddg�} t dddddg�}tj|j| j�\}}}tjddddddgtjd�}tj||� tjdddddd gtjd�}tj||� tjddddd
d gtjd�}tj||� d S ) Nr r r
r3 r4 )r r r r' r r ) r r r r@ r r r r r6 )rA rB rC r. r/ rD r0 r1 r r r �test_outer_join_indexer2 s rF c C s� t ddddg�} t dddddg�}tj|j| j�\}}}tjddddgtjd�}tj||� tjddddgtjd�}tj||� tjddddgtjd�}tj||� d S ) Nr r r
r3 r4 )r r r ) r r r5 r@ r r r r r6 )rA rB rC r. r/ rD r0 r1 r r r �test_inner_join_indexer2 s rG c C s� t ddddddgdddg�ddddddgd�} t| �} t ddddddgdddg�ddddddgddd d
d d gd
�}t|�jddg�d }t| |j� ddgddgdd�}| j|ddgd�}|jddgdd�}tj||� t ddddddgdddgdd�ddddddgd�} t| �} t ddddddgdddgdd�ddddddgddd d
d d gd
�}t|�jddg�d }t| |j� ddgddgdd�}| j|ddgd�}|jddgdd�}tj||� d S )Nr( r) �cr r )�Cat1�Int1g�������?g333333�?g�������?gffffff�?g �?g�������?)�Cat�Int�FactorrK rL rM rI rJ r )Zleft_onZright_on�how)�on)ZaxisT)�ordered) r r Z set_indexr Z reset_indexr Zdropr Zassert_frame_equal)r( r) r* r r r r �&test_merge_join_categorical_multiindex% sD rQ )Znumpyr r% Z pandas._libsr r Zpandasr r r r Zpandas._testingZ_testingr r r+ r2 r= r! r? rE rF rG rQ r r r r �<module> s
$ z |