Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/base/__pycache__/reshaping.cpython-37.pyc
Назад
B �5�gp7 � @ sl d dl Z d dlZd dlZd dlmZmZmZ d dlZ d dl m Z d dl m Z d dlmZ G dd� de�ZdS )� N)�is_datetime64tz_dtype�is_interval_dtype�is_period_dtype)�ExtensionArray)�ExtensionBlock)�BaseExtensionTestsc @ sp e Zd ZdZej�dddg�dd� �Zej�dddg�dd� �Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zej�d#d$d%gejjd&d'gd(d)gd*�g�d+d,� �Zej�d-ejjd$d%gd.d/ggd.d/gd*�ej�d&d'd0g�ej�d1d2d3g�ej�d4d5d6d7d8g�g�ej�d9d:d;g�d<d=� ��Zd>d?� Z d@dA� Z dBdC� Z dDS )E�BaseReshapingTestsz&Tests for reshaping and concatenation.�in_frameTFc C s� t �|�}|rt �|�}t j||gdd�}t|�t|�d ksBt�|rR|jd }n|j}||jksft�t|j d�r�t |j j d t �s�t�t |j j d t�s�t�d S )NT)� ignore_index� r �blocks)�pd�Series� DataFrame�concat�len�AssertionError�dtypes�dtype�hasattrZ_mgr� isinstancer r Zarraysr )�self�datar �wrapped�resultr � r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/base/reshaping.py� test_concat s z BaseReshapingTests.test_concatc C s� t j|�ddg�ddgd�}t j|�ddg�ddgd�}|rXt �d|i�}t �d|i�}t �||g�}|r�t �d|�ddddg�i�}| �||� n$t �|�ddddg��}| �||� d S )N� r )�indexr � �a)r r �taker r �assert_frame_equal�assert_series_equal)r Z data_missingr Z valid_blockZna_blockr �expectedr r r �test_concat_all_na_block( s z+BaseReshapingTests.test_concat_all_na_blockc C s4 t �d|d d� i�}t �ddddgi�}t �ddddgi��d�}|||g}t �|�}t �d d � |D ��}| �||� t �d d � |D ��}t �d d � |D ��}| �||� t �||�t�g�}t �|�d �|�d �g�}| �||� t �|d |d �t�g�}t �|d �d �|d �d �g�}| �||� d S )N�Ar r r r! �b�c�categoryc S s g | ]}|� t��qS r )�astype�object)�.0�xr r r � <listcomp>@ s z?BaseReshapingTests.test_concat_mixed_dtypes.<locals>.<listcomp>c S s g | ] }|d �qS )r' r )r- r. r r r r/ D s c S s g | ]}|d � t��qS )r' )r+ r, )r- r. r r r r/ E s r, )r r r+ r r# r$ r, )r r �df1�df2Zdf3�dfsr r% r r r �test_concat_mixed_dtypes7 s "z+BaseReshapingTests.test_concat_mixed_dtypesc C s* t �d|d d� i�}t �ddddgi�}t �|d d� dddgd��}t j||gdd�}| �||� t j|d |d gdd�}| �||� t jddddgidddgd�}t �|jt|d d� �|g |jd �tjdddgd��}t j||gdd�}| �||� t j|d |d gdd�}| �||� d S ) Nr' r �Br r )r' r4 )�axis)r )r ) r r r r# �_from_sequence�listr �np�nan)r r �na_valuer0 r1 r% r r r r �test_concat_columnsQ s z&BaseReshapingTests.test_concat_columnsc C s� t �d|d d� i�}t �d|dd� i�}t �|jt|d d� �|g |jd�|dd� d��}t j||gddd �}| �||� d S ) Nr' r r4 � )r )r' r4 r F)r5 �copy)r r r6 r7 r r r# )r r r: r0 r1 r% r r r r �'test_concat_extension_arrays_copy_falsei s z:BaseReshapingTests.test_concat_extension_arrays_copy_falsec C s� t �d|d d� i�}t �d|d d� i�}t j||gdd�}t �|jttd��dgd dd�|jdgd ttd�� dd�d��}| �||� d S ) Nr! � r( T)r ���)Z allow_fill)r! r( )r r r r" r7 �ranger# )r r r! r( r r% r r r �test_concat_with_reindexv s &z+BaseReshapingTests.test_concat_with_reindexc C s� |d d� }|dd� }t �|��t j|dddgd��\}}t �|jt|�|g |jd��}t �|j|gt|� |jd��}| �||� | �||� d S )Nr r r? r )r )r )r r �alignr6 r7 r r$ ) r r r: r! r( �r1�r2�e1�e2r r r � test_align� s $ z BaseReshapingTests.test_alignc C s� |d d� }|dd� }t �d|i��t jd|idddgd��\}}t �d|jt|�|g |jd�i�}t �d|j|gt|� |jd�i�}| �||� | �||� d S )Nr r r? r' r )r )r )r r rC r6 r7 r r# ) r r r: r! r( rD rE rF rG r r r �test_align_frame� s , z#BaseReshapingTests.test_align_framec C sz t j|dd�}t �dt�t|�d �i�}|�|�\}}t j|jt|�|g |j d�|j d�}| � ||� | � ||� d S )Nr! )�name�colr )r ) r r r r8 Zaranger rC r6 r7 r rJ r$ r# )r r r: �ser�dfrD rE rF r r r �test_align_series_frame� s z*BaseReshapingTests.test_align_series_framec C sJ t �ddgt|� i�}||d< t �dgt|� |d��}| �||� d S )Nr' r r4 )r' r4 )r r r r# )r r rM r% r r r �,test_set_frame_expand_regular_with_extension� s z?BaseReshapingTests.test_set_frame_expand_regular_with_extensionc C sJ t �d|i�}dgt|� |d< t �|dgt|� d��}| �||� d S )Nr' r r4 )r' r4 )r r r r# )r r rM r% r r r �,test_set_frame_expand_extension_with_regular� s z?BaseReshapingTests.test_set_frame_expand_extension_with_regularc C s< t jddgt|� itd�}||d< |jd |jks8t�d S )Nr' r )r )r r r r, r r r )r r rM r r r � test_set_frame_overwrite_object� s z2BaseReshapingTests.test_set_frame_overwrite_objectc C s6 t �|d d� dddgdddgd��}t �ddddgddddgd��}t �||�}t �dddgdddgdddg|j|d |d |d g|jd�d ��}| �||d d d d g � t j||dd�}t �ddddtjgdddtjdgdddddg|j|d |d |d |d |g|jd�d ��}| �||d d d d g � d S )Nr r r r )�ext�int1�key� )�int2rT )r )rS rV rT rR rR rS rT rV �outer)�how)r r �merger6 r r# r8 r9 )r r r: r0 r1 �res�expr r r � test_merge� s$ & $ ,z BaseReshapingTests.test_mergec C s� |d d� \}}t |�j||g|jd�}t�|ddgd��}tj||dd�}t�|ddgddgd��}| �||� tj|jddg |dd�}|jddg jd d �}| �||� d S ) Nr )r r )rT �valrT )�on)rT �val_x�val_yr T)Zdrop) �typer6 r r r rY r# �ilocZ reset_index)r r r! r( rT rM r r% r r r � test_merge_on_extension_array� s z0BaseReshapingTests.test_merge_on_extension_arrayc C s� |d d� \}}t |�j|||g|jd�}t�|dddgd��}t�|dddgd��}tj||dd�}t�|�dddddg�dddddgdddddgd ��}| �||� d S ) Nr )r r r )rT r] rT )r^ r )rT r_ r` )ra r6 r r r rY r" r# ) r r r! r( rT r0 r1 r r% r r r �(test_merge_on_extension_array_duplicates� s z;BaseReshapingTests.test_merge_on_extension_array_duplicates�columnsr' r4 )r' r! )r' r( rW �inner)�namesc C s� t �|d d� |d d� d��}||_|�� }|�t��� }|�t�}t|t j�rr|j|j d d �df jks�t �n"t |j |j d d �df jk�s�t �|�t�}| � ||� d S )Nr? )r' r4 r )r r re �stackr+ r, r r r rb r �allr Z assert_equal)r r re rM r r% r r r � test_stack� s " z BaseReshapingTests.test_stackr r! r( )r4 r( )r' r4 )r! r( r) )r r r )r' r! r )r' r( r )r' r! r )r4 r! r )r4 r) r �obj�series�framec s� � d t |�� � |dkr(tj� |d�}ntj� � d�|d�}|j}tt|���tj� �fdd�td|�D ��}�x&|D �] }|j |d��t � �fdd��j D ��s�t �|dkr�|�� }|j |d�jd dd �} | ��| � |�t�j |� jjd �} |dk�r~t� j��r&| j�t�� � �s t �| �t�} t� j��rR| j�t�� � �sHt �| �t�} t� j��r~| j�t�� � �stt �| �t�} ��t��| ��| � qvW d S ) Nrl )r )r' r4 c 3 s | ]}t �� |�V qd S )N)� itertools� permutations)r- �i)�levelsr r � <genexpr>7 s z2BaseReshapingTests.test_unstack.<locals>.<genexpr>r )�levelc 3 s" | ]}t �| jt� ��V qd S )N)r �arrayra )r- rK )r r r r rr = s r )r5 )rs Z fill_value)r r r r Znlevelsr7 rA rn �chain� from_iterableZunstackri re r Zto_frameZ droplevelr# r+ r, r r: r r �applyr r ) r r r rk rL �n� combinationsrs rM Zaltr% r )r rq r r � test_unstack s< z BaseReshapingTests.test_unstackc C s@ |� � }t|�t|�ks t�|d |d< |d |d ks<t�d S )Nr r )Zravelra r )r r r r r r � test_ravelY s z BaseReshapingTests.test_ravelc C sf |� � }t|�t|�ks t�||k s(t�|j|jd d d� ksBt�|d |d< |d |d ksbt�d S )Nr@ r r )� transposera r �shape)r r r r r r �test_transposeb s z!BaseReshapingTests.test_transposec C s t j|d d� |d d� d�ddddgd�}|j}t jt|�j|d gd |jd �t|�j|d gd |jd �t|�j|d gd |jd �t|�j|d gd |jd �d �ddgd�}| �||� | �t�t�|��|� | �t�t�|dg ��|dg � d S )NrU )r' r4 r! r( r) �d)r r r )r r r )r! r( r) r r' r4 ) r r �Tra r6 r r# r8 r| )r r rM r r% r r r �test_transpose_framep s , z'BaseReshapingTests.test_transpose_frameN) �__name__� __module__� __qualname__�__doc__�pytest�markZ parametrizer r& r3 r; r> rB rH rI rN rO rP rQ r\ rc rd r Z MultiIndex� from_tuplesrj Z from_productrz r{ r~ r� r r r r r sH / r )rn �numpyr8 r� Zpandas.core.dtypes.commonr r r Zpandasr Zpandas.api.extensionsr Zpandas.core.internalsr Z pandas.tests.extension.base.baser r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка