Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/__pycache__/test_subclass.cpython-37.pyc
Назад
B �5�g�\ � @ s` d dl Zd dlZd dlm mZ d dlZd dlm Z m Z m Z m Z d dl mZ G dd� d�ZdS )� N)� DataFrame�Index� MultiIndex�Seriesc @ 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"� Zej�d#�d$d%� �Zd&d'� Zd(d)� Zd*d+� Zej�d,d-d.g�d/d0� �Z d1d2� Z d3d4� Z e j d5d6� �Z!d7d8� Z"d9d:� Z#d;d<� Z$d=d>� Z%d?S )@�TestDataFrameSubclassingc s G �fdd�dt ��G � �fdd�dt�� td�td�d�}� |�}t|� �sRt�|j}t|��sft�|�� dksvt�|dd � }t|� �s�t�|�� dks�t�t� d d g�}� d dgd dgg|d�}t|d � �s�t�t� ddg�}� d dgd dgg|d�}t|d ���st�d S )Nc s$ e Zd Ze� fdd��Zdd� ZdS )zQTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeriesc s � S )N� )�self)� CustomSeriesr �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/test_subclass.py� _constructor s z^TestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeries._constructorc S s dS )N�OKr )r r r r �custom_series_function s zhTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomSeries.custom_series_functionN)�__name__� __module__� __qualname__�propertyr r r )r r r r s r c s< e Zd ZdZ� fdd�Ze�fdd��Z�Zdd� Z� Z S )zTTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFramez� Subclasses pandas DF, fills DF with simulation results, adds some custom plotting functions. c s t � j||� d S )N)�super�__init__)r �args�kw)� __class__r r r # s z]TestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame.__init__c s � S )Nr )r )�CustomDataFramer r r & s zaTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame._constructorc S s dS )Nr r )r r r r �custom_frame_function, s zjTestDataFrameSubclassing.test_frame_subclassing_and_slicing.<locals>.CustomDataFrame.custom_frame_function) r r r �__doc__r r r Z_constructor_slicedr � __classcell__r )r r )r r r s r � )�col1Zcol2r � � )�Ar )r �Br � � )�columnsr )r � )r r$ ) r r �range� isinstance�AssertionErrorr r r r � from_tuples)r �dataZcdfZ cdf_seriesZcdf_rowsZmcolZ cdf_multiZ cdf_multi2r )r r r �"test_frame_subclassing_and_slicing s" z;TestDataFrameSubclassing.test_frame_subclassing_and_slicingc C s� t jdddgdddgd�dddgd�}d |_|jd ks:t�|d g jd ksNt�|jddgd d �f jd ksnt�|jd dgd d �f jd ks�t�|jd d�d d �f jd ks�t�t �|�}t �||� |j|jks�t�|j|jks�t�d S ) Nr r! r" )�X�Y�a�b�c)�index�XXXr+ r ) �tm�SubclassedDataFrameZtestattrr' �loc�ilocZround_trip_pickle�assert_frame_equal� _metadata)r �dfZ unpickledr r r �test_dataframe_metadataH s " z0TestDataFrameSubclassing.test_dataframe_metadatac C s� t jdddgdddgddd gd �d d d gd�}|jd d �df }t jdddgtd�dd�}t �||� t|t j�svt�|jd d �df }t jdddgtd�dd�}t �||� t|t j�s�t�|jd d �df }t jddd gtd�dd�}t �||� t|t j��st�|jd d d �f }t jdddgtd�d d�}t �||� t|t j��sRt�|jdd d �f }t jdddgtd�d d�}t �||� t|t j��s�t�|jd d d �f }t jddd gtd�d d�}t �||� t|t j��s�t�d S )Nr r! r" � r � � � � )r+ r, �Zr- r. r/ )r0 r+ �abc)r0 �namer, r? ZXYZ) r2 r3 r4 �SubclassedSeries�list�assert_series_equalr& r' r5 )r r8 �res�expr r r �test_indexing_sliced\ s4 * z-TestDataFrameSubclassing.test_indexing_slicedc C s6 G dd� dt �}tjtdd�� |� j W d Q R X d S )Nc @ s e Zd Zedd� �ZdS )zFTestDataFrameSubclassing.test_subclass_attr_err_propagation.<locals>.Ac S s | j S )N)Z i_dont_exist)r r r r �bar� s zJTestDataFrameSubclassing.test_subclass_attr_err_propagation.<locals>.A.barN)r r r r rH r r r r r � s r z.*i_dont_exist.*)�match)r �pytestZraises�AttributeErrorrH )r r r r r �"test_subclass_attr_err_propagation s z;TestDataFrameSubclassing.test_subclass_attr_err_propagationc C sP t jdddgdddgd�td�d�}t jdddgdddgd �td �d�}|j|d d �\}}t jdtjdtjdgdtjdtjdgd�td �d�}t jddtjdtjgddtjdtjgd �td �d�}t|t j�s�t�t �||� t|t j�s�t�t �||� |j �|j �\}}t|t j ��s t�t � ||j � t|t j ��s>t�t � ||j � d S )Nr r" r )r- r. �ACE)r0 r! r: )r/ �d�ABDr )�axis�ABCDE) r2 r3 rC �align�np�nanr& r' r6 r- r/ rB rD )r �df1�df2�res1�res2�exp1�exp2r r r �test_subclass_align� s( $ $ z,TestDataFrameSubclassing.test_subclass_alignc C s* t jdddgdddgd�td�d�}t jdddgtd �d d �}|j|d d �\}}t jdtjdtjdgdtjdtjdgd�td�d�}t jddtjdtjgtd�d d �}t|t j�s�t�t � ||� t|t j�s�t�t � ||� |�|�\}}t|t j�s�t�t � ||� t|t j��st�t � ||� d S )Nr r" r )r- r. rM )r0 r! r: rO �x)r0 rA r )rP rQ ) r2 r3 rC rB rR rS rT r& r' r6 rD )r r8 �srW rX rY rZ r r r � test_subclass_align_combinations� s" $$ z9TestDataFrameSubclassing.test_subclass_align_combinationsc C sL t �ddgi�}x6|�� D ]*\}}t|t j�s2t�t �||j| � qW d S )Nr- r )r2 r3 Ziterrowsr& rB r' rD r4 )r r8 �i�rowr r r �test_subclass_iterrows� s z/TestDataFrameSubclassing.test_subclass_iterrowsc C sx t jdddgdddgddd ggd d d gd ddgd�}|�� }t jddddddddd g td�td�gd�}t �||� d S )Nr r! r" r: r r; r< r= r> r- r. r/ r+ r, r? )r0 r# Z aaabbbcccZ XYZXYZXYZ)r0 )r2 r3 �stackrB rC rD )r r8 rE rF r r r �test_subclass_stack� s (z,TestDataFrameSubclassing.test_subclass_stackc C s� t jddddgddddgd d d d gd dddggtjtttd�td���ddgd�tjtttd�td���ddgd�d�}t jddgddgddgddgd d gd d gd dgddggtjtttd�td �td ���dddgd�td d gdd �d�}|�� }t �||� |�d�}t �||� t jddgddgddgddgd d gd d gd dgddggtjtttd�td �td!���dddgd�td"d#gdd �d�}|�d�}t �||� d S )$Nr � � � � � � � � � � �! �( �) �* �+ �AABB�cdcd�aaa�ccc)�names�WWXX�yzyz�www�yyy)r0 r# �AAAABBBB�ccddccdd�yzyzyzyz�Wr+ )rA �WXWXWXWX�y�z) r2 r3 r r( rC �zipr rb r6 )r r8 rF rE r r r �test_subclass_stack_multi� sL * $ z2TestDataFrameSubclassing.test_subclass_stack_multic C s� t jddddgddddgd d d d gd dddggtjtttd�td���ddgd�tjtttd�td���ddgd�d�}t jddgddgddgddgd d gd d gd dgddggtjtttd�td �td ���dddgd�td d gdd �d�}|�� }t �||� |�d�}t �||� t jd!d"gddgd#d$gddgd%d&gd d gd'd(gddggtjtttd�td �td)���dddgd�td*d+gdd �d�}|�d�}t �||� d S ),Nr rd g (@g *@rg rh g 6@g 7@rk rl g @@g �@@ro rp g E@g �E@rs rt ru rv )rw rx ry rz r{ )r0 r# r| r} r~ r r+ )rA g $@g &@g 4@g 5@g >@g ?@g D@g �D@r� r� r� ) r2 r3 r r( rC r� r rb r6 )r r8 rF rE r r r � test_subclass_stack_multi_mixed sR $ z8TestDataFrameSubclassing.test_subclass_stack_multi_mixedc C sx t jdddgdddgddd ggd d d gd ddgd�}|�� }t jddddddddd g td�td�gd�}t �||� d S )Nr r! r" r: r r; r< r= r> r- r. r/ r+ r, r? )r0 r# Z XXXYYYZZZZ abcabcabc)r0 )r2 r3 �unstackrB rC rD )r r8 rE rF r r r �test_subclass_unstackO s (z.TestDataFrameSubclassing.test_subclass_unstackc C s� t jddddgddddgd d d d gd dddggtjtttd�td���ddgd�tjtttd�td���ddgd�d�}t jddddddddgd d d dd dd dggtdd gdd �tjtttd �td �td ���dddgd�d�}|�� }t �||� |�d�}t �||� t jdd dd dd dd gdd ddddddggtd!d"gdd �tjtttd �td �td#���dddgd�d�}|�d�}t �||� d S )$Nr rd re rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt ru rv )rw rx ry rz r{ )r0 r# r r )rA �WWWWXXXX�yyzzyyzz�cdcdcdcdr/ rN �ABABABAB) r2 r3 r r( rC r� r r� r6 )r r8 rF rE r r r �test_subclass_unstack_multi^ s0 * $& & z4TestDataFrameSubclassing.test_subclass_unstack_multic C s� t jddddgddddgd d d d gd dddggtjtttd�td���ddgd�tjtttd�td���ddgd�d�}t jddddddddgd d d dd dd dggtdd gdd �tjtttd �td �td ���dddgd�d�}|�� }t �||� |�d�}t �||� t jdd dd dd dd gdd ddddddggtd!d"gdd �tjtttd �td �td#���dddgd�d�}|�d�}t �||� d S )$Nr rd g (@g *@rg rh g 6@g 7@rk rl g @@g �@@ro rp g E@g �E@rs rt ru rv )rw rx ry rz r{ )r0 r# r r )rA r� r� r� r/ rN r� ) r2 r3 r r( rC r� r r� r6 )r r8 rF rE r r r �!test_subclass_unstack_multi_mixed� s: $ z:TestDataFrameSubclassing.test_subclass_unstack_multi_mixedc C s� t �ddddddgddddddgddddddgd ��}|jd d d d �}t �dddd �dddd �d��}d\|j_|j_t �||� d S )Nr r �C�One�Twog �?g @g @)r0 r# �valuesr0 r# r� )r r r� )r� r� )r0 r# )r2 r3 Zpivotr0 rA r# r6 )r r8 Zpivoted�expectedr r r �test_subclass_pivot� s z,TestDataFrameSubclassing.test_subclass_pivotc C s� t �ddgddgddgddgd ��}tj|d d gd �}t jddd dgddd dgddddgddddggd d ddgd�}t �||� d S )N�John�Mary�Doe�Bog @g @� � )�first�last�height�weightr� r� )Zid_varsr� r� �variable�value)r# )r2 r3 �pdZmeltr6 )r ZcheeseZmeltedr� r r r �test_subclassed_melt� s z-TestDataFrameSubclassing.test_subclassed_meltc C s� t j�d� t j�d�}t�dddd�ddd d�d d d d�d ddd�tttd�|��d��}|j |d< |� � |� � dddddd gd d d d ddgddddddgddddddgd�}t�|�}|� ddg�dddg }t j |ddgddd �}t�||� d S ) N�{ r" r- r. r/ )r r r! rN �e�fg @g333333�?gffffff�?g������ @g�������?g�������?)ZA1970ZA1980ZB1970ZB1980r+ �idi� i� r r r! )r+ r r �yearr� r� r+ r r )r_ �j)rS �random�seed�randnr2 r3 �dictr� r% r0 �tolistZ set_indexr� Z wide_to_longr6 )r r\ r8 Zexp_datar� Z long_framer r r � test_subclassed_wide_to_long� s$ z5TestDataFrameSubclassing.test_subclassed_wide_to_longc s� dd� � dd� �t jddddgd d dd gddd d gd d d dggddddgd�}|�� fdd�� |j� fdd�dd� t jdddd gd d ddgddd d gd d d dggddddgd�}|j�fdd�dd�}t|t j�s�t�t �||� t �ddd gddd gddd gddd gg�}|jd d� dd�}t|t j��s4t�t �||� |jd d� dd d �}t|t j��sft�t �||� t �ddd gddd gddd gddd gg�}|jd!d� dd�}t|t j��r�t�t �||� d S )"Nc S s t | tj�st�d S )N)r& r2 rB r' )r` r r r �check_row_subclass s zJTestDataFrameSubclassing.test_subclassed_apply.<locals>.check_row_subclassc S s | d dkr | d d7 < | S )Nr� r� r� g �?r )r` r r r �stretch s z?TestDataFrameSubclassing.test_subclassed_apply.<locals>.stretchr� r� r� g @r� r� g @r� r� r� r� r� r� r� )r# c s � | �S )Nr )r\ )r� r r �<lambda> � z@TestDataFrameSubclassing.test_subclassed_apply.<locals>.<lambda>c s � | �S )Nr )r\ )r� r r r� r� r )rP g @c s � | �S )Nr )r\ )r� r r r� & r� r! r" c S s t �dddg�S )Nr r! r" )r2 rB )r\ r r r r� , r� c S s dddgS )Nr r! r" r )r\ r r r r� 0 r� �expand)rP Z result_typec S s dddgS )Nr r! r" r )r\ r r r r� 6 r� )r2 r3 �applyr& r' r6 rB rD )r r8 r� �resultr )r� r� r �test_subclassed_apply s< * *z.TestDataFrameSubclassing.test_subclassed_applyz*ignore:.*None will no longer:FutureWarningc C sD t �dddgdddgddd gd ��}t||�� }t|t j�s@t�d S ) Nr r! r" r: r r; r< r= r> )r r r� )r2 r3 �getattrr& rB r' )r Zall_reductionsr8 r� r r r �test_subclassed_reductions: s $ z3TestDataFrameSubclassing.test_subclassed_reductionsc C sX t �dddddgdtjdddgdd d d dgd ��}|�� }t|t j�sJt�t �d d d gd ddgddd gd��}|�� }t|t j�s�t�t jddddgddddgdd d dgd d d d!ggtj t t t d"�t d#���d$d%gd&�tj t t t d'�t d(���d)d*gd&�d+�}t � t �� |jd d,�}W d Q R X t|t j��s2t�t �� }|�� }t|t j��sTt�d S )-Nr� ZMylaZLewisg 8@g 5@rn � FT)ZPersonZAge�Singler r r" r r; r< r= )r r r� r rd re rf rg rh ri rj rk rl rm ro rp rq rr rs rt ru rv )rw rx ry rz r{ )r0 r# )�level)r2 r3 rS rT �countr& rB r' r r( rC r� Zassert_produces_warning� FutureWarning)r r8 r� r r r �test_subclassed_countB s* $* $ z.TestDataFrameSubclassing.test_subclassed_countc C sB t jddgddgd�ddgd�}|�ddg�}t|t j�s>t�d S )Nr! r: r )Znum_legsZ num_wingsZfalcon�dog)r0 )r2 r3 �isinr& r' )r r8 r� r r r � test_isinc s z"TestDataFrameSubclassing.test_isinc C s` t �dddgdddgddd gd ��}|�� }t|t j�s<t�t �� }|�� }t|t j�s\t�d S ) Nr r! r" r: r r; r< r= r> )r r r� )r2 r3 Z duplicatedr& rB r' )r r8 r� r r r �test_duplicatedk s $z(TestDataFrameSubclassing.test_duplicated� idx_method�idxmax�idxminc C sD t �dddgdddgddd gd ��}t||�� }t|t j�s@t�d S ) Nr r! r" r: r r; r< r= r> )r r r� )r2 r3 r� r& rB r' )r r� r8 r� r r r �test_idxu s $ z!TestDataFrameSubclassing.test_idxc C s� t �ddddgddddgg�}t �ddddg�}|�|�}t|t j�sJt�t �ddddgddddgg�}t �ddddg�}|�|�}t|t j�s�t�d S )Nr r ������r! )r2 r3 rB �dotr& r' )r r8 r] r� r r r �test_dot| s z!TestDataFrameSubclassing.test_dotc C s\ t �dddgdddgddd gd ��}|�� }t|t j�s<t�|jd d �}t|t j�sXt�d S ) Nr r! r" r: r r; r< r= r> )r r r� F)r0 )r2 r3 Z memory_usager& rB r' )r r8 r� r r r �test_memory_usage� s $ z*TestDataFrameSubclassing.test_memory_usagec C s| dddddg}dddd g}t jtj�d d �||d �}t jtj�d d �|d d � |d �}|j|d ddd�}t|t j�sxt�d S )Nr- r. r/ rN r� �one�two�three�fourr r: )r0 r# r TZkendall)rP Zdrop�method) r2 r3 rS r� r� Zcorrwithr& rB r' )r r0 r# rU rV Zcorrelsr r r � test_corrwith� s z&TestDataFrameSubclassing.test_corrwithc C s� d}t jd|dd�}tjtjtjtjgtjtjtjgtjtjtjgd�|d�}|�|dd � �}t|tj�spt�|�|d �}t|tj �s�t�|�d�}t|tj �s�t�d S ) Nr" z1/1/1990Z53s)Zperiods�freq)r r r� )r0 r� z 1989-12-31) r� Z date_ranger2 r3 rS rT Zasofr& r' rB )r �N�rngr8 r� r r r � test_asof� s z"TestDataFrameSubclassing.test_asofc C s@ t �dddgdddgddd gd ��}|�� }t|t j�s<t�d S ) Nr r! r" r: r r; r< r= r> )r r r� )r2 r3 r� r& rB r' )r r8 r� r r r � test_idxmin_preserves_subclass� s $z7TestDataFrameSubclassing.test_idxmin_preserves_subclassc C s@ t �dddgdddgddd gd ��}|�� }t|t j�s<t�d S ) Nr r! r" r: r r; r< r= r> )r r r� )r2 r3 r� r& rB r' )r r8 r� r r r � test_idxmax_preserves_subclass� s $z7TestDataFrameSubclassing.test_idxmax_preserves_subclassc C sF t ddddgi�}t�ddddgi�}|�|�s4t�|�|�sBt�d S )Nr- r r! r" )r r2 r3 �equalsr' )r rU rV r r r �test_equals_subclass� s z-TestDataFrameSubclassing.test_equals_subclassN)&r r r r* r9 rG rL r[ r^ ra rc r� r� r� r� r� r� r� r� r� rJ �mark�filterwarningsr� r� r� r� Z parametrizer� r� r� �tdZskip_if_no_scipyr� r� r� r� r� r r r r r s: 7# 9>'2 6! r )�numpyrS rJ Z pandas.util._test_decorators�utilZ_test_decoratorsr� Zpandasr� r r r r Zpandas._testingZ_testingr2 r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка