Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/merge/__pycache__/test_merge_asof.cpython-37.pyc
Назад
B �5�g�� � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZm Z m Z m Z d dl m Z d dlmZ G dd� d�Zej�dddid d d �g�ej�d d d gdd gg�dd� ��Zdd� Zej�dd d d�dd d�d dd�g�dd� �Zdd� Zd d � Zd d � Zd d!� ZdS )"� N)�Index� Timedelta� merge_asof�read_csv� to_datetime)� MergeErrorc @ s< e Zd Zd�dd�Zejdd�dd� �Zdd � 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'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z ej j d8e!d9�e"j#d:d;�gd<d=gd>�d?d@� �Z$dAdB� Z%dCdD� Z&dEdF� Z'dGdH� Z(dIdJ� Z)dKdL� Z*dMdN� Z+dOdP� Z,dQdR� Z-dSdT� Z.dUdV� Z/dWdX� Z0dYdZ� Z1d[d\� Z2d]d^� Z3d_d`� Z4dadb� Z5dcdd� Z6dedf� Z7dgdh� Z8didj� Z9dkdl� Z:dmdn� Z;ej j dodpdq� drdq� gdsdtgd>�ej � dudvdwg�dxdy� ��Z<dzd{� Z=d|d}� Z>d~d� Z?d�d�� Z@d�d�� ZAd�d�� ZBd�S )�� TestAsOfMergeFc C sD |ddd|�}t |�}|r4|jddgdd�jdd �}t|j�|_|S ) NZreshape�merge�data�time�ticker�last)ZkeepT)�drop)r Zdrop_duplicates� reset_indexr r )�self�datapath�name�dedupe�path�x� r �`/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/merge/test_merge_asof.py� read_data s zTestAsOfMerge.read_dataT)Zautousec C s\ | � |d�| _| j |ddd�| _| � |d�| _| � |d�| _| � |d�| _| � |d�| _d S ) Nz trades.csvz quotes.csvT)r zasof.csvz tolerance.csvzallow_exact_matches.csvz%allow_exact_matches_and_tolerance.csv)r �trades�quotes�asof� tolerance�allow_exact_matches�!allow_exact_matches_and_tolerance)r r r r r � setup_method s zTestAsOfMerge.setup_methodc C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdd d gd ��}t||dd�}t�||� dS )zdoc-string examples� � � �a�b�c)r# �left_val� � � � )r# � right_val)r# r& r+ )�onN)�pd� DataFramer �tm�assert_frame_equal)r �left�right�expected�resultr r r �test_examples1* s $ z TestAsOfMerge.test_examples1c C sv t jtdddddg�dddddgdddd d gd d d d d gd�ddddgd�}t jtddddddddg�ddddddddgdddddd dd gd d d d!d d"d#d$gd%�ddd&d'gd�}t||ddd(� t||ddtd)�d*� t jtdddddg�dddddgdddd d gd d d d d gtjdtjtjtjgtjd tjtjtjgd+�ddddd&d'gd�}t||ddtd,�d-d.�}t�||� d/S )0zdoc-string examplesz20160525 13:30:00.023z20160525 13:30:00.038z20160525 13:30:00.048�MSFT�GOOG�AAPLg������I@g\��(��@g��(\��@g �X@�K � �d )r r �price�quantityr r r< r= )�columnsz20160525 13:30:00.030z20160525 13:30:00.041z20160525 13:30:00.049z20160525 13:30:00.072z20160525 13:30:00.075g ��@g\��(�I@g ��Q��I@g��(\X@g�z�GJ@g= ףp��@g{�G�I@g= ףp�I@g J@gq= ף�X@gףp= ��@g�p= �J@)r r �bid�askr? r@ )r, �byZ2ms)r, rA r )r r r< r= r? r@ �10msF)r, rA r r N) r- r. r r r �np�nanr/ r0 )r r r r3 r4 r r r �test_examples26 sn z TestAsOfMerge.test_examples2c C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdd tjgd ��}t||ddd�}t�||� dS )zdoc-string examplesr r! r" r# r$ r% )r# r& r' r( r) r* )r# r+ )r# r& r+ �forward)r, � directionN)r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �test_examples3� s $"z TestAsOfMerge.test_examples3c C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdd d gd ��}t||ddd�}t�||� dS )zdoc-string examplesr r! r" r# r$ r% )r# r& r' r( r) r* )r# r+ )r# r& r+ �nearest)r, rG N)r- r. r r/ r0 )r r1 r2 r3 r4 r r r �test_examples4� s $ z TestAsOfMerge.test_examples4c C s2 | j }| j}| j}t||ddd�}t�||� d S )Nr r )r, rA )r r r r r/ r0 )r r3 r r r4 r r r � test_basic� s zTestAsOfMerge.test_basicc C sd | j }| j�� }|j�d�|_| j�� }|j�d�|_|j�d�|_t||ddd�}t�||� d S )N�categoryr r )r, rA ) r r �copyr �astyper r r/ r0 )r r3 r r r4 r r r �test_basic_categorical� s z$TestAsOfMerge.test_basic_categoricalc C sL | j }| j�d�}| j}t||dddd�}|j|_||j }t�||� d S )Nr Tr )� left_index�right_onrA ) r r � set_indexr r �indexr> r/ r0 )r r3 r r r4 r r r �test_basic_left_index� s z#TestAsOfMerge.test_basic_left_indexc C s: | j }| j}| j�d�}t||dddd�}t�||� d S )Nr Tr )�left_on� right_indexrA )r r r rR r r/ r0 )r r3 r r r4 r r r �test_basic_right_index� s z$TestAsOfMerge.test_basic_right_indexc C sF | j �d�}| j�d�}| j�d�}t||dddd�}t�||� d S )Nr Tr )rP rV rA )r rR r r r r/ r0 )r r3 r r r4 r r r �!test_basic_left_index_right_index� s z/TestAsOfMerge.test_basic_left_index_right_indexc C s� | j �ddg�}| j�d�}tjtdd�� t||ddd� W d Q R X | j �d�}| j�ddg�}tjtdd�� t||ddd� W d Q R X d S ) Nr r< z left can only have one index)�matchT)rP rV r? z right can only have one index)r rR r �pytest�raisesr r )r r r r r r �test_multi_index� s z TestAsOfMerge.test_multi_indexc C s� | j �d�}| j�d�}d}tjt|d�� t||dddd� W d Q R X | j �d�}| j�d�}d}tjt|d�� t||dddd � W d Q R X d S ) Nr z:Can only pass argument "left_on" OR "left_index" not both.)rY r< T)rU rP rV z<Can only pass argument "right_on" OR "right_index" not both.r? )rQ rP rV )r rR r rZ r[ r r )r r r �msgr r r �test_on_and_index� s z TestAsOfMerge.test_on_and_indexc C s4 | j }| j}| j}t||dddd�}t�||� d S )Nr r )r, �left_by�right_by)r r r r r/ r0 )r r3 r r r4 r r r �test_basic_left_by_right_by s z)TestAsOfMerge.test_basic_left_by_right_byc C sZ | j }| j}| j}||jdk }t||ddd�}tj|j|jdkddgf< t� ||� d S )Nr6 r r )r, rA r? r@ ) r r r r r rC rD �locr/ r0 )r r3 r r �qr4 r r r �test_missing_right_by s z#TestAsOfMerge.test_missing_right_byc C s` t jtdddddg�dddddgdd d d d gd d d d dgdddddgd�dddddgd�}t jtdddddd g�ddddddgd d ddd dgd d d d d d!gd d"d#d$d%d&gd'�dddd(d)gd�}t jtdddddg�dddddgdd d d d gd d d d dgdddddgtjd d d tjgtjd"d%d tjgd*�dddddd(d)gd�}t||dddgd+�}t�||� d S ),Nz20160525 13:30:00.023z20160525 13:30:00.046z20160525 13:30:00.048z20160525 13:30:00.050r6 r7 r8 �ARCA�NSDQ�BATSg������I@g\��(��@g��(\��@g �X@r9 r: r; )r r �exchr< r= r r rh r< r= )r> z20160525 13:30:00.030z20160525 13:30:00.041z20160525 13:30:00.045z20160525 13:30:00.049g�G�z��@g\��(�I@g ��Q��I@g ��@g��(\X@g{�G�I@g= ףp�I@g J@g= ףp��@gq= ף�X@)r r rh r? r@ r? r@ )r r rh r< r= r? r@ )r, rA )r- r. r rC rD r r/ r0 )r r r r3 r4 r r r � test_multiby sR zTestAsOfMerge.test_multibyc C s` t jtdddddg�dddddgdd d d d gd d d d dgdddddgd�dddddgd�}t jtdddddd g�ddddddgd d ddd dgd d d d d d!gd d"d#d$d%d&gd'�dddd(d)gd�}t jtdddddg�dddddgdd d d d gd d d d dgdddddgtjd d d tjgtjd"d%d tjgd*�dddddd(d)gd�}t||dddgd+�}t�||� d S ),Nz20160525 13:30:00.023z20160525 13:30:00.046z20160525 13:30:00.048z20160525 13:30:00.050r r r' re rf rg g������I@g\��(��@g��(\��@g �X@r9 r: r; )r r rh r< r= r r rh r< r= )r> z20160525 13:30:00.030z20160525 13:30:00.041z20160525 13:30:00.045z20160525 13:30:00.049g�G�z��@g\��(�I@g ��Q��I@g ��@g��(\X@g{�G�I@g= ףp�I@g J@g= ףp��@gq= ף�X@)r r rh r? r@ r? r@ )r r rh r< r= r? r@ )r, rA )r- r. r rC rD r r/ r0 )r r r r3 r4 r r r � test_multiby_heterogeneous_typesY sR z.TestAsOfMerge.test_multiby_heterogeneous_typesc C sL t jtd�ddgtd�ddgtd�ddgtd�ddggddd gd ��d�}t jtd �ddd gtd �ddd gtd�dddgtd�dddggddd dgd ��d�}t jtd�ddd gtd�ddd gtd�dddgtd�dddggddd dgd ��d�}t||dddd gd�}t�||� tjt dd�� t||dddd gdgd� W d Q R X d S )NZ20160602r r# r' Z20160603r$ r Zk1Zk2)r> Z20160502g �?g @Z20160503g @g @�valueT)rP rV rA z(left_by and right_by must be same length)rY )rP rV r_ r` ) r- r. r rR r r/ r0 rZ r[ r )r r1 r2 r3 r4 r r r �test_multiby_indexed� sB z"TestAsOfMerge.test_multiby_indexedc C sH | � |d�}| � |d�}| j |ddd�}t||ddd�}t�||� d S ) Nz asof2.csvz trades2.csvz quotes2.csvT)r r r )r, rA )r r r/ r0 )r r r3 r r r4 r r r � test_basic2� s zTestAsOfMerge.test_basic2c C sD dd� }|| j �}|| j�}|| j�}t||dd�}t�||� d S )Nc S s | | j dk jddd�jdd�S )Nr6 r r )ZaxisT)r )r r r )r r r r �<lambda>� s z0TestAsOfMerge.test_basic_no_by.<locals>.<lambda>r )r, )r r r r r/ r0 )r �fr3 r r r4 r r r �test_basic_no_by� s z TestAsOfMerge.test_basic_no_byc C s� | j }| j}d}tjt|d�� t||dddd� W d Q R X tjtdd�� t||ddgdd� W d Q R X tjtdd�� t||dd � W d Q R X d S ) Nz6incompatible merge keys \[1\] .* must be the same type)rY r r? r )rU rQ rA z can only asof on a key for left)r, rA )rA )r r rZ r[ r r )r r r r] r r r �test_valid_join_keys� s z"TestAsOfMerge.test_valid_join_keysc C sR t �| j| jg��ddg�jdd�}t| j|ddd�}| �|d�}t� ||� d S )Nr r T)r )r, rA zasof.csv) r- �concatr � sort_valuesr r r r r/ r0 )r r rc r4 r3 r r r �test_with_duplicates� s z"TestAsOfMerge.test_with_duplicatesc C sz t �dddgdddgd��}t �dddgdddgd��}t||dd�}t �dddgdddgdddgd��}t�||� d S ) Nr r( r' )�keyr& )ru r+ ru )r, )ru r& r+ )r- r. r r/ r0 )r �df1�df2r4 r3 r r r �test_with_duplicates_no_on� s z(TestAsOfMerge.test_with_duplicates_no_onc C s@ | j }| j}d}tjt|d�� t||dddd� W d Q R X d S )Nz/allow_exact_matches must be boolean, passed foo)rY r r �foo)r, rA r )r r rZ r[ r r )r r r r] r r r � test_valid_allow_exact_matches s z,TestAsOfMerge.test_valid_allow_exact_matchesc C s | j }| j}t||ddtd�d� t|�� |�� dddd� d}tjt|d�� t||dddd� W d Q R X tjt|d�� t|�� |�� ddd d� W d Q R X d }tjt|d�� t||ddtd� d� W d Q R X tjt|d�� t|�� |�� ddd d� W d Q R X d S ) Nr r Z1s)r, rA r rS r z6incompatible tolerance .*, must be compat with type .*)rY g �?ztolerance must be positive���)r r r r r rZ r[ r )r r r r] r r r �test_valid_tolerance s< z"TestAsOfMerge.test_valid_tolerancec C s� | j jddd�}| jjddd�}|jjr,t�|jjr8t�tjtdd�� t ||ddd� W d Q R X | j �d�}|jjszt�|jjr�t�tjtdd�� t ||ddd� W d Q R X | j�d�}|jjs�t�|jjs�t�t || jddd� d S ) Nr F)Z ascendingzleft keys must be sorted)rY r )r, rA zright keys must be sorted) r rs r r Z is_monotonic�AssertionErrorrZ r[ � ValueErrorr )r r r r r r �test_non_sortedG s z TestAsOfMerge.test_non_sortedr �1dayr )�daysr zdatetime.timedelta)Zidsc C s4 | j }| j}t||dd|d�}| j}t�||� d S )Nr r )r, rA r )r r r r r/ r0 )r r r r r4 r3 r r r �test_tolerance_ s z TestAsOfMerge.test_tolerancec C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdtjd gd ��}t||dddd�}t�||� d S )Nr r! r" r# r$ r% )r# r& r' r( r* � )r# r+ )r# r& r+ rF )r, rG r )r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �test_tolerance_forwardm s $"z$TestAsOfMerge.test_tolerance_forwardc C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdtjd gd ��}t||dddd�}t�||� d S )Nr r! r" r# r$ r% )r# r& r' r( r* r� )r# r+ )r# r& r+ rI )r, rG r )r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �test_tolerance_nearestz s $"z$TestAsOfMerge.test_tolerance_nearestc C s� t �t jtd�ddt�d�d�t�d�d��}t �t jtd�ddt�d�d�td�d ��}t ||d t d �d �}t �t jtd�ddt�d�d�t�d�td �d��}t � ||� d S )Nz 2016-01-02�Dr! �UTC)�start�freq�periods�tz)�date�value1z 2016-01-01ZABCDE)r� �value2r� z1 day)r, r ZBCDEE)r� r� r� ) r- r. � date_ranger �pytz�timezonerC Zarange�listr r r/ r0 )r r1 r2 r4 r3 r r r �test_tolerance_tz� s0 z TestAsOfMerge.test_tolerance_tzc C s� t �dddgdddgd��}t �dd d d d gdd d d d gd ��}t �dddgdddgdd tjgd��}t||dddd�}t�||� d S )Ng�������?g @g������%@r# r$ r% )r# r& g �?g @gffffff @g @g '@)r# r+ r )r# r& r+ rI g �?)r, rG r )r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �test_tolerance_float� s z"TestAsOfMerge.test_tolerance_floatc C sL | j �d�}| j�d�}| j�d�}t||dddtd�d�}t�||� d S )Nr Tr r� )rP rV rA r )r rR r r r r r/ r0 )r r3 r r r4 r r r �test_index_tolerance� s z"TestAsOfMerge.test_index_tolerancec C s, t | j| jdddd�}| j}t�||� d S )Nr r F)r, rA r )r r r r r/ r0 )r r4 r3 r r r �test_allow_exact_matches� s z&TestAsOfMerge.test_allow_exact_matchesc C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdd d gd ��}t||dddd�}t�||� d S )Nr r! r" r# r$ r% )r# r& r' r( r* r� )r# r+ )r# r& r+ rF F)r, rG r )r- r. r r/ r0 )r r1 r2 r3 r4 r r r � test_allow_exact_matches_forward� s $ z.TestAsOfMerge.test_allow_exact_matches_forwardc C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgdd d gd ��}t||dddd�}t�||� d S )Nr r! r" r# r$ r% )r# r& r' r( r* r� )r# r+ )r# r& r+ rI F)r, rG r )r- r. r r/ r0 )r r1 r2 r3 r4 r r r � test_allow_exact_matches_nearest� s $ z.TestAsOfMerge.test_allow_exact_matches_nearestc C s2 t | j| jddtd�dd�}| j}t�||� d S )Nr r Z100msF)r, rA r r )r r r r r r/ r0 )r r4 r3 r r r �&test_allow_exact_matches_and_tolerance� s z4TestAsOfMerge.test_allow_exact_matches_and_tolerancec C s� t �tdg�dgd��}t �tddg�ddgd��}t||dd �}t �tdg�dgdgd ��}t�||� t||dd d �}t �tdg�dgdgd ��}t�||� t||dd td �d�}t �tdg�dgtjgd ��}t�||� d S )Nz2016-07-15 13:30:00.030�bob)r �usernamez2016-07-15 13:30:00.000r r' )r �versionr )r, )r r� r� F)r, r rB )r, r r ) r- r. r r r/ r0 r rC rD )r rv rw r4 r3 r r r �'test_allow_exact_matches_and_tolerance2 s: z5TestAsOfMerge.test_allow_exact_matches_and_tolerance2c C s� t �tddg�ddgd��}t �tddg�ddgd��}t||d d td �d �}t �tddg�ddgtjtjgd ��}t�||� d S )Nz2016-07-15 13:30:00.030r� Zcharlie)r r� z2016-07-15 13:30:00.000r r' )r r� r FrB )r, r r )r r� r� ) r- r. r r r rC rD r/ r0 )r rv rw r4 r3 r r r �'test_allow_exact_matches_and_tolerance36 s( z5TestAsOfMerge.test_allow_exact_matches_and_tolerance3c C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgtjd d gd ��}t||ddddd�}t�||� d S )Nr r! r" r# r$ r% )r# r& r( � r) r� )r# r+ )r# r& r+ rF F)r, rG r r )r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �.test_allow_exact_matches_and_tolerance_forward[ s $"z<TestAsOfMerge.test_allow_exact_matches_and_tolerance_forwardc C s� t �dddgdddgd��}t �ddd d d gddd d d gd ��}t �dddgdddgtjd d gd��}t||ddddd�}t�||� d S )Nr r! r" r# r$ r% )r# r& r( r� r) r� r* )r# r+ )r# r& r+ rI F)r, rG r r )r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �.test_allow_exact_matches_and_tolerance_nearesto s $"z<TestAsOfMerge.test_allow_exact_matches_and_tolerance_nearestc C s� t �dddddgdddddgd d d d d gd��}t �dddddgdddddgdddddgd��}t �dddddgdddddgd d d d d gdtjdddgd��}t||d d dd�}t�||� d S )Nr r! r" � � �X�Y�Zr# r$ r% �d�e)r# r$ r& r) r� � )r# r$ r+ )r# r$ r& r+ rF )r, rA rG )r- r. rC rD r r/ r0 )r r1 r2 r3 r4 r r r �test_forward_by� s z TestAsOfMerge.test_forward_byc C s� t �dddddgdddddgd d d d d gd��}t �dddddgdddddgdddddgd��}t �dddddgdddddgd d d d d gdddddgd��}t||d d dd�}t�||� d S )Nr 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+ rI )r, rA rG )r- r. r r/ r0 )r r1 r2 r3 r4 r r r �test_nearest_by� s z TestAsOfMerge.test_nearest_byc C s� t jtdddddg�dddddgd d d d d gd�dddgd�}t jtddddddddg�ddddddddgddddd d d d gd �ddd!gd�}t||ddd"�}t jtdddddg�dddddgd d d d d gddddd gd#�dddd!gd�}t�||� d S )$Nz20160525 13:30:00.020z20160525 13:30:00.030z20160525 13:30:00.040z20160525 13:30:00.050z20160525 13:30:00.060r r' r( g�������?g333333�?g�������?gffffff�?g �?)r ru r� r ru r� )r> z20160525 13:30:00.015z20160525 13:30:00.025z20160525 13:30:00.035z20160525 13:30:00.055z20160525 13:30:00.065g������ @g������@gffffff@g333333@g @g������@g������@gffffff@)r ru r� r� )r, rA )r ru r� r� )r- r. r r r/ r0 )r rv rw r4 r3 r r r � test_by_int� sH zTestAsOfMerge.test_by_intc C s� t jdddddddgtd�d �d d gd �}t jd ddgdddgd�d dgd �}|�d �jdd�}t||d d�}t jtd�dddddddgdddddddgd�d d dgd �}t�||� d S )Ng ףp= @gHP�s�b?g�z�G!9@g�����@u@gH�z��>@g�����C�@g�q����?�ABCDEFG)r< �symbolr� r< )r> g g �?g Y@g-C ��6?g{�G�z�?g�������?)r< �mpvr� T)r )r, ZBGACEDF)r� r< r� )r- r. r� rs r r r/ r0 )r rv rw r4 r3 r r r � test_on_float� s zTestAsOfMerge.test_on_floatc C s� t �|�j}tjdddddddgtd�d �d d gd �}||j�|_tjd dddgtd�d�d dgd �}||j�|_|�d �jdd�}t ||d d�}tjtd�dddddddgtd�d�d d dgd �}||j�|_t � ||� d S )Nr! r' � r; �N �x �O r� )rk r� r� rk )r> r �P �} �xyzw)rk r4 r4 T)r )r, �BACEGDFZxxxxxyz)r� rk r4 ) rC �dtype�typer- r. r� rk rs r r r/ r0 )r �any_real_dtyper� rv rw r4 r3 r r r �test_on_specialized_type s$ z&TestAsOfMerge.test_on_specialized_typec C s$ t �|�j}tjdddddddgddd dd ddgtd �d �d d dgd�}||j�|_tjddddgdddd gtd�d�dd dgd�}||j�|_|�d�jdd�}t ||dd d�}tjtd�ddd d dddgdddddddgt j dt j t j t j ddgd �d d ddgd�}||j�|_t � ||� d S ) Nr! r' r� r; r� r� r� r r( r� )rk ru r� r� ru rk )r> r r� r� r� )rk ru r4 r4 T)r )r, rA r� r �y)r� ru rk r4 ) rC r� r� r- r. r� rk rs r r rD r/ r0 )r r� r� rv rw r4 r3 r r r � test_on_specialized_type_by_int9 s* z-TestAsOfMerge.test_on_specialized_type_by_intc C s4 t jtd�dddddddddg ddddd d d d d g d�dddgd�}t jdddddddddg dddddddddg dddddddd dg d �ddd gd�}|�d�jd d �}|�d�jd d �}t||ddd!�}t jtd�dddddddddg dddd dd d d d g dddd ddddd g d"�dddd gd�}t�||� d S )#NZ AAABBBCCCr r' r( g�G�z @g�ׁsF @g�� @g)\��()@g�G�z.)@g )@gfffff�w@g33333�w@g �w@)r� rh r< r� rh r< )r> g g �?g Y@g @g @�@g-C ��6?g{�G�z�?g�������?g�������?g �?)rh r< r� r� T)r )r, rA )r� rh r< r� )r- r. r� rs r r r/ r0 )r rv rw r4 r3 r r r �test_on_float_by_int] sH z"TestAsOfMerge.test_on_float_by_intc C sp d}t �dddgdddgd��}t �dd d d d gdd d d d gd ��}tjt|d�� t||dd� W d Q R X d S )Nz@Incompatible merge dtype, .*, both sides must have numeric dtyper r! r" r# r$ r% )r& r# r' r( r) r* )r+ r# )rY )r, )r- r. rZ r[ r r )r r] r1 r2 r r r � test_merge_datatype_error_raises� s $z.TestAsOfMerge.test_merge_datatype_error_raisesc C s| d}t �dddgt �dddg�d��}t �dd d d d gt �dd dd dg�d��}tjt|d�� t||dd� W d Q R X d S )NzHincompatible merge keys \[0\] .* both sides category, but not equal onesr r! r" r# r$ r% )r& r# r' r( r) r* r� )r+ r# )rY )r, )r- r. � CategoricalrZ r[ r r )r r] r1 r2 r r r �,test_merge_datatype_categorical_error_raises� s z:TestAsOfMerge.test_merge_datatype_categorical_error_raisesc C s` t �dgdgt �dg�d��}t||dddgd�}t �dgdgt �dg�d��}t�||� d S )Nr )r r� �zr r� r� )r, rA )r- r. r� r r/ r0 )r �dfr4 r3 r r r �:test_merge_groupby_multiple_column_with_categorical_column� s zHTestAsOfMerge.test_merge_groupby_multiple_column_with_categorical_column�funcc C s | S )Nr )r r r r rn � � zTestAsOfMerge.<lambda>c C s t | �S )N)r )r r r r rn � r� �numeric�datetime�sider1 r2 c C s� d|� d�}|ddt jg�}|dddg�}t�|dddgd ��}t�|d d d gd ��}tjt|d��, |dkr~t||dd� nt||dd� W d Q R X d S )Nz"Merge keys contain null values on z sideg �?g @g $@r# r$ r% )r# r& r r) r� )r# r+ )rY r1 )r, )rC rD r- r. rZ r[ r~ r )r r� r� r] ZnullsZ non_nullsZdf_nullr� r r r �test_merge_on_nans� s z TestAsOfMerge.test_merge_on_nansc C s� t �t �dg��d�dgdgd��}t �t �dg��d�dgdgd��}t||dd d �}t jt jddd �dddggdd d d gd�}t�||� d S )Nz 2018-01-01r� r' r# )�by_col�on_col�valuesr r$ r� r� )rA r, )r� Zvalues_xZvalues_y)r> )r- r. � DatetimeIndex� tz_localizer � Timestampr/ r0 )r r1 r2 r4 r3 r r r �test_merge_by_col_tz_aware� s z(TestAsOfMerge.test_merge_by_col_tz_awarec C s� t �t �dg��d�dgdgdgd��}t �t �dg��d�dgdgd gd��}t||d d gd d �}t jt jddd�dddggd d d dgd�}tjtjgt d�|d< t � ||� d S )Nz 2018-01-01r� ZHELLOr' r# )�by_col1�by_col2r� rk ZWORLDr r$ r� r� r� )rA r, )r� Zvalue_x)r> )r� Zvalue_y) r- r. r� r� r r� rC �arrayrD �objectr/ r0 )r r1 r2 r4 r3 r r r �test_by_mixed_tz_aware� s z$TestAsOfMerge.test_by_mixed_tz_awarec C s t jttddddddgdddd d dg��d d gd �}t �|d d�|d < t jttdd ddddgdddd d dg��d dgd �}t �|d d�|d < t jttddddddgdddd d dgdtjdd tjtjg��d d dgd �}t �|d d�|d < t||d td�dd�}t � ||� d S )Nr r! r" r� � r� r r' r( r� r r1 )r> �ms� r� � r2 Z1msrI )r, r rG ) r- r. r� �zipZ to_timedeltarC rD r r r/ r0 )r r1 r2 r3 r4 r r r � test_timedelta_tolerance_nearest� s& $ $ z.TestAsOfMerge.test_timedelta_tolerance_nearestc C s� t �dddgdddgd��}t �dd d gdddgd ��}|d �|�|d < |d �|�|d < t �dddgdddgtjd dgd��}|d �|�|d < t||d dd�}t�||� d S )Nr r" r� r r' r( )r# r& r! r� r� )r# r+ r# g �?g @)r# r& r+ )r, r )r- r. rN rC rD r r/ r0 )r Z any_int_dtyper1 r2 r3 r4 r r r �test_int_type_tolerance s "z%TestAsOfMerge.test_int_type_tolerancec C s t jddddd�}t jdddd gd g|d d � d �}t �|d gd dg d��}t||ddgd�}t jdddd g|d d � d gd dg d�t jddddd�d�}t�||� t||ddgd�}t j|d gd dg tjdddd gd�tdd dddg�d�}t�||� d S ) Nz 2019-10-01Z30minr! r� )r� r� r� g�������?g�������?gffffff�?g333333�?�xyzr )r> rS g�G�z�@r� g��Q� �@)� from_date�abcTr� )r1 r2 rP rQ r( )r� r� r� z2019-10-01 00:30:00)rS )r1 r2 rV rU )r� r� r� r r' ) r- r� r. r r/ r0 rC rD r )r rS r1 r2 r4 r3 r r r �test_merge_index_column_tz0 s( " z(TestAsOfMerge.test_merge_index_column_tzc C s� t jdddd�tdd� }t jddd�}t jd d it �|�d �}t jd d it �|�d �}t jd d d�t �|�d �}t||ddtdd�d�}t�||� d S )Nz1/1/2020z 1/20/2020Z2D)r� �endr� g�������?)�secondsz2/1/2020)r� r� �val1ry )rS �val2�bar)r� r� Tg �?)rP rV r )r- r� r r. r� r r/ r0 )r Zdr1Zdr2rv rw r3 r4 r r r �%test_left_index_right_index_toleranceQ s z3TestAsOfMerge.test_left_index_right_index_toleranceN)F)C�__name__� __module__� __qualname__r rZ Zfixturer r5 rE rH rJ rK rO rT rW rX r\ r^ ra rd ri rj rl rm rp rq rt rx rz r| r �mark� parametrizer r� � timedeltar� 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� r� r� r� r� r r r r r s| T AA2 2 ( 2% = $9 $!r �kwargsr, r T)rP rV r z2019-06-01 00:09:12z2019-06-01 00:10:29g �?c C sP t jd|i|d�}t jd|i|d�}tjtdd�� t||f| � W d Q R X d S )Nr )rS z@Incompatible merge dtype, .*, both sides must have numeric dtype)rY )r- r. rZ r[ r r )r� r r1 r2 r r r �#test_merge_asof_non_numerical_dtypeh s r� c C sj t �dddgdddgd��} t �dddgdd d gd ��}tjtd d �� t| |ddddd� W d Q R X d S )NZ12Z13Z15r# r$ r% )r# � left_val1r� r� ro )r# r& z@Incompatible merge dtype, .*, both sides must have numeric dtype)rY r� r& )rU rQ r_ r` )r- r. rZ r[ r r )r1 r2 r r r �*test_merge_asof_non_numerical_dtype_objectz s r� )rV rP � left_time)rU rV r2 )rP rQ c C s� t dddgdd�}tjdddgdd dgd �|d �}tjd dd dddgidd dddgd �}t||f| �}tjdddgdd dgdddgd�|d �}t�||� d S )Nr r! r" �test)r r# r$ r% r� )r1 r� )rS r2 r' r( r) r* )r1 r� r2 )r r- r. r r/ r0 )r� rS r1 r2 r4 r3 r r r � test_merge_asof_index_behavior� s & r� c C s� t jddddgitdddgdd �d �} t jd d d dgitdddgdd �d �}t| |ddd�}t �dddgdddgd d dgd��}t�||� d S )Nr$ r" r� r� r r' r( r# )r )rS r% r� � � r )rU rQ )r# r$ r% )r- r. r r r/ r0 )r1 r2 r4 r3 r r r �&test_merge_asof_numeri_column_in_index� s &&$r� c C s� t jddddgit jjdddgddd ggdd gd �d �} t jd d ddgit jjdddgddd ggddgd �d �}t| |ddd�}t �dddgdddgd ddgd��}t�||� d S )Nr$ r" r� r� r r' r( r# r% r� )�names)rS r� r� r� r r� )rU rQ )r# r$ r% )r- r. Z MultiIndexZ from_arraysr r/ r0 )r1 r2 r4 r3 r r r �+test_merge_asof_numeri_column_in_multiindex� s ( ($r� c C s� t jddddgitdddgdd �d �} t jd d d dgitdddgdd �d �}tjtdd�� t| |ddd� W d Q R X | �� �ddg�} |�� �dd g�}tjtdd�� t| |ddd� W d Q R X d S )Nr$ r" r� r� �1�2�3r# )r )rS r% r� r� r� �m�n�oz@Incompatible merge dtype, .*, both sides must have numeric dtype)rY )rU rQ ) r- r. r rZ r[ r r r rR )r1 r2 r r r �3test_merge_asof_numeri_column_in_index_object_dtype� s && r c C s� t �ddgt �d�t �d�gd��} t jt �d�ddd �}t �d d d d gi�}t|| |dddd�}t �d d d gtjddg|d��}t�||� t| |d|ddd�}t �ddgt �d�t �d�gd d gd��}t�||� d S )Nr' r) z2021/01/01 00:37z2021/01/01 01:40)r# �tsz2021/01/01 00:00r( Z1h)r� r� r� r$ r� � r* r FZbackward)rU rQ r rG )r$ r# r )r# r r$ ) r- r. r� r� r rC rD r/ r0 )r2 Zts_merger1 r4 r3 r r r �test_merge_asof_array_as_on� s6 r )r� �numpyrC rZ r� Zpandasr- r r r r r Zpandas._testingZ_testingr/ Zpandas.core.reshape.merger r r� r� r� r� r� r� r� r r r r r r �<module> sB _
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка