Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/merge/__pycache__/test_multi.cpython-37.pyc
Назад
B �5�g�w � @ s� d dl Zd dlZd dlZd dlmZmZmZmZm Z d dl m Z d dl mZ d dlmZ ejdd� �Zejdd� �Zejd d � �Zejd d � �Zejd d� �Zejdd� �ZG dd� d�ZG dd� d�ZdS )� N)� DataFrame�Index� MultiIndex�Series� Timestamp)�concat)�mergec C sP ddddddddddg } ddddddddddg }t j�t| ��}t| ||d ��S ) z=left dataframe (not multi-indexed) for multi-index join tests�bar�foo�baz�qux�snap�two�one�three)�key1�key2�data)�np�random�randn�lenr )r r r � r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/merge/test_multi.py�left s r c C sn t ddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd�} ttj�dd �| dddgd�S )z:right dataframe (multi-indexed) for multi-index join testsr r r r r r r r � � � r r )�levels�codes�names� �j_one�j_two�j_three)�index�columns)r r r r r )r% r r r �right s . r' c C sb t dddddgdddddgdddddgddd ddgd d d d dgd�dddddgd��ddddg�S )N�A�B�C�AM�IP�OPZhbwZnhbZhboi� i? i� i� i\ )�Origin� Destination�Period�TripPurp�Tripsr. r/ r0 r1 r2 )r& )r � set_indexr r r r � left_multi* s r4 c C sv t dddddddgdddddddgdddddddgd d d d d d d gd d dd dddgd�dddddgd��ddddg�S )Nr( r) r* �E�Fr+ r, r- �a�b�c�d �P �Z �K �# �7 )r. r/ r0 �LinkType�Distancer. r/ r0 r@ rA )r& )r r3 r r r r � right_multi8 s rB c C s dddgS )Nr. r/ r0 r r r r r � on_cols_multiF s rC c C s dddddgS )Nr. r/ r0 r1 r@ r r r r r �idx_cols_multiK s rD c @ s e Zd Zdd� Zdd� Zej�dddg�dd � �Zej�dddg�d d � �Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zej�ddejeeg�d d � �Zej�d d d g�d!d"� �Zejd#d$� �Zejd%d&� �Zejd'd(� �Zd)d*� Zd+d,� Z d-d.� Z d/d0� Z d1d2� Z dS )3�TestMergeMultic C s� t ddddgdddggdddd d d d d d d g dd d dd d d dd d g gd d gd�| _ttj�dd �| jdddgd�| _ddddddddddg }ddddddddddg }tj�t|��}t|||d��| _d S )Nr r r r r r r r r r r �first�second)r r r r! r" r# r$ )r% r& r )r r r ) r r% r r r r Zto_joinr r )�selfr r r r r r � setup_methodQ s* . zTestMergeMulti.setup_methodc C s� ddg}|j |||d�jdd�}t||�� ||d�}t�||� |j |||dd�jdd�}t||�� ||dd�}t�||� d S )Nr r )�on�howT)�drop)rJ rK �sort)�join� reset_indexr �tm�assert_frame_equal)rH r r' � join_type�on_cols�result�expectedr r r �test_merge_on_multikeyo s z%TestMergeMulti.test_merge_on_multikeyrM FTc sh dddg�dd� � � �fdd�}t ttt�td�td �d ���}ttj�|d �ddgd �}|� d dtj� d dt |��� tj� t |��}|j | �� }� |�|d< � |� |d< |j�dd� ||||� tj|jd d d�df< tj|jdd d�df< tj|jdd d�df< � |�|d< tj� t |��}|j |d d�f }� |� |d< |j�dd� ||||� d S )N�1st�2nd�3rdc sB dd� � � fdd�}|| d �|| d �d | d � d�d S ) Nc S s | | kr dS t | �S )Nr )�ord)r7 r r r �<lambda>� � zNTestMergeMulti.test_left_join_multi_index.<locals>.bind_cols.<locals>.<lambda>c s | � � �td� S )Nr7 )�maprZ )�ts)�iordr r r[ � r\ rW rY g Y@rX r g ��@)�fillna)�df�fr )r_ r � bind_cols� s z<TestMergeMulti.test_left_join_multi_index.<locals>.bind_colsc s� | j |�d|d�}t| �t|�d k s*t�|d �� �� r>t�|d �� �� rRt�tj|d |d dd� � |jd d �d d�f �}tj|d |dd� |jd ks�t�|r�t� ||j �d d �� t | |� � �|dd �}t �t|��|_t� ||� d S ) Nr )rJ rK rM r �4th�5thF)Z check_names���� mergesort)�kind)rJ rM rK )rN r �AssertionErrorZisna�anyrP Zassert_series_equal�iloc�namerQ � sort_valuesr rO r �aranger% )r r' rM �resrT �out)rc �icolsr r � run_asserts� s z>TestMergeMulti.test_left_join_multi_index.<locals>.run_assertsr7 �zr )i� r )r& r i� rd re T)�inplace� r �% r �+ ���)�listr] �chrr rn rZ r r �choice�insert�randintr Z permutationrk �copyr3 �nan�loc)rH r r' rM rr �lc�ir )rc rq r �test_left_join_multi_index� s* " z)TestMergeMulti.test_left_join_multi_indexc C sJ ddg}|j ||dd|d�}|j ||dd|d�}||j }t�||� d S )Nr r Tr )Zleft_onZ right_indexrK rM r' )Zright_onZ left_indexrK rM )r r& rP rQ )rH r r' rM rS Zmerged_left_rightZmerge_right_leftr r r �test_merge_right_vs_left� s z'TestMergeMulti.test_merge_right_vs_leftc C s� t td�tjddgdddggddgd �d d �}ttd �ddddddgd d��}t||�� ddgd�}ttd �ddddddgd gd td�d��}t� ||� d S )N� r( r) r r r �lev1�lev2)r �Amount)rl ZAAABBBr )r� r� �col)rJ )r� r� r� r� ) r �ranger � from_productr ry r rO rP rQ )rH �sra rT rU r r r �.test_merge_multiple_cols_with_mixed_cols_index� s z=TestMergeMulti.test_merge_multiple_cols_with_mixed_cols_indexc C s| t �dd�}t�|d�}|d d d� }t||tj�d�d��}t|d d d� |d d d� tj�d�d��}t||dd � d S ) Nr! i' r rx i N )r r Zvalue1)r r Zvalue2�outer)rK )rP Z rands_arrayr Ztiler r r r )rH r r ra Zdf2r r r � test_compress_group_combinations� s *z/TestMergeMulti.test_compress_group_combinationsc C s� ddg}t dddgd ddgd tjt�d �tjd �d ��}t�d dg�}t dddgi|d�}|j||d�}|�� }tj |d< d|j |j dk|j dk@ df< d|j |j dk|j dk@ df< t �||� |j|ddd� |j||dd�}t �||� t dddgd ddgd tjdddgd tjd �tjt�d �tjd �d��}t�d dg�}t dddgi|d�}|j||d�}|�� }tj |d< d|j |j dk|j dk@ df< d|j |j dk|j dk@ df< t �||� |j|dd�}|j||dd�}t �||� d S )N�k1�k2r r r � r r � � )Zdtype)r� r� �v)r r )r r �v2� � )r% )rJ rg T)rh rt )rJ rM )r� r� Zk3r� )rh )r r �arrayrn Zint64r � from_tuplesrN r~ r r� r� r� rP rQ rm Zfloat32Zint32)rH rS r r% r' rT rU r r r �#test_left_join_index_preserve_order� s@ z2TestMergeMulti.test_left_join_index_preserve_orderc C sx t ddddgddddgddd d gdd d d gddd dgddd dgddddgdd ddgddddgddddgg ddddgdddd d d d d d!d"g d#�}t dd ddgddddgdddd"gddd d gddd d gdddd gdddd gdddd!gdddd$gdd dd%gdddd gdd d dgdd d d&gddd d'ggd(d)d*d+gd,��d(d)d*g�}|j|dddgd-d.�}t ddddd gddddd!gddddtjgddd d d'gdd d d dgdd d d d&gddd dtjgddd dd gddd dd gdddddgddddd"gdd dddgddddd gddddd ggddddd+gddddd d d d d d d d d!d"gd#�}t�||� |j|dddgd-d/d0�}|jdddgd1d2�}t�||� d S )3N�X�Yr* r7 �W�e�V�Qr( �h�R�Dr� r8 r9 r) rb �g�j�dZcolaZcolbZcolc�tagr r r r r� r� � r� � r� )r& r% i����i����rx ���Zcol1Zcol2Zcol3�val)r& r )rJ rK T)rJ rK rM rg )rh )r r3 rN r r rP rQ rm )rH r r' rT rU r r r �+test_left_join_index_multi_match_multiindex# sh $ z:TestMergeMulti.test_left_join_index_multi_match_multiindexc C s\ t ddgddgddgddggdd gddddgd �}t dd gdd gdd gddgddgddgddgddggddgd��d�}|j|ddd�}t ddd gddd gdddgdddgddtjgddd gdddgddtjggdd dgddddddddgd �}t�||� |j|dddd�}|jddd �}t�||� t||� � ddd �}t� t |��|_ t�||� d S ) Nr9 r r8 r r7 r r r� r� )r& r% r� �w�xr� �yrs �rr� �qr� �char)r& r )rJ rK T)rJ rK rM rg )rh )rK rJ ) r r3 rN r r rP rQ rm r rO rn r r% )rH r r' rT rU Z expected2r r r � test_left_join_index_multi_matchh sD z/TestMergeMulti.test_left_join_index_multi_matchc C s� t td�tj�d�tj�d�td�tj�d�d�dddddgd �}t d d tjtjtjgd d tjtjtjgd��}t||ddd�}|jdgdd�}|�|�}t � ||� d S )NZabcder� )�id�v1r� �dummy�v3r� r� r� r� r� )r& r7 r8 gX9��v��?g��ʡE�@)r� Zsv3r )rJ rK r )�axis) r ry r r r r r rL rN rP rQ )rH r r' rT �rdfrU r r r �test_left_merge_na_buglet� s z(TestMergeMulti.test_left_merge_na_bugletc C s� dddgdddgdddgddt jgdddgdd dgdd t jgd d d gdd dgg }t|d d dgd�}ddt jgddt jgddt jgd dt jgd dt jgdd t jgg}t|d d dgd�}|j|dd�}|�d�j|�d�dd�}|�dt j�}t�||� d S )Ni� r( g �?r) i� i� i� g @r* i� g @�year�panelr )r& r� )rK i���)r r r r r` �replacerP rQ )rH r �frameZ other_data�otherrT rU r r r �test_merge_na_keys� s* z!TestMergeMulti.test_merge_na_keys�klassNc C s� t dddgdddgdgd�}t�|j�|_|jj}|d k rB||�}t dddgd d d gd ��}|j|d|gd d�}t�||� t d d d gdddgdddgd��}|j||jjgd d�}t�||� d S )Nr r r z 2016-01-01z 2017-01-01z 2018-01-01r7 )r& i� i� i� )r7 �key_1�inner)rJ rK )�key_0�a_x�a_y)r �pd� to_datetimer% r� r rP rQ )rH r� ra � on_vectorrU rT r r r �test_merge_datetime_index� s z(TestMergeMulti.test_merge_datetime_index� merge_typer r' c C s t dddgitjtd�dgtd�dggddgd �d �}t tjg ddgd �d gd �}tjtd�dgtd�dggddgd �}|d kr�t ddgd d gd�|d �}|j|d ddgd�}|j|d d�}n<t d d gddgd�|d �}|j|dddgd�}|j|dd�}t�||� t�||� d S )Nr g �?z 1950-01-01r( z 1950-01-02r) �dater� )r )r r% �state)r% r& r )r r� )rK rJ )rK )r� r r' )r r r� r r rN rP rQ )rH r� r r' Zexpected_indexrU Z results_mergeZ results_joinr r r �(test_merge_datetime_multi_index_empty_df� s2 z7TestMergeMulti.test_merge_datetime_multi_index_empty_dfc C s6 t dddgdddgdddgd�d d d gd ��d �}|S ) Nr r r r gffff��Ag����PAi^ )� household_id�male�wealthr� r� r� )r& )r r3 )rH � householdr r r r� s zTestMergeMulti.householdc C sh t dddddddgdddddd tjgd d d d d dtjgdddddddgd�ddddgd��ddg�}|S )Nr r r r� � nl0000301109� nl0000289783� gb00b03mlx29� lu0197800237� nl0000289965zABN Amro�RobecozRoyal Dutch Shellz AAB Eastern Europe Equity FundzPostbank BioTech Fondsg �?g�������?g333333�?g333333�?g �?)r� �asset_idrl �sharer� r� rl r� )r& )r r r r3 )rH � portfolior r r r� % s( zTestMergeMulti.portfolioc C s| t ddddddgddddddgddddd d gd d d dd dgddddddgddddddgd���ddg�jddd d gd �}|S ) Nr r gffff��Ag����PAg x�AzABN Amror� zRoyal Dutch Shellz AAB Eastern Europe Equity FundzPostbank BioTech Fondsg �?g�������?g333333�?g333333�?g �?r r r� r� r� r� r� )r� r� rl r� r� r� r� r� r� r� rl r� )r& )r r3 �reindex)rH rU r r r rU B s2 zTestMergeMulti.expectedc C s. |� � }|� � }|j|dd�}t�||� d S )Nr� )rK )r~ rN rP rQ )rH r� r� rU rT r r r �test_join_multi_levelsi s z%TestMergeMulti.test_join_multi_levelsc C sD |� � }|� � }t|�� |�� dgdd��ddg�}t�||� d S )Nr� r� )rJ rK r� )r~ r rO r3 rP rQ )rH r� r� rU rT r r r �(test_join_multi_levels_merge_equivalencer s z7TestMergeMulti.test_join_multi_levels_merge_equivalencec C sn |� � }|� � }|j|dd�}t|tddgitjdtjfgddgd�d �gd d d �j|j d �}t � ||� d S )Nr� )rK r� g �?r� r� r� )r )r% r T)r� rM )r& ) r~ rN r r r r� r r r� r& rP rQ )rH r� r� rU rT r r r � test_join_multi_levels_outer s z+TestMergeMulti.test_join_multi_levels_outerc C s� |� � }|� � }d|j_tjtdd�� |j|dd� W d Q R X |� � }|j�ddg� tjtdd�� |j|dd� W d Q R X d S )Nr z+cannot join with no overlapping index names)�matchr� )rK r� z'columns overlap but no suffix specified)r~ r% rl �pytestZraises� ValueErrorrN Z set_names)rH r� r� Z portfolio2r r r � test_join_multi_levels_invalid� s z-TestMergeMulti.test_join_multi_levels_invalidc C s t dddddddgddddddtjgd d d d d d d gd�dddgd��ddg�}t dddddgdddddgddddd gd ���dd g�}t ddddddddgddddddddgddddddddgd d d d d d d d gdddddddd gd ���ddd g�jdd gd�}t|�� |�� dgd!d"��ddd g�}t�||� t ddddddddddddg dddddddddddd g d d ddddddddd d g d d d d d d d d d d d d g d d dddddddd d d g d ���ddd g�jdd gd�}t|�� |�� dgd#d"��ddd g�}t�||� d S )$Nr r r r� r� r� r� r� g �?g�������?g333333�?g333333�?g �?)r� r� r� r� r� r� )r& �� �� �� � � g�S�緖�?g�D������g �>���?g� ��?gH�RE�?)r� �t� log_returnr� )r� r� r� r� r� r� r� )rJ rK r� ) r r r r3 r� r rO rP rQ )rH r� r� rU rT r r r �test_join_multi_levels2� s� z&TestMergeMulti.test_join_multi_levels2) �__name__� __module__� __qualname__rI rV r� �mark� parametrizer� r� r� r� r� r� r� r� r� r �asarrayr r r� r� �fixturer� r� rU r� r� r� r� r� r r r r rE P s* 77E2 / ' rE c @ sL e Zd Zdd� Zdd� Zej�ddej e e g�dd� �Z d d � Z d d � ZdS ) �TestJoinMultiMultic C sD t |�� |�� ||d��|��� }|j||d��� }t�||� d S )N)rK rJ )rK )r rO r3 � sort_indexrN rP rQ )rH r4 rB rR rC rD rU rT r r r �test_join_multi_multiF s z(TestJoinMultiMulti.test_join_multi_multic C s` |j |jd�}|j |jd�}t|�� |�� ||d��|��� }|j||d��� }t�||� d S )N)r& )rK rJ )rK ) rL r& r rO r3 r� rN rP rQ )rH r4 rB rR rC rD rU rT r r r � test_join_multi_empty_framesX s z/TestJoinMultiMulti.test_join_multi_empty_frames�boxNc C s� t dddgdddgdgd�}t�|j�|_|jj}|d k rB||�}t dddgd d d gd ��}|j|d|gd d�}t�||� t d d d gdddgdddgd��}|j||jjgd d�}t�||� d S )Nr r r z 2016-01-01z 2017-01-01z 2018-01-01r7 )r& i� i� i� )r7 r� r� )rJ rK )r� r� r� )r r� r� r% r� r rP rQ )rH r� ra r� rU rT r r r r� m s z,TestJoinMultiMulti.test_merge_datetime_indexc C s� t jdddgddgd�}tddd gd d d gd �|d�}t jddddgddgd�}tddddgddddgd �|d�}|�|�}t|�� |�� dgd d ��dddg�}t�||� d S ) N)�K0ZX0)r� ZX1)�K1ZX2�keyr� )r ZA0�A1�A2�B0ZB1ZB2)r( r) )r% )r� ZY0)r� ZY1)�K2ZY2)r� ZY3r� ZC0ZC1ZC2ZC3ZD0ZD1ZD2ZD3)r* r� r� )rJ rK ) r r� r rN r rO r3 rP rQ )rH Z index_leftr Z index_rightr' rT rU r r r �test_single_common_level� s z+TestJoinMultiMulti.test_single_common_levelc C s� t jddgddggddgd�}t jdd d gddgd�}t|d d d ddgid�}t|ddddgid�}|�|�}t|d d ddgdddtjgd�d�}t�||� d S )Nr r r r� r7 r8 )r )r� r )r r )r r r� r! � � �( )r% r r� r r Zfing)r� r� ) r r� r� r rN r r rP rQ )rH Zmidx1Zmidx3r r' rT rU r r r �test_join_multi_wrong_order� s z.TestJoinMultiMulti.test_join_multi_wrong_order)r� r� r� r� r� r� r� r� r r� r r r� r r r r r r r� E s r� )�numpyr r� Zpandasr� r r r r r Zpandas._testingZ_testingrP Zpandas.core.reshape.concatr Zpandas.core.reshape.merger r� r r' r4 rB rC rD rE r� r r r r �<module> s& z
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка