Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/merge/__pycache__/test_merge.cpython-37.pyc
Назад
B �5�g�] � @ s d dl mZm Z mZ d dlZd dlZd dlZd dlZd dlm Z m Z d dl m Z d dl Zd dl mZmZmZmZmZmZmZmZmZmZmZmZmZ d dl m Z d dl m Z d dl!m"Z" d dl#m$Z$m%Z% dZ&d Z'e'e&fd d �Z(d d � Z)dd� Z*ej+e)� dd� d�dd� �Z,ej+e)� dd� d�dd� �Z-ej+e*� dd� d�dd� �Z.ej+dd � �Z/G d d � d �Z0d d � Z1G d!d"� d"�Z2ej+d#d$� �Z3ej+d%d&� �Z4G d'd(� d(�Z5ej+d)d*� �Z6ej+d+d,� �Z7G d-d.� d.�Z8ej9j:d/ed0d1gd0d1gd2d3�ed4d5gd2d6�ed7d8gd2d6�ed7d8gd2d6�ed d8d2d9�ed:d;gd2d6�gd<d� d=�d>d?� �Z;ej9�:d@dAdBgdddCdCd1fdDddAdBgddCdEd1fdddAdBgdEdCd1fdAdBgdddCdCdfdFddAdBgddCdEdfdddAdBgdEdCdfg�dGdH� �Z<ej9�:dId d dJdKidLdMgfd d dJdNid dMgfd d dJdOidPdQgfd d dJdRdSgidPdQgfdTd dJdUidTd gfdVdVdJdWidXdVgfdYdYdJdUidYdZgfdTdTdJdWid[dTgfdTdYdJdWidTdYgfdTdTdJd\idTd[gfd d dJd]id^d gfdTdTi d[d_gfd d i dPdQgfg �d`da� �Z=ej9�:dbd&edcdddegdfdgej>gdhdidjgdk��fdAedcddd7degdfdgdlej>gdhdiej>djgdk��fg�dmdn� �Z?ej9�:dodpdqdrg�dsdt� �Z@ej9�:dJd$d&hd d du�g�dvdw� �ZAej9�:dxdydTdTeBdT�dzfg�d{d|� �ZCej9�:d}d~dg�ej9�:d�dEdCg�d�d�� ��ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHej9�:d�d�d�g�d�d�� �ZId�d�� ZJej9�:d�d�d�g�ej9�:d�dTdEd��d�d�gfdTdEd��d�d�gfg�d�d�� ��ZKej9�:d�d�d�g�ej9�:d�d�dTid�d�gfd�dTid�d�gfg�d�d�� ��ZLej9�:d�d�d�g�ej9�:d�d�dEid�dEig�d�d�� ��ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTej9�:d�dd�g�d�d�� �ZUdS )�� )�date�datetime� timedeltaN)�is_categorical_dtype�is_object_dtype)�CategoricalDtype) � Categorical�CategoricalIndex� DataFrame� DatetimeIndex� Float64Index� Int64Index� IntervalIndex� MultiIndex� PeriodIndex� RangeIndex�Series�TimedeltaIndex� UInt64Index)�concat)� MergeError�merge�2 � c C s^ t t| ��}t�t�||| ��}t|�|k rPt�t |�|d |t|� � �}t�|� |S )N)�list�range�npZasarray�tile�len�random�shuffle)�ngroups�nZ unique_groups�arr� r$ �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/merge/test_merge.py� get_test_data. s " r& c C sT t dgdd�t dgdd�t dg�t dg�t dg�t t�d�g�t tjdd d �g�gS ) N� �int64)�dtype�Int64g�G�z��?�fooTz 2018-01-01z US/Eastern)�tz)r �pd� Timestampr$ r$ r$ r% � get_series9 s r/ c C s8 t tjgdd�t tjgdd�t tjgdd�t tjg�gS )Nr* )r) �float�object)r r �nanr- �NaTr$ r$ r$ r% � get_series_naE s r4 c C s | j jS )N)r) �name)�xr$ r$ r% �<lambda>N � r7 )�params�idsc C s | j S )zV A parametrized fixture returning a variety of Series of different dtypes )�param)�requestr$ r$ r% �series_of_dtypeN s r= c C s | j jS )N)r) r5 )r6 r$ r$ r% r7 W r8 c C s | j S )zk A duplicate of the series_of_dtype fixture, so that it can be used twice by a single function )r; )r< r$ r$ r% �series_of_dtype2W s r> c C s | j jS )N)r) r5 )r6 r$ r$ r% r7 ` r8 c C s | j S )zU A parametrized fixture returning a variety of Series with all NA values )r; )r< r$ r$ r% �series_of_dtype_all_na` s r? c C sR t ddgddgddgd��} t ddddd gddddd gdddddgd ��}| |fS ) Nr r' � �a�b)�col1� col_conflict�col_left� � � )rC rD � col_right)r )�df1�df2r$ r$ r% �dfs_for_indicatori s rL c @ s| e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zej�d)d*d*d+�d*d,d-�d.d*d/�d.d,d0�g�d1d2� �Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z ej�d=d>d?d?d@d?d?gfd*d@d?d?d?d?gfg�ej�dAdBdCg�dDdE� ��Z dFdG� Z!dHdI� Z"dJdK� Z#dLdM� Z$dNdO� Z%dPdQ� Z&ej�dRdSdTdUdVdWdXdYg�dZd[� �Z'd\d]� Z(d^d_� Z)d`da� Z*dbdc� Z+ddde� Z,dfdg� Z-dhdi� Z.djdk� Z/dldm� Z0dndo� Z1dpdq� Z2drds� Z3dtdu� Z4dvdw� Z5dxdy� Z6ej�dAdCdzg�ej�d{e7d?d|d}g�e7d?d|d}d~d~d~g�fe8dd�d�g�e8dd�d�e9j:e9j:e9j:g�fe;d?d|d�g�e;d?d|d�d~d~d~g�fe<d?d|d�g�e;d?d|d�d~d~d~g�fe=�>d�d�d�g�e=�>d�d�d�e?j@e?j@e?j@g�feAdd�d�gdSd��eAdd�d�e9j:e9j:e9j:gdSd��feBd�d�d�g�eBd�d�d�e9j:e9j:e9j:g�fg�d�d�� ��ZCd�d�� ZDej�dAdBdCg�d�d�� �ZEd�d�� ZFd�d�� ZGd~S )�� TestMergec C s� t t� t� tj�t�tj�t�d��| _| j| jd dk | _t ttd d�ttd td d�tj�td �d��| _t d d d d d d d gtj�d�d��| _ t dtj�d�id d d d gd�| _ d S )N)�key1�key2�data1�data2rO r' rH )r" r@ )r! r" )rN rO �valuerA rB �c�d�e� )�key�v1�v2rG )�index) r r& r r �randn�N�df�NGROUPSrK �left�right)�self�methodr$ r$ r% � setup_methodw s $zTestMerge.setup_methodc C sR t � }t dddgiddgdd�}t||ddd�}t dg ig dd�}t�||� d S ) NrA r' r@ r r( )rZ r) T)� left_index� right_index)r r �tm�assert_frame_equal)ra �df_emptyZdf_a�result�expectedr$ r$ r% �test_merge_inner_join_empty� s z%TestMerge.test_merge_inner_join_emptyc C s4 t | j| j�}t | j| jddgd�}t�||� d S )NrN rO )�on)r r] rK rf rg )ra �joined�expr$ r$ r% �test_merge_common� s zTestMerge.test_merge_commonc C sZ t ddddgddddgddddgddddgd��}|�t�}|}t||�}t�||� d S )Nr' r r@ rF )r r' r@ rF )r �astyper0 r rf rg )ra r_ r` rj ri r$ r$ r% � test_merge_non_string_columns� s 0 z'TestMerge.test_merge_non_string_columnsc C sN | j �d�}| j�d�}t||dd�}t| j | jdd��d�}t�||� d S )NrN )rl )r] � set_indexrK r rf rg )ra r_ r` ri rj r$ r$ r% �test_merge_index_as_on_arg� s z$TestMerge.test_merge_index_as_on_argc C s� t dddddddgtj�d�d��}t dtj�d �iddddgd �}t||d d d dd�}t||d d ddd�}t�||jd d �|jf � t||d d d d d�}t||d d dd d�}t�||jd d �|jf � d S )NrA rB rS rT rU rV )rW rX rY rG )rZ rW Tr_ F)�left_onre �how�sortr` )�right_onrd ru rv ) r r r r[ r rf rg �loc�columns)ra r_ r` Zmerged1�merged2r$ r$ r% �(test_merge_index_singlekey_right_vs_left� s " z2TestMerge.test_merge_index_singlekey_right_vs_leftc C s� t dddddddgtj�d�d��}t dtj�d �iddddgd �}t||d d d d�}|j|d d�j|j }t� ||� t||d d d d�}|j|d d�j|j }t� ||jd d �|j f � d S )NrA rB rS rT rU rV )rW rX rY rG )rZ rW T�inner)rt re ru )rl )rw rd ru ) r r r r[ r �joinrx rZ rf rg ry )ra r_ r` ri rj r$ r$ r% � test_merge_index_singlekey_inner� s " z*TestMerge.test_merge_index_singlekey_innerc C s� d}t jtjj|d�� t| j| jdd� W d Q R X d}t jtjj|d�� t| j| jdd� W d Q R X d}t jtjj|d�� t| j| jddd � W d Q R X d }t jt|d�� t| j | j d gd d gd � W d Q R X d S )Nz&Must pass right_on or right_index=True)�matchT)rd z$Must pass left_on or left_index=True)re zRCan only pass argument "on" OR "left_on" and "right_on", not a combination of bothrW )rt rl z)len\(right_on\) must equal len\(left_on\)rN rO )rt rw ) �pytest�raisesr- �errorsr r r_ r` � ValueErrorr] rK )ra �msgr$ r$ r% �test_merge_misspecified� s z!TestMerge.test_merge_misspecifiedc C s� d}t jt|d�� t| j| jddddgd� W d Q R X d}t jt|d�� t| j| jdddgdd� W d Q R X t jt|d��$ t| j| jdddgddgd� W d Q R X d S ) Nz>right_index parameter must be of type bool, not <class 'list'>)r r_ FrN rO )ru rd re z=left_index parameter must be of type bool, not <class 'list'>)r� r� r� r r] rK )ra r� r$ r$ r% �&test_index_and_on_parameters_confusion� s. z0TestMerge.test_index_and_on_parameters_confusionc C sT t | j| jdd�}| jd �� d �� }t|�|ks8t�d|ksDt�d|ksPt�d S )NrW )rl r@ Zv1_xZv1_y)r r_ Z value_counts�sumr �AssertionError)ra �mergedZexp_lenr$ r$ r% �test_merge_overlap s z TestMerge.test_merge_overlapc C s t ddddgddddgd��}t ddd dgd d d d gd��}|j|ddddd�}tddddddtjgdd�}t�|d |� tdtjddddd gdd�}t�|d |� tddddddtjgdd�}t�|d |� td tjd d d d d gdd�}t�|d |� d S )Nr+ �barZbazr' r@ rF rG )�lkeyrR ZquxrH � rV r )�rkeyrR r� r� �outerT)rt rw ru rv )r5 �value_x�value_y)r r r r r2 rf �assert_series_equal)ra r_ r` r� rn r$ r$ r% �%test_merge_different_column_key_names s z/TestMerge.test_merge_different_column_key_namesc C sz t ddd�td�d�}t ddd�td�d�}t||d d d d �}d |d < |d dk�� sZt�d |d< |d dk�� svt�d S )Nr r' )rA rB � )rZ r+ r� )rS rT T)rd re �copyr� rA �peekaboorT )r r r �allr� )ra r_ r` r� r$ r$ r% �test_merge_copy# s zTestMerge.test_merge_copyc C s� t ddd�td�d�}t ddd�td�d�}t||d d d d �}|r||jjd |jjd ks^t�|jjd |jjd ks�t�n8d |d< |d d k�� s�t�d|d< |d dk�� s�t�d S )Nr r' )rA rB r� )rZ r+ r� )rS rT TF)rd re r� r@ r� rA r� rT )r r r �_mgr�arraysr� r� )ra �using_array_managerr_ r` r� r$ r$ r% �test_merge_nocopy/ s zTestMerge.test_merge_nocopyc C s� t dddddgttd��d�ddgd�}t ddddd dgttd ��d ��}t||dd d �}t dddddddd dg t�ddddddd tjtjg �dddddddd dg d�dddgd�}t�||� d S )Nr' r@ rF rH )rW rR rR rW )ry rG r� )rW �rvaluer� )rl ru r )rW rR r� r� ) r r r r r �arrayr2 rf rg )ra r_ r` rm rj r$ r$ r% �"test_intelligently_handle_join_keyE s &" z,TestMerge.test_intelligently_handle_join_keyc C s� t dgdgd��}t dgdgd��}t||dd�}|d jd ksDt�t d gdgd��}t d gd gd��}t||dd�}|d jd ks�t�t ddgi�}t ddgi�}t�dg�}t�dg�}t||||dd�}|d jd ks�t�d S )Nr' r� )rW rX r@ � r� )ru rW r( TFr �bool�val)rt rw ru �key_0)r r r) r� r r� )ra rJ rK r] r� r� r$ r$ r% � test_merge_join_key_dtype_castX s z(TestMerge.test_merge_join_key_dtype_castc C s� t dddddgt�d�d�ddgdd �}t d t�d �idd �}tjddddd dgdd �}t||d|dd�}t|||ddd�}t�|d |d � |d �� �� s�t �|d �� �� s�t �t dt�d�idgd�}t d t�d �i�}t�dddddg�}t�ddddd dg�}t||||dd�}t�|d t dddddddd dg dd�� t dt�d�i�}t d t�d �i�}tjddddddgtj d �}t||d|dd�}t�|d t |dd�� d S )Nr' r@ rF rH )rW rR rR rW r( )ry r) r� r� )r) rG r� )rt rw ru )ry r� )r5 r T)rd rw ru ) r r �aranger� r rf r� Znotnar� r� r r( )ra r_ r` rW r� rz r� r� r$ r$ r% � test_handle_join_key_pass_arrayo s. & z)TestMerge.test_handle_join_key_pass_arrayc C sx t �� }tddgi|gd�}tdddgi||gd�}dd � dd � d d � d d � �}tjt|d �� t||� W d Q R X d S ) Nr6 rA )rZ �yrB rS z>No common columns to perform merge on. Merge options: left_on=z , right_on=z , left_index=Fz, right_index=)r )r �nowr r� r� r r )ra �dtrJ rK r� r$ r$ r% �&test_no_overlap_more_informative_error� s z0TestMerge.test_no_overlap_more_informative_errorc C s� t ddd�}t ddd�}t ddd�}t ddd�}tddgi|gd �}td d d gi||gd �}t||� tddd d gi|||gd �}td d dddddgi||||||gd �}t||� tddd gi||gd �}td d dgi||gd �}t||� d S )Ni� rH r' r@ rF rG r6 rA )rZ r� rB rS �qrT rU �f�g�h)r r � _check_merge)ra r� �dt2�dt3Zdt4rJ rK r$ r$ r% � test_merge_non_unique_indexes� s & z'TestMerge.test_merge_non_unique_indexesc C sv t ddd�}t ddd�}t ddd�}tdddd d gi||||gd �}td d ddddgi|||||gd �}t||� d S )Ni� rH r' r@ rF r6 rA rB rS rT )rZ r� rU r� r� z h�i)r r r� )ra r� r� r� rJ rK r$ r$ r% �(test_merge_non_unique_index_many_to_many� s "z2TestMerge.test_merge_non_unique_index_many_to_manyc C sZ t dgdgd��}t dg i�}t||ddd�}t�||� t||ddd�}t�||� d S )Nr' r@ )rW rR rW r_ )rl ru r` )r r rf rg )ra r_ r` ri r$ r$ r% � test_left_merge_empty_dataframe� s z)TestMerge.test_left_merge_empty_dataframe�kwargT)rd re r6 )rd rw rA )rt re )rt rw c C sj t dddgd�}t dddgd�}t ddddddgtjg td�td �}t||fd |i|��}t�||� d S ) NrA rB rS )ry r6 r� �z)r) )ry rZ r) ru )r r- �Indexr1 r rf rg )ra � join_typer� r_ r` �exp_inri r$ r$ r% �!test_merge_left_empty_right_empty� s z+TestMerge.test_merge_left_empty_right_emptyc sx t dddgd�� t dddgdd d gd d d ggdddgd��t tjtjgd td�tjtjgd td�tjtjgd td�ddd gdd d gdd d gd�ddddddgd�}|dd� }|j�t�|_� �fdd�}� �fdd�}x0ddd�ddd�gD ]}|||� |||� q�W ddd�}|||� dddg|d< |||� ddd �}|||� tjtjgd td�|d< |||� d S ) NrA rB rS )ry r' r@ rF rG rH r� rV r � r6 r� r� )r) )rA rB rS r6 r� r� r c sH t � �fddi|��}t�|| � t � �fddi|��}t�|| � d S )Nru r| r_ )r rf rg )rn r� ri )r_ r` r$ r% �check1� s z>TestMerge.test_merge_left_empty_right_notempty.<locals>.check1c sH t � �fddi|��}t�|| � t � �fddi|��}t�|| � d S )Nru r` r� )r rf rg )rn r� ri )r_ r` r$ r% �check2� s z>TestMerge.test_merge_left_empty_right_notempty.<locals>.check2T)rd re )rd rw )rt re )rt rw )r r r� r2 r1 rZ rp )ra �exp_outr� r� r� r� r$ )r_ r` r% �$test_merge_left_empty_right_notempty� s6 * z.TestMerge.test_merge_left_empty_right_notemptyc s� t dddgdddgddd ggd d d gd ��t dddgd ��t dddgdddgddd gtjtjgd td�tjtjgd td�tjtjgd td�d�d d d dddgd ���dd� ��j�t��_��fdd�� � �����fdd��d S )Nr' r@ rF rG rH r� rV r r� rA rB rS )ry r6 r� r� )r) )rA rB rS r6 r� r� r c sH t � �fddi|��}t�|| � t � �fddi|��}t�|| � d S )Nru r| r` )r rf rg )rn r� ri )r_ r` r$ r% r� % s z>TestMerge.test_merge_left_notempty_right_empty.<locals>.check1c s� t ��fddi|��}t�|| � t ��fddi|��}t�|| � x@ddd�ddd�ddd �ddd �gD ]}� �|� ��|� qjW d S ) Nru r_ r� T)rd re r6 )rd rw rA )rt re )rt rw )r rf rg )rn r� ri )r� r� r� r� r_ r` r$ r% r� + s z>TestMerge.test_merge_left_notempty_right_empty.<locals>.check2)r r r� r2 r1 rZ rp )ra r$ )r� r� r� r� r_ r` r% �$test_merge_left_notempty_right_empty s * z.TestMerge.test_merge_left_notempty_right_emptyc C s~ t ||d�ddgd�}|d d� }t t|jd d�t|jd d�t|jd d�d�ddd gd�}|j|dd �}t�||� d S ) N)rW rR rW rR )ry r )r) )r� rW r� r� r� )rl )r r �dtypesr rf rg )ra r= r> r] rh rj �actualr$ r$ r% �test_merge_empty_frame; s z TestMerge.test_merge_empty_framec C sd t ||d�ddgd�}t ||d�ddgd�}t |||d�dddgd�}|j|dd�}t�||� d S ) N)rW rR rW rR )ry )rW r� r� r� r� )rl )r r rf rg )ra r= r? Zdf_leftZdf_rightrj r� r$ r$ r% �test_merge_all_na_columnM s z"TestMerge.test_merge_all_na_columnc C s t jjdddd�t jjdddd�tddd�tddd �td dd �tddd�tddd �tdd d �tdd d �td d d�td dd �tddd �g d�}t�|�}|j�� }|�� t�|t j�d�d��}|j |ddd�}t ||ddd�}t � ||� |j�� |j�� k� � �s t �d S )Nr r� )�sizei� r' � i� r@ rG i� rF i� rH )Zvar1Zvar2�var3rV )r� Zvar8r� F)rl rv )r r �randintr r � from_dictr� �uniquerv r rf rg r� r� )ra rT r] r� �newri rn r$ r$ r% �test_merge_nosortb s( zTestMerge.test_merge_nosort)rv �valuesFr' r ru r_ r` c C sB t ddddgi�}|j|d||d�}t |dgd�}t�||� d S )NrA r' r )rl ru rv )ry )r r rf rg )ra rv r� ru r] ri rj r$ r$ r% � test_merge_same_order_left_right� s z*TestMerge.test_merge_same_order_left_rightc C s� t ddgddgd��}t dgdgd��}|j|ddd�}t ddd �ddd �dtjd �dtjd �ddd �d ���d ��� dd d d g }tj||dd� d S )Nr r' )�i1�i2)r� �i3r� �_)rl �rsuffixg )r r' )r� r� �i1_r� Nr� r� r� F)Z check_dtype)r r} r r2 rr � reset_indexrf rg )ra rJ rK ri rj r$ r$ r% �test_merge_nan_right� s z TestMerge.test_merge_nan_rightc C s� t ddgddgd��}t dgdgd��}|j|dd d �}t ddd �d tjd �ddd �dtjd �d ��d dddg }t�||� d S )Nr r' g �?g �?)r� r� gffffff�?)r� r� r� r� )r� rl )r r' g )r� r� r� r� r� r� r� )r r} r r2 rf rg )ra rJ rK ri rj r$ r$ r% �test_merge_nan_right2� s z TestMerge.test_merge_nan_right2c s@ G � fdd�dt �� � | j�}|j| jdd�}t|� �s<t�d S )Nc s e Zd Ze� fdd��ZdS )z0TestMerge.test_merge_type.<locals>.NotADataFramec s � S )Nr$ )ra )� NotADataFramer$ r% � _constructor� s z=TestMerge.test_merge_type.<locals>.NotADataFrame._constructorN)�__name__� __module__� __qualname__�propertyr� r$ )r� r$ r% r� � s r� rN )rl )r r] r rK � isinstancer� )ra Znadri r$ )r� r% �test_merge_type� s zTestMerge.test_merge_typec C s� t ddddd�tdd�d�}ttd�d �}|j|d d �}|j|d d �}tt ddddd�t ddddd�gtdd�tdd�gd��}|r�|�t�}t�||� d S ) Ni� � rH �8 r i�W )rT �tr� )ry T)Z ignore_index) r r r r �appendrp r1 rf rg )ra r� rT r] ri rj r$ r$ r% �test_join_append_timedeltas� s z%TestMerge.test_join_append_timedeltasc C s� t �d�}tt||gddgd��}tt|gdgd��}|j|ddd�}tt||gtd�d�t|tjgtd�d�d ��}t� ||� d S ) Ni ���A�B)rZ �rr_ )r� ru ZAB)�0Z0r) r Z timedelta64r r r} r r- r3 rf rg )ra �td�lhs�rhsri rj r$ r$ r% � test_join_append_timedeltas2� s z&TestMerge.test_join_append_timedeltas2c C s� t dddgi�}td d gddgdd�}xrdD ]j}|�|��d�}|d jdksRt�|j|ddd �}t ddgtjd d gdd �d �ddgd �}t � ||� q,W d S )N� entity_id�e �f �days)rZ r5 )z datetime64[D]z datetime64[h]z datetime64[m]z datetime64[s]zdatetime64[ms]zdatetime64[us]zdatetime64[ns]zdatetime64[ns]T)rt re �nat)r) )r� r� )ry ) r r rp �to_framer) r� r r r� rf rg )ra rJ �sr) rK ri rn r$ r$ r% �test_other_datetime_unit� s z"TestMerge.test_other_datetime_unit�unit�Dr� �mr� �ms�us�nsc C s� t dddgi�}td d gddgdd�}d|� d�}|�|��d�}|d jdksTt�|j|dd d �}t ddgtjd d g|d �d �ddgd�}t � ||� d S )Nr� r� r� r� )rZ r5 zm8[�]zm8[ns]T)rt re r� )r) )r� r� )ry ) r r rp r� r) r� r r r� rf rg )ra r� rJ r� r) rK ri rn r$ r$ r% �test_other_timedelta_unit� s z#TestMerge.test_other_timedelta_unitc C s� t dddgdddgddd gd ��}t dddgdddgddd gd ��}d d d g|_d d d g|_t dddgdddgddd gdddgddd gd��}d d d d d g|_t�t||�|� dd d g|_d}tjt|d�� t||� W d Q R X d S )Nr' r@ rF rG rH r� rV r r� )rW rX rY rW r+ r� )rW rX rY Zv3Zv4rN z;Data columns not unique: Index\(\['foo'\], dtype='object'\))r )r ry rf rg r r� r� r )ra r] rK rj r� r$ r$ r% �&test_overlapping_columns_error_message s "" z0TestMerge.test_overlapping_columns_error_messagec C s� t tjdddd�ddgd��}t tjdddd�dddgd��}t tjdd dd�ddtjtjgtjdddgd ��}t||d d d �}t�||� d S )N�20151010r@ z US/Eastern)�periodsr, r' )rW rR �20151011rF rG )rW r� r� rW r� )rl ru )r r- � date_ranger r2 r rf rg )ra r_ r` rj ri r$ r$ r% �test_merge_on_datetime64tz# s z$TestMerge.test_merge_on_datetime64tzc C s� t ddgtjdddd�d��}t ddgtjdddd�d��}t dddgttjdddd��tjg tjgttjdddd�� d ��}t||d d d �}t�||� |d jdks�t �|d jdks�t �d S )Nr' r@ r� z US/Eastern)r r, )rW rR rF r )rW r� r� rW r� )rl ru r� zdatetime64[ns, US/Eastern]r� ) r r- r r r3 r rf rg r) r� )ra r_ r` rj ri r$ r$ r% � test_merge_datetime64tz_values= s z(TestMerge.test_merge_datetime64tz_valuesc C s� t jdd�}tt jd|jd�gdgt jd|jd�gd�ddd gd �}|d d � }|j|dd �}tttd �t|d �t|d �ttd �t|d �d�dddddgd �}t� ||� d S )N�UTC)r, Z2018g @Z2019)r rR �date2r rR r )ry r )rl )r) )r� �date2_xr r� �date2_yr� r r� r ) r- ZDatetimeTZDtyper r. r, r r r0 rf rg )ra Zdtzr` r_ ri rj r$ r$ r% � test_merge_on_datetime64tz_emptyX s z*TestMerge.test_merge_on_datetime64tz_emptyc C s� t tjddddd�dgd�}d|d < t t�d d d g�d d��}|d j�d�j�d�|d< t||ddd�}t tjddddd�dgd tj gd tj gd d gd d��}t � ||� d S )Nz2017-10-29 01:00rG �Hz Europe/Madrid)r �freqr, r )ry r' rR z2017-10-29 03:00:00z2017-10-29 04:00:00z2017-10-29 05:00:00r@ )r rR r r� )ru rl rV rF )r r� r� ) r r- r � to_datetimer� Z tz_localizeZ tz_convertr r r2 rf rg )ra rJ rK ri rj r$ r$ r% �+test_merge_datetime64tz_with_dst_transitionq s$ z5TestMerge.test_merge_datetime64tz_with_dst_transitionc C s� t jdddd�}tttt|���|dgd�}t||g�}|j|dddd �}tt� tj dtj d �� d �� d d �d �ddg|�� jd�}t�||� d S )Nz 2016-01-01� �M)r r Zpnum)rZ ry Tr| )rd re ru )r) r@ ���r' Zpnum_xZpnum_y)ry rZ )r- � period_ranger r r r r r r r r� r( �repeat�reshape� sort_indexrZ rf rg )ra rZ r] rK ri rj r$ r$ r% �"test_merge_non_unique_period_index� s $z,TestMerge.test_merge_non_unique_period_indexc C s� t tjdddd�ddgd��}t tjdddd�dddgd��}t tjdd dd�ddtjtjgtjdddgd ��}t||d d d �}t�||� d S )Nr� r@ r� )r r r' )rW rR r rF rG )rW r� r� rW r� )rl ru )r r- r r r2 r rf rg )ra r_ r` rj ri r$ r$ r% �test_merge_on_periods� s z TestMerge.test_merge_on_periodsc C s� t ddgtjdddd�d��}t ddgtjdddd�d��}tjdddd�}tjdddd�}t dddgt|�tjg tjgt|� d ��}t||d d d �}t�||� |d jdks�t �|d jdks�t �d S )Nr' r@ r� r� )r r )rW rR rF r )rW r� r� rW r� )rl ru r� z Period[D]r� ) r r- r r r3 r rf rg r) r� )ra r_ r` Zexp_xZexp_yrj ri r$ r$ r% �test_merge_period_values� s z"TestMerge.test_merge_period_valuesc C sZ |\}}|� � }|� � }tddddddgddtjtjtjtjgddtjtjtjtjgtjdddddgtjdddddgd ��}td d d d d d gd d d gd �|d< |ddddddg }t||dddd�}t�||� |j|dddd�}t�||� t�||� t�||� |}|jddid�}t||dddd�} t�| |� |j|dddd�} t�| |� d S )Nr r' r@ rF rG rH rA rB )rC �col_conflict_xrE �col_conflict_yrI � left_only�both� right_only)� categories�_mergerC r rE r rI r� T)rl ru � indicatorZ custom_name)ry ) r� r r r2 r r rf rg �rename) ra rL rJ rK Zdf1_copyZdf2_copyZ df_result�testZdf_result_custom_nameZtest_custom_namer$ r$ r% �test_indicator� sN zTestMerge.test_indicatorc C sh |\}}d}t jt|d�� t||dddd� W d Q R X t jt|d�� |j|dddd� W d Q R X d S )Nz<indicator option can only accept boolean or string arguments)r rC r� rH )rl ru r )r� r� r� r )ra rL rJ rK r� r$ r$ r% �#test_merge_indicator_arg_validation s z-TestMerge.test_merge_indicator_arg_validationc C s� |\}}t ||dddd�}|jdk�� s,t�|j |dddd�}|jdk�� sPt�t ||dddd�}|jdk�� stt�|j |dddd�}|jdk�� s�t�t ||dddd�}|jd k�� s�t�|j |dddd�}|jd k�� s�t�d S ) NrC r_ T)rl ru r r r` r r| r )r r r� r� )ra rL rJ rK Ztest2Ztest3Ztest4r$ r$ r% �%test_merge_indicator_result_integrity s z/TestMerge.test_merge_indicator_result_integrityc C s |\}}x�dD ]�}t dddg|ddgi�}d|� d�}tjt|d�� t||ddd d � W d Q R X tjt|d�� |j|ddd d � W d Q R X qW t ddgddgd ��}d }tjt|d�� t||ddd d � W d Q R X tjt|d�� |j|ddd d � W d Q R X d S )N)Z_right_indicatorZ_left_indicatorr rC r' r@ zECannot use `indicator=True` option when data contains a column named z;|Cannot use name of an existing column for indicator column)r r� T)rl ru r )rC �custom_column_namez:Cannot use name of an existing column for indicator columnr% )r r� r� r� r )ra rL rJ r� r� Z df_badcolumnr� r$ r$ r% � test_merge_indicator_invalid% s( z&TestMerge.test_merge_indicator_invalidc C s� t ddgddgd��}t dddgdddgd��}t ddddgddddgd��}td d d d gd d d gd �|d < t||ddgddd�}t�||� |j|ddgddd�}t�||� d S )Nr r' rA rB )rC �col2rF r6 r� r r r )r r rC r' r� T)rl ru r )r r r rf rg )ra Zdf3Zdf4Zhand_coded_resultZtest5r$ r$ r% �%test_merge_indicator_multiple_columnsG s z/TestMerge.test_merge_indicator_multiple_columnsc C s^ t ddddgddddgd �td �d �}t ddddd gd ddddgd�td�d �}|�� }|�� }t||dddd�}t�||� t�||� t ddddgddddgddddgd dddgd�td �ddddgd�}t||dddd�}t�||� t ddddgddddgd dddgd �td �d �}t||ddd �}t�||� t�||� t�||� t||ddd �}t�||� t ddddgddddgd dddgd �dddgtd �d �}|�d�} t| |dddd �}t�||� |�t d gd!gd�d gd ��} t|| ddd"d� d#} tj t | d$�� t|| dddd� W d Q R X tj t | d$�� t|| ddd � W d Q R X |jt dgd%gd�d&gd �dd'�} t| |ddd(d� d)} tj t | d$�� t| |dddd� W d Q R X tj t | d$�� t| |ddd � W d Q R X t| | dd*d � d+} tj t | d$�� t| | ddd(d� W d Q R X d,} tj t | d$�� t| | dd"d � W d Q R X d-} tj t | d$�� t||dd.d � W d Q R X t ddddgd/d0d/d0gddddgd �td �d �}t dddgd/d0d/gd ddgd1�td&�d �}t dddgd/d0d/gdddgd ddgd2�td&�d �} d3} tj t | d$�� t||ddd � W d Q R X t||ddgdd �}t�|| � d S )4NrA rB rS rT �cat�dogZweasel�horse)rA rB rG )rZ rU ZmeowZbarkzum... weasel noise?ZnayZchirp)rA rS rH Tz1:1)rd re �validate)�a_xrB �a_yrS r- r. )rZ ry Z one_to_one)rA rB rS )rl r, )rB rA rS )ry rZ )rd rw r, �mooZ one_to_manyzBMerge keys are not unique in right dataset; not a one-to-one merge)r �cowrF )rv Z many_to_onezAMerge keys are not unique in left dataset; not a one-to-one mergeZ many_to_manyzCMerge keys are not unique in right dataset; not a many-to-one mergezBMerge keys are not unique in left dataset; not a one-to-many mergez!Not a valid argument for validateZ jibberishr r' )rA rB rT )rA rB rS rT zQMerge keys are not unique in either left or right dataset; not a one-to-one merge) r r r� r rf rg rr r� r� r� r r� )ra r_ r` Z left_copyZ right_copyri rj Z expected_2Z expected_3Zleft_index_resetZ right_w_dupsr� Z left_w_dupsZexpected_multir$ r$ r% �test_validationZ s� zTestMerge.test_validationc C s: t g g g d��}tjdd�� t||dd� W d Q R X d S )N)rA rB rS �raise)�divide)rA rB )rl )r r Zerrstater )ra rA r$ r$ r% �)test_merge_two_empty_df_no_division_error s z3TestMerge.test_merge_two_empty_df_no_division_errorr� zindex,expected_indexr@ rG Nz 2001-01-01z 2002-02-02z 2003-03-03rF )r' r@ )r@ rF )rF rG z 2001-01-02z 2001-01-03)r Z1dZ2dZ3dc C s� t dddgdddgd�|d�}t ddddddd gi�}|j|d d |d �}t dddgdddgdddgtjddgtjddgtjd d ggd d dgd�}|j|d d� t�||� d S )Nr r' r@ )rA rW )rZ rB rF rG rH rW T)rt re ru rA )ry )�inplace)r r r r2 rr rf rg )ra ru rZ �expected_indexrJ rK ri rj r$ r$ r% �$test_merge_on_index_with_more_values s ' z.TestMerge.test_merge_on_index_with_more_valuesc C s� t dddgdddgd��}t ddddgi�}t dddd gddddgddddgd�dd dgdddtjgd �}|j|d d d d �}t�||� d S )Nr' r@ rF r )rA rW rB )rA rW rB rA rW )ry rZ Tr` )rt re ru )r r r2 r rf rg )ra r_ r` rj ri r$ r$ r% � test_merge_right_index_rightR s "z&TestMerge.test_merge_right_index_rightc C s� t ddgddgd��}t ddgddgd��}|j|dd g|d �}|d kr`t ddgddgd��}nt ddgddgd��}t�||� d S ) Nr* �pig�( r� )�animal� max_speedZquetzal�P r; r< )rl ru r` )r r rf rg )ra ru �left_df�right_dfri rj r$ r$ r% � test_merge_preserves_row_ordera s z(TestMerge.test_merge_preserves_row_orderc C s� t dddgtdddgtd�d�d��}t ddddgitddd g�d �}|j|d d d d�}t dddd gtdddd g�ddddgd�dddtjgd �}|jdd dgd�}t� ||� d S )Nr' r@ rF rA rB �abc)r )rA rW rS )rZ rW Tr` )rt re ru )rA rW rB r )ry ) r r r r r r r2 Zreindexrf rg )ra r_ r` ri rj r$ r$ r% �8test_merge_take_missing_values_from_index_of_other_dtypem s zBTestMerge.test_merge_take_missing_values_from_index_of_other_dtypec C sr t t�d��d�d dddddgd �}t t�d��d �d ddd d gd �}x|jjD ] }d |j_qTW |�|� d S )Nr� )rG rH r' rA rB rS rT rU )ry )rH rG r6 r� F) r r r� r r� r� �flagsZ writeabler )ra rP rQ r# r$ r$ r% �test_merge_readonly� s $" z TestMerge.test_merge_readonly)Hr� r� r� rc rk ro rq rs r{ r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� �mark� parametrizer� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r" r# r$ r& r( r1 r4 r r r- r3 r r r � from_tuplesr r2 r r r7 r8 r@ rB rD r$ r$ r$ r% rM v s� ! 1* & $ ? " ; * rM c C sR xLdD ]D}| j ||d�}t| �� |�� |dd�}|�d�}tj||dd� qW d S )N)r| r_ r� )ru T)ru rv rZ F)Z check_names)r} r r� rr rf rg )r6 r� ru ri rj r$ r$ r% r� � s r� c @ sJ e Zd Zej�dddgeddg��d�g�dd� �Zej�de j e j e j e j e jg�ej�de j e je je jg�d d � ��Zej�d d d dgdddgd d dgdddgd�fd d dgddgd dgddgd�fd d gdddgd d gddgd�fg�dd� �Zdd� Zdd� Zdd� Zdd � Zej�d d d d ged d d g��d�fd!ddged d d g��d�fd d ged"d#ged$�fd d ged"d#ged$�fg�d%d&� �Zej�d ed d gd'd$�d d d(gfed d gd)d$�d d d(gfd d d gd*d+d,gfd!ddgd*d+d,gfd d d gd*d+d,gfej d-d d.d/�d0d1gfej d-d d.d/�d d gfej d-d d.d/�d!dgfej d2dd3�ej d2dd4d5�fg �d6d7� �Z ej�d8d d gd9fg d:fd gd;fd gd<fg�d=d>� �Z ej�d8d d gd9fg d:fd gd;fd gd<fg�d?d@� �Z ej�dAd:d#d dBgd"dCdggfd9d#d dBgd"dCdggfd<d#d dBgd"dCdggfd;d"dCdgd#d dBggfg�dDdE� �Z dFS )G�TestMergeDtypes� right_valsr+ r� �categoryc C sv t ddgtddg��d�ddgddgtddgdd �tddgd d �d ��}t d |i�}t||d d �}t|jj�srt�d S )Nr+ r� rJ r' r@ g �?g @�uint64)r) �int32)r� r� �Cr� �E�Fr� )rl )r r rp r r r� r) r� )ra rI r_ r` ri r$ r$ r% �test_different� s z TestMergeDtypes.test_different�d1�d2c C s@ t �|�}t �|�}tt jdddgd |d�ddgd t jt �d �t jd�d ��}t�d d g�}td t jddg|d�i|d�}|j|ddgd�}|� � } |j dkr�t �d�}t jt j |d�| d < d| j | j dk| jdk@ d f< d| j | j dk| jdk@ d f< t�|| � |j|ddgdd�}| jddgddd� t�|| � d S )Nr r' r@ r )r) r+ r� r� � )�k1�k2�v)r@ r� )r' r+ rY rH rV )rZ rT rU )rl r� �float64T)rl rv Z mergesort)�kindr5 )r r) r r� r� r( r rG r} r� rX r2 rx rT rU rf rg Z sort_values) ra rQ rR Zdtype1Zdtype2r_ rZ r` ri rj r$ r$ r% �test_join_multi_dtypes� s& z&TestMergeDtypes.test_join_multi_dtypesz int_vals, float_vals, exp_valsr' r@ rF g �?g @g @)�X�Yc C sd t d|i�}t d|i�}t |�}|j|ddd�}t�||� |j|ddd�}t�||ddg � d S )NrZ r[ )rt rw )r r rf rg )ra Zint_valsZ float_valsZexp_valsr� r� rj ri r$ r$ r% �test_merge_on_ints_floats� s z)TestMergeDtypes.test_merge_on_ints_floatsc C s� t ddgddgd�ddgd�}t d gd gd �dd gd�}|j|dd d�}t ddgddgtjdgd�ddd gd�}t�||� d S )Ng �?g @r� r� )rW rX rW rX )ry r@ �� )rW rY rY r_ )rl ru g i@)rW rX rY )r r r r2 rf rg )ra rJ rK ri rj r$ r$ r% �test_merge_key_dtype_cast� s z)TestMergeDtypes.test_merge_key_dtype_castc C s� t ddddgi�}t ddddgi�}t dgdgd ��}t�t��" |j|ddd �}t�||� W d Q R X t�t��* |j|ddd �}t�||ddg � W d Q R X t dtjtjdgi�}t�d ��* |j|ddd �}t�||ddg � W d Q R X d S ) NrZ r' r@ rF r[ g�������?g @g @)rZ r[ )rt rw )r rf �assert_produces_warning� UserWarningr rg r r2 )ra r� r� rj ri r$ r$ r% �!test_merge_on_ints_floats_warning� s z1TestMergeDtypes.test_merge_on_ints_floats_warningc C st t dtddgtd�i�}t dddgi�}t dddgitd�}t||dd�}t�||� t||dd�}t�||� d S )NrW TF)r) )rl )r r r1 r rf rg )ra rJ rK rj ri r$ r$ r% �(test_merge_incompat_infer_boolean_object s z8TestMergeDtypes.test_merge_incompat_infer_boolean_objectc C sx t dtddtjgtd�i�}t dddgi�}t dddgitd�}t||dd�}t�||� t||dd�}t�||� d S )NrW TF)r) )rl )r r r r2 r1 r rf rg )ra rJ rK rj ri r$ r$ r% �5test_merge_incompat_infer_boolean_object_with_missing s zETestMergeDtypes.test_merge_incompat_infer_boolean_object_with_missingzdf1_vals, df2_valsr rA rB g FT)r) c C s\ t d|i�}t d|i�}t||dgd�}t|jj�s8t�t||dgd�}t|jj�sXt�d S )Nr� )rl )r r r r� r) r� )ra �df1_vals�df2_valsrJ rK ri r$ r$ r% �!test_merge_incompat_dtypes_are_ok' s z1TestMergeDtypes.test_merge_incompat_dtypes_are_okrK rS rL r� �1�2z1/1/2011r� )r r z 2011-01-01z 2011-01-02Z20130101)r z US/Eastern)r r, c C s� t d|i�}t d|i�}d|d j� d|d j� d�}t�|�}tjt|d�� t||dgd� W d Q R X d|d j� d|d j� d�}t�|�}tjt|d�� t||dgd� W d Q R X d S )Nr� zYou are trying to merge on z and z9 columns. If you wish to proceed you should use pd.concat)r )rl )r r) �re�escaper� r� r� r )ra rd re rJ rK r� r$ r$ r% � test_merge_incompat_dtypes_error? s z0TestMergeDtypes.test_merge_incompat_dtypes_errorzexpected_data, howr� r| r` r_ c C sR t dgdg|d�}t dgdg|d�}t|||d�}t |dg|d�}t�||� d S )N)r' �id)ry r) )r@ )ru )r r rf rg )ra Zany_nullable_numeric_dtyperu � expected_datarQ rR ri rj r$ r$ r% �test_merge_EA_dtypeo s z#TestMergeDtypes.test_merge_EA_dtypec C sR t dgdg|d�}t dgdg|d�}t|||d�}t |dg|d�}t�||� d S )N)rA rl )ry r) )rB )ru )r r rf rg )ra ru rm Zany_string_dtyperQ rR ri rj r$ r$ r% �test_merge_string_dtype� s z'TestMergeDtypes.test_merge_string_dtypezhow, expected_datarG rH c C s\ t ddgddgd��}t ddgddgd��}t|||d �}t |d d d gd �}t�||� d S )NTFr' rH )r� r� rF rG )r� rM )ru r� r� rM )ry )r r rf rg )ra ru rm rJ rK ri rj r$ r$ r% �test_merge_bool_dtype� s z%TestMergeDtypes.test_merge_bool_dtypeN)!r� r� r� r� rE rF r rp rP r r( rL Zint16Zint8Zuint8rW Zfloat32Zfloat16rY r\ r^ ra rb rc r1 r� rf r- r rk rn ro rp r$ r$ r$ r% rH � sd & (!& . $ rH c C sN t j�d� ttt jjddgdd���tddg��t jjdddgdd�d ��S ) Ni� r+ r� )r� )r� �one�two�three)rZ r[ )r r �seedr r �choicerp �CDTr$ r$ r$ r% r_ � s r_ c C s4 t j�d� ttddg��tddg��ddgd��S )Ni� r+ r� r' r@ )rZ �Z)r r rt r r rp rv r$ r$ r$ r% r` � s r` c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d � dd � dd � g�dd� �Z dd� Z dd� Z ej � dddg�ej � dddddgddgddgfddddgddgddgfd d d d gd d gd d gfg�d d � ��Zd d!� Zd"S )#�TestMergeCategoricalc C sV t ||dd�}|j�� }ttddgd�t�d�t�d�gdddgd �}t�||� d S ) NrZ )rl r+ r� )r �OZY_xZY_y)rZ ) r r� r r r r r) rf r� )ra r_ r� ri rj r$ r$ r% �test_identical� s z#TestMergeCategorical.test_identicalc C sV t ||dd�}|j�� }ttddgd�t�d�t�d�gddd gd �}t�||� d S ) NrZ )rl r+ r� )r ry r( r[ rw )rZ ) r r� r r r r r) rf r� )ra r_ r` r� ri rj r$ r$ r% � test_basic� s z TestMergeCategorical.test_basicc C sD t dddddd�dddddd�d��}t d d d d d d�d d d d d d�d ��}t||d ddd �}|�� }|�� }|d �d�|d< t||d ddd �}|d �tdg��|d< t�||� |�� }|d �d�|d< t||d ddd �}t�||� |�� }|d �d�|d< |�� }|d �d�|d< t||d ddd �}t�||� d S )NrA rB rS rT rU )r r' r@ rF rG �null)rS rT r� r� )rA rB r_ )ru rt rw rJ )r r r� rp r rf rg )ra r` r_ r] rj Zcrightri Zcleftr$ r$ r% �test_merge_categorical� s. z+TestMergeCategorical.test_merge_categoricalc C s� t tdddgdddgd�dddgd��}t tdddgdddgd�d d d gd ��}t||d gd�}t tdddg�dddgd d d gd��}t�||� d S )Nr� r� rM )r ZA0�B0ZC0)�Foo�LeftZC1�B1�A1)r �Rightr )rl )r r� r� )r r r rf rg )ra rJ rK ri rj r$ r$ r% �'tests_merge_categorical_unordered_equal� s z<TestMergeCategorical.tests_merge_categorical_unordered_equalc C s� |j |j�d�d�}t||dd�}|j�� }ttddgd�t� d�td d gd�gdd d gd �}t � ||� |j j �|j j �s�t�|jj �|jj �s�t�d S )NrJ )rw rZ )rl r+ r� )r ry r' r@ r[ rw )rZ )�assignrw rp r r� r r r r r) rf r� rZ r� Z#_categories_match_up_to_permutationr� )ra r_ r` r� ri rj r$ r$ r% �test_other_columns s z'TestMergeCategorical.test_other_columns�changec C s | S )Nr$ )r6 r$ r$ r% r7 / r8 z TestMergeCategorical.<lambda>c C s | � tdddg��S )Nr+ r� Zbah)rp rv )r6 r$ r$ r% r7 0 r8 c C s | � tdd��S )NT)�ordered)rp rv )r6 r$ r$ r% r7 1 r8 c C s� ||j �d��}|j|d�}t|j jj�s.t�t||d|d�}|j� � }t t �d�t �d�t �d�gdddgd �}t � ||� d S ) Nr1 )rZ rZ )rl ru ry r( r[ rw )rZ )rZ rp r� r r� r) r� r r� r r r rf r� ) ra r� r� r_ r` rZ r� ri rj r$ r$ r% � test_dtype_on_merged_different, s (z3TestMergeCategorical.test_dtype_on_merged_differentc s� d� t dddddddd d d g � d d dddgd � � fdd�dD �� fdd�dD �d��}|�dd� �}t||t|j�d�}t�||� d S )NrH rA rB rS rT rU r� r� r� r� �jr� �wr6 r� r� r@ c s$ g | ] }|gd � D ]}|�qqS )r@ r$ )�.0�each�letter)r� r$ r% � <listcomp>N s zKTestMergeCategorical.test_self_join_multiple_categories.<locals>.<listcomp>)r� r" �u�p�oc s g | ]}|g� D ]}|�qqS r$ r$ )r� r� r� )r� r$ r% r� S s ) ZaaZbb�cc�dd�ee�ffZgg�hh�iiZjj)rA rB rS rT c S s | � d�S )NrJ )rp )r6 r$ r$ r% r7 f r8 zITestMergeCategorical.test_self_join_multiple_categories.<locals>.<lambda>)rl )r �applyr r ry rf rg )ra r] ri r$ )r� r% �"test_self_join_multiple_categoriesE s z7TestMergeCategorical.test_self_join_multiple_categoriesc C s, t tddd�dgtddd�dggddgd�}|d �d �|d< t tddd�dgtddd �d ggdd gd�}|d �d �|d< t t�d ��� ddgt�d��� dtjgt�d��� tjd ggddd gd�}t||ddgd�}t� ||� t t�d ��� ddggddd gd�}t||ddgd�}t� ||� d S )Ni� r' g�������?r@ g�������?r Znum2)ry rJ rF gffffff�?Znum4z 2001-01-01z 2001-01-02z 2001-01-03r� )ru rl r| ) r r rp r- r. r r2 r rf rg )ra r] rK Zexpected_outerZ result_outerZexpected_innerZ result_innerr$ r$ r% � test_dtype_on_categorical_datesm s$ ** z4TestMergeCategorical.test_dtype_on_categorical_datesr� TFz.category_column,categories,expected_categoriesr@ r' �False�Truec C s� t ddddg|d��}|d �t||d��|d< t ddgddgd ��}|�|�}t ddg|ddgd ��}|d �t||d��|d< t�||� d S ) Nr' r@ rF rG )rl r) r) )r� r� )rl �num)rl r) r� )r rp rv r rf rg ) ra Zcategory_columnr Zexpected_categoriesr� rJ rK ri rj r$ r$ r% �/test_merging_with_bool_or_int_cateorical_column� s zDTestMergeCategorical.test_merging_with_bool_or_int_cateorical_columnc C s\ t tddtjgdd�dd��}t||dd�}t tddtjgdd�ddd��}t�||� d S ) Nr' r@ r* )r) )r� r� r� )rl )r� �B_xZB_y)r r r r2 r rf rg )ra r] ri rj r$ r$ r% �test_merge_on_int_array� s z,TestMergeCategorical.test_merge_on_int_arrayN)r� r� r� rz r{ r} r� r� r� rE rF r� r� r� r� r� r$ r$ r$ r% rx � s& +( (rx c C s t ddddgidddgd�S )NrA r� r� r r@ r' )rZ )r r$ r$ r$ r% r> � s r> c C s t ddddgidddgd�S ) NrB i, �d r] rF r' r@ )rZ )r r$ r$ r$ r% r? � s r? c @ sr e Zd Zej�dddeddgddgd�d d gd �fdd eddgddgd�d d gd �fd dedddgddejgd�d d dgd �fd d edddgejddgd�dd d gd �fddeejddgdddgd�dd d gd �fdd eddejgdddgd�d d dgd �fddedddejgejdddgd�dd d dgd �fdd edddejgejdddgd�dd d dgd �fg�dd� �Z dS )�TestMergeOnIndexeszhow, sort, expectedr| Fr� r� r] r� )rA rB r@ r' )rZ Tr_ r r` i, rF r� c C s$ t ||dd||d�}t�||� d S )NT)rd re ru rv )r rf rg )ra r> r? ru rv rj ri r$ r$ r% �test_merge_on_indexes� s 0z(TestMergeOnIndexes.test_merge_on_indexesN) r� r� r� r� rE rF r r r2 r� r$ r$ r$ r% r� � s8 ""&&$$ r� rZ r� r� � index_col)r r5 g �?g @)r5 r' r@ )�start�stopr5 z 2018-01-01z 2018-01-02c C s t | �jS )N)�typer� )r6 r$ r$ r% r7 � r8 )r: c C sb t dddgi| d�}t dddgi| d�}|j|dgd �}t ddgddgd �| d�}t�||� d S ) N� left_datar' r@ )rZ � right_datag �?g @r� )rl )r� r� )r r rf rg )rZ r_ r` ri rj r$ r$ r% �test_merge_index_types� s r� z-on,left_on,right_on,left_index,right_index,nmr� r| F)NNNTTr� T)NNNTTNc C s� t dddddgitjddgddggd d gd �d �}tddddgtjddgddggd d gd �|d �}t ddgddgd�tjddgdggd d gd �d �}|d k r�t||| ||||d�} t�| |� n4d} tjt | d�� t||| ||||d�} W d Q R X d S )Nr� r' r@ rF rG rA rB r r� r| )�names)rZ )rZ r5 )r� r� )rl rt rw rd re z$Cannot merge a Series without a name)r ) r r � from_productr r rf rg r� r� r� ) rl rt rw rd re �nmrA rB rj ri r� r$ r$ r% �test_merge_series s< " r� z!col1, col2, kwargs, expected_cols�suffixes)� �_dupr� Z0_dup)Nr� )�_x�_yZ0_xZ0_yr� r� rA )Nr� g )r� Nz0.0_xrB �b_yr- )Nr� )Z_aNZ0_ar. c C s� t | dddgi�}t |dddgi�}t ddgddgddgg|d�}|j|fddd �|��}t�||� t||fddd �|��}t�||� d S ) Nr' r@ rF rG rH r� )ry T)rd re )r r rf rg )rC r' �kwargsZ expected_colsrA rB rj ri r$ r$ r% �test_merge_suffix; s r� z how,expectedr� r] i, �< �F iX i� i )r� r� ZB2r= c C sb t dddgdddgd��}t dddgd d d gd��}t||d | d d�}d ddg|_t�||� d S )Nr� r] r' r� r� r= )r� r� i, iX i� i r� )r� r� )rl ru r� r� )r r ry rf rg )ru rj r> r? ri r$ r$ r% �test_merge_duplicate_suffix[ s r� zcol1, col2, suffixes)rA rA )NN)rA rA )r� N)r r )Nr� c C sX t | dddgi�}t |dddgi�}d}tjt|d�� t||dd|d � W d Q R X d S ) Nr' r@ rF rG rH z'columns overlap but no suffix specified)r T)rd re r� )r r� r� r� r )rC r' r� rA rB r� r$ r$ r% �test_merge_suffix_errory s r� )r_ r` c C sT t ddddgi�}t ddddgi�}t�t�� t||ddd d hd � W d Q R X d S ) NrA r' r@ rF rB rG rH Tr_ r` )rd re r� )r rf r_ � FutureWarningr )r� rA rB r$ r$ r% �test_merge_suffix_warns� s r� zcol1, col2, suffixes, msg)rA rA )rA rB rS z(too many values to unpack \(expected 2\)z1not enough values to unpack \(expected 2, got 1\)c C sT t | dddgi�}t |dddgi�}tjt|d�� t||dd|d� W d Q R X d S ) Nr' r@ rF rG rH )r T)rd re r� )r r� r� r� r )rC r' r� r� rA rB r$ r$ r% � test_merge_suffix_length_error� s r� � cat_dtyperq rr �reversec C s� t dddgdd�t dddgdd�d�}ttdddg��|d �dd d gd ���d �}dddg}dd d g}|r||�� |�� tt|��|| �|d ���d �}|j|ddd�}tdd d gdd d gtdddg��|d �d���d �}t�||� d S )NrA rB rS F)r r� )rq rr rq r' r@ rF )r+ r_ r+ )r+ r` T)rd re )r_ r` r+ ) r r r rp rr r� r rf rg ) r� r� Z cat_dtypesrJ Zdata_fooZ data_rightrK ri rj r$ r$ r% �test_merge_equal_cat_dtypes� s( & r� c C s� t dddgdd�} ttddg��| �ddgd���d �}ttdddg��| �d ddgd ���d �}|j|d d d �}tddgd dgtddg��| �d���d �}t�||� d S )NrA rB rS F)r r� r' r@ )r+ r_ r+ rF )r+ r` T)rd re )r_ r` r+ )r r r rp rr r rf rg )r� rJ rK ri rj r$ r$ r% � test_merge_equal_cat_dtypes2� s "$r� c C sj t dtt�dd�t�dd�gdd�i�} | �� }|d �d�|d< t|| ddd �}| �� }t�||� d S ) NrA r r' r@ �interval)r) rJ r| )ru rl ) r r r- ZIntervalr� rp r rf rg )r` r_ ri rj r$ r$ r% � test_merge_on_cat_and_ext_array� s &r� c s� ddddg} dddg}t j| |fdd gd �}t|d �}d |d < t|d �}d |d < d� d�|j|d � �fd�}� fdd�| D ��fdd�| D � }t j||gdd gd �}t|d �}d |d < t�||� d S )NrA rB rS rT rg rh �3r� r| )r� )ry r� rl r� r� )rl r� c s g | ] }|� �qS r$ r$ )r� r� )�l_sufr$ r% r� � s z1test_merge_multiindex_columns.<locals>.<listcomp>c s g | ] }|� �qS r$ r$ )r� r� )�r_sufr$ r% r� � s )r r� r r rf rg ) �letters�numbersrZ Zframe_xZframe_yri Zexpected_labelsr6 rj r$ )r� r� r% � test_merge_multiindex_columns� s" r� c C s� t dddgdddgd��} t dddgt�d d d g�d ��}t| |d dd�}t dddgdddgt�d d dg�d��}t�||� d S )NrA rB rS rg rh �4)r6 r� r� �2000Z2001Z2002)r� r� r_ r� )ru rl r3 )r6 r� r� )r r- r r rf rg )rJ rK ri rj r$ r$ r% � test_merge_datetime_upcast_dtype s r� � n_categoriesrH � c C s� t dgtt| �� �}tt| d �dg|d�}tdggdgt dgt�| �d�d�}t||dddd �}td d � t| d �D �d d g|d�}t�||� d S )Nr r' rR )ry rZ r� )r r_ T)ru rd re c S s& g | ] }|d k r|dgn|t jg�qS )r@ g @)r r2 )r� r� r$ r$ r% r� ) s z9test_categorical_non_unique_monotonic.<locals>.<listcomp>r� r� ) r r r r r r� r rf rg )r� rd rJ rK ri rj r$ r$ r% �%test_categorical_non_unique_monotonic s r� c C s� t ddddddgdddg�ddddddgd�} t| �} t ddddddgdddg�ddddddgddd d d d gd �}t|��ddg�d }t| |�� ddgddgdd�}|jddgdd�}| j|ddgd�}t�||� t ddddddgdddgdd�ddddddgd�} t| �} t ddddddgdddgdd�ddddddgddd d d d gd �}t|��ddg�d }t| |�� ddgddgdd�}|jddgdd�}| j|ddgd�}t�||� d S )NrA rB rS r r' )�Cat1�Int1g�������?g333333�?g�������?gffffff�?g �?g�������?)�Cat�Int�Factorr� r� r� r� r� r_ )rt rw ru )Zaxis)rl T)r� ) r r rr r r� Zdropr} rf rg )rA rB rj ri r$ r$ r% �&test_merge_join_categorical_multiindex0 sD r� �funcr Z merge_asof)r� �err_msg)rt rd rt rd )rw re rw re c C sx t ddgddgd��}t ddgddgd��}d |d � d |d � d �}tjt|d �� tt| �||f|� W d Q R X d S )Nr' r@ rF rG )rA rB rH r� )rA rS zCan only pass argument "r z" OR "z " not both\.)r )r r� r� r �getattrr- )r� r� r� r_ r` r� r$ r$ r% �/test_merge_join_cols_error_reporting_duplicatesh s r� c C sx t ddgddgd��}t ddgddgd��}d |d � d |d � d �}tjt|d �� tt| �||f|� W d Q R X d S )Nr' r@ rF rG )rA rB rH r� )rA rS z Must pass "r z" OR "z"\.)r )r r� r� r r� r- )r� r� r� r_ r` r� r$ r$ r% �,test_merge_join_cols_error_reporting_missingy s r� r� c C sj t ddgddgd��}t ddgddgd��}d }tjt|d ��" tt| �||fd d i|�� W d Q R X d S ) Nr' r@ rF rG )rA rB rH r� )rA rS zZCan only pass argument "on" OR "left_index" and "right_index", not a combination of both\.)r rl rA )r r� r� r r� r- )r� r� r_ r` r� r$ r$ r% �1test_merge_join_cols_error_reporting_on_and_index� s r� c C sv t ddgddgd��} t ddgddgd��}t| |dddd�}t ddgddgddgddgddgd��}t�||� d S ) Nr' r+ )r6 r� r` Tr6 )ru rd rw )r6 Zx_xZz_xZx_yZz_y)r r rf rg )r_ r` ri rj r$ r$ r% �test_merge_right_left_index� s r� c C s� t dgdgd���ddg�} t ddgi��dg�}t g g d�tjd��ddg�}t| |dgdd�}t�||� t|| ddgd �}t�||� d S ) Nr' r@ )rA rB rA rB )r) T)rt re )rd rw )r rr r r( r rf rg )rJ rK rj ri r$ r$ r% �$test_merge_result_empty_index_and_on� s r� c C s� t dddgddd��} t dddgdd��}t ddddgddddgddddggddddgd �}t�t�� t| |dd �}W d Q R X t�||� t�t�� t|| dd d � W d Q R X t�||� d S ) Nr' r@ rF )rA rB �b_x)rA rB rA r� r� )ry )rl )r� r� )rl r� )r rf r_ r� r rg )r_ r` rj ri r$ r$ r% �-test_merge_suffixes_produce_dup_columns_warns� s 0 r� c C sp t dddgdddggdddgd�} t ddgdd��}t| |dd�}t ddddggdd d d gd�}t�||� d S ) Nr' r@ rA rB )ry rF )rA rB )rl r� r� )r r rf rg )r_ r` ri rj r$ r$ r% �3test_merge_duplicate_columns_with_suffix_no_warning� s " r� c C s� t ddddgddddggddddgd�} t ddgdd��}t�t�� t| |dd �}W d Q R X t dddddggddddd gd�}t�||� d S ) Nr' r@ rA rB r� )ry rF )rA rB )rl r� )r rf r_ r� r rg )r_ r` ri rj r$ r$ r% �Btest_merge_duplicate_columns_with_suffix_causing_another_duplicate� s ( "r� c C s� t ddgddggt�ddg�d�} t dd gd d ggd d gd�}t|| dddd�}t dd ddgd d ddggt�d d ddg�d�}t�||� d S )Nr' r@ rF rG rA g �\@)ry r� r� r� r� r6 r� r| T)ru rd re )r r- r� r rf rg )rJ rK ri rj r$ r$ r% �%test_merge_string_float_column_result� s " ,r� c C sX t dgdgdgd�} t dgdgdgd�}tjtdd�� t| |dgdd � W d Q R X d S ) NrZ rM � )�datary rZ i� zCan only pass argument)r T)rl rd )r r� r� r r )Zdf_1Zdf_2r$ r$ r% �/test_mergeerror_on_left_index_mismatched_dtypes� s r� r) r* c C s� t ddgddgd�| d�}t tjtjgddgd�| d�}t||dd d �}t ddtjtjgddtjtjgtjtjddgd �| d�}t�||� t||dd d �}t tjtjddgddtjtjgtjtjddgd �| d�}t�||� d S ) Nr' r@ )rW rC )r) rF rG )rW r' rW r� )rl ru )rW rC r' )rW r' rC )r r r2 r rf rg )r) r_ r` ri rj r$ r$ r% �test_merge_outer_with_NaN� s r� )Vr r r r ri �numpyr r� Zpandas.core.dtypes.commonr r Zpandas.core.dtypes.dtypesr Zpandasr- r r r r r r r r r r r r r Zpandas._testingZ_testingrf Zpandas.api.typesrv Zpandas.core.reshape.concatr Zpandas.core.reshape.merger r r\ r^ r& r/ r4 Zfixturer= r> r? rL rM r� rH r_ r` rx r> r? r� rE rF r� r� r� r2 r� r� r� �tupler� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r$ r$ r$ r% �<module> s� < & t7 + " " $ $ 8
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка