Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/internals/__pycache__/test_internals.cpython-37.pyc
Назад
B �5�gU� � @ s� d dl mZm Z d dlZd dlZd dlZd dlZd dlmZ d dl m m Z d dl mZ d dlZd dlmZmZmZmZmZmZmZmZmZ d dlmZ d dl m m Z d dl!m"Z"m#Z#m$Z$ d dl%m&Z&m'Z'm(Z( d dl)m*Z*m+Z+ e j,Z-ej.e+e(gd �d d � �Z/ej.d d � �Z0dd� Z1dd� Z2dZ3dd e+fdd�Z4d/dd�Z5d0dd�Z6G dd� d�Z7G dd � d �Z8d d � Z9G d d � d �Z:G d!d"� d"�Z;G d#d$� d$�Z<G d%d&� d&�Z=d'd(� Z>d)d*� Z?d+d,� Z@d-d.� ZAdS )1� )�date�datetimeN)�BlockPlacement)� is_scalar) � Categorical� DataFrame� DatetimeIndex�Index� IntervalIndex�Series� Timedelta� Timestamp� period_range)� DatetimeArray� SparseArray�TimedeltaArray)� BlockManager�SingleBlockManager� make_block)�ensure_block_shape� new_block)�paramsc C s | j S )zS Fixture to test both the internal new_block and pseudo-public make_block. )�param)�request� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/internals/test_internals.py� block_maker2 s r c C s t d�S )Nz�a: f8; b: object; c: f8; d: object; e: f8;f: bool; g: i8; h: complex; i: datetime-1; j: datetime-2;k: M8[ns, US/Eastern]; l: M8[ns, CET];)� create_mgrr r r r �mgr: s r c C sX t �| j|j� | j|jks t�t| jt�s0t�t|jt�s@t�t �| jj|jj� d S )N) �tm�assert_numpy_array_equal�values�dtype�AssertionError� isinstance�mgr_locsr �as_array)�left�rightr r r �assert_block_equalC s r) c C s: t �| d �}t jjj|| |jfdt| �d d��� S )Nr )r � )�x�shape�strides)�np�arange�libZ stride_tricksZ as_strided�itemsize�len�copy)r, �arrr r r �get_numeric_matK s r5 � c C sX t |�}t|�}|dkr tf}|f| }t|�}| dkrJ|�| �| }�n�| dkrhd|�| �| }�n�| dkr�t�dd� |�� | D �|�}�n�| dkr�tj|tj d �}�n�| d kr�|d �d �}�n~| � d ��r@t � d| �} | dk s�t d| � ���| �� d } |dk�st d��tt�t�d | d�j}t|t|�d�}�n| dk�rZ|d �d�}n�| dk�r�tddddddddddg �}n�| dk�r�td d d d d d d d d d g �}n�| d k�r6|d! d"k�s�t �td#d$� |dd!� D ���s�t �| �d%��r�tj} nd&} t| | ddd| dd'| d(g | d)�}|j�� } | |d 7 } ntd*| � d+���|||t|�d,�S )-a� Supported typestr: * float, f8, f4, f2 * int, i8, i4, i2, i1 * uint, u8, u4, u2, u1 * complex, c16, c8 * bool * object, string, O * datetime, dt, M8[ns], M8[ns, tz] * timedelta, td, m8[ns] * sparse (SparseArray with fill_value=0.0) * sparse_na (SparseArray with fill_value=np.nan) * category, category2 N)�float�f8Zf4�f2�int�i8�i4�i2�i1Zuint�u8Zu4�u2�u1)�complexZc16Zc8y �?)�object�string�Oc S s g | ]}d |d���qS )�A�dr )�.0�ir r r � <listcomp>� s z create_block.<locals>.<listcomp>)�b�bool)r" )r �dtzM8[ns]g e��AzM8[ns]zM8[nszM8\[ns,\s*(\w+\/?\w*)\]zincompatible typestr -> r r* z)must have only 1 num items for a tz-aware)�tz)�ndim)� timedelta�tdzm8[ns]zm8[ns])�category� � � )Z category2�arK �crG )�sparseZ sparse_na���r6 c s s | ]}|d kV qdS )r* Nr )rH �sr r r � <genexpr>� s z create_block.<locals>.<genexpr>Z_nag � � )� fill_valuezUnsupported typestr: "�")� placementrO )r r2 �Nr5 �astyper. �reshape�ravel�ones�bool_� startswith�re�searchr# �groupsr r/ �_datar r �all�endswith�nanr Z sp_values�view� ValueError) �typestrr` � item_shape� num_offset�maker� num_itemsr, �matr! �mrN r^ r4 r r r � create_blockU sV " rx c C s0 |d kr t }tt| td|�dd�tt�|���S )Nr r )r` rr )ra r rx �slicer r. r/ )rq �num_rowsr r r �create_single_mgr� s r{ c C sN |dkrt f}d}g }i }x�| �d�D ]�}|�� }t|�s<q&|�d�ddd� \}}|�� }|�� �d�}|�|� tt�t|��| �}y|| �|� W n t k r� |||< Y nX |t|�7 }q&W t |�}g } d} xD|� � D ]8\}}|�d�d } | � t | ||| d�� | t|�7 } q�W t| d d � d �} tt| �|gd d � |D � �S )aU Construct BlockManager from string description. String description syntax looks similar to np.matrix initializer. It looks like this:: a,b,c: f8; d,e,f: i8 Rules are rather simple: * see list of supported datatypes in `create_block` method * components are semicolon-separated * each component is `NAME,NAME,NAME: DTYPE_ID` * whitespace around colons & semicolons are removed * components with same DTYPE_ID are combined into single block * to force multiple blocks with same dtype, use '-SUFFIX':: 'a:f8-1; b:f8-2; c:f8-foobar' Nr �;�:rS �,�-)rr rs c S s | j d S )Nr )r% )rK r r r �<lambda>� � z create_mgr.<locals>.<lambda>)�keyc S s g | ]}t t�|���qS r )r r. r/ )rH �nr r r rJ � s z create_mgr.<locals>.<listcomp>)ra �split�stripr2 � partition�extend�listr. r/ �KeyErrorr �items�appendrx �sortedr �tuple) �descrrr �offsetZ mgr_itemsZblock_placementsrG �namesZblockstrr` �blocksrs rq Zsblocksr r r r � s@ r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� TestBlockc C sB t ddddg�| _t ddg�| _t ddd g�| _t d d g�| _d S ) Nr7 r rS rU rB � rC r* rT rL r\ )rx �fblock�cblock�oblock� bool_block)�self�methodr r r � setup_method� s zTestBlock.setup_methodc C s t ddg�}|jtjks t�d S )Nr< r )rx r" r. �int32r# )r� Z int32blockr r r �test_constructor� s zTestBlock.test_constructorc C s4 dd� }|| j � || j� || j� || j� d S )Nc S s t t�| �| � d S )N)r) r �round_trip_pickle)�blkr r r �_check s z%TestBlock.test_pickle.<locals>._check)r� r� r� r� )r� r� r r r � test_pickle� s zTestBlock.test_picklec C s: t | jjt�st�t�| jjjtj dddgtj d�� d S )Nr rS rU )r" ) r$ r� r% r r# r r r& r. �array�intp)r� r r r � test_mgr_locs s zTestBlock.test_mgr_locsc C sJ | j j| j jjkst�| j j| j jjks,t�t| j �t| j j�ksFt�d S )N)r� r, r! r# r" r2 )r� r r r � test_attrs s zTestBlock.test_attrsc C s( | j �� }|| j k st�t| j |� d S )N)r� r3 r# r) )r� �copr r r � test_copy s zTestBlock.test_copyc C s0 | j �� }|�d� t|jt�s$t�t�|jj t j ddgt j d�� |j d dk�� sZt�| j �� }|�d� t|jt�s~t�t�|jj t j ddgt j d�� |j d dk�� s�t�| j �� }|�d� t�|jj t j ddgt j d�� |j d dk�� s�t�| j �� }tjtd d�� |�d� W d Q R X d S )Nr rS rU )r" r* )�matchrT )r� r3 �deleter$ r% r r# r r r& r. r� r� r! rl �pytest�raises� IndexError)r� Znewbr r r � test_delete s( zTestBlock.test_deletec C s� t jddd��dd��d�}t|�}|jjd }t|jt �s@t �|� d� t|jt �sZt �t|�d ��}|jjd }t|jt �s�t �|� dd g� t|jt �s�t �d S ) N� r; )r" r\ rU zm8[ns]r r* zM8[ns]rT ) r. r/ rc ro r �_mgrr� r$ r! r r# r� r )r� r4 �dfr� r r r �test_delete_datetimelike5 s z"TestBlock.test_delete_datetimelikec C s� t j�dd�}t|dddgdd�}|�� }d|d d �< |jdk�� sHt�t|�dksXt�t|dg dgdd�t|dg dgdd�t|dg dgdd�g}x t ||�D ]\}}t ||� q�W d S ) NrT rU r* r] rS )r` rO i���r ) r. �random�randnr �_splitr! rl r# r2 �zipr) )r� r! r� �result�expected�res�expr r r � test_splitF s zTestBlock.test_splitc C s& | j }t�t�� |j W d Q R X d S )N)r� r �assert_produces_warning�DeprecationWarningZis_categorical)r� r� r r r � test_is_categorical_deprecatedY s z(TestBlock.test_is_categorical_deprecatedN) �__name__� __module__� __qualname__r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zej�ddd g�d d � �Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zej�ddejfd ejfg�d d � �Zej�dd ejfd ejfd!ejfg�d"d#� �Zd$d%� Zd&d'� Zej�d(d)d*d+d,d-g�d.d/� �Zd0d1� Z d2d3� Z d4d5� Z ej�dd6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGg�dHdI� �Z dJdK� Z dLdM� Z!dNdO� Z"dPdQ� Z#dRdS� Z$ej�ddTdUg�dVdW� �Z%ej�ddXdYdZd[d\g�d]d^� �Z&d_d`� Z'ej�dadbdcdbdddegdfg�dgdh� �Z(diS )j�TestBlockManagerc C s* t d�}|jdkst�t|�dks&t�d S )Nza,b,c: f8-1; d,e,f: f8-2rS r] )r �nblocksr# r2 )r� r r r r r� a s zTestBlockManager.test_attrsc C s� t d�}|j|j }}tt�dg��|d _tt�dg��|d _d}tjt |d�� t ||�}|� � W d Q R X tt�dg��|d _tt�dg��|d _t ||�}|� d� d S )Nz a:bool; a: f8r r* zGaps in blk ref_locs)r� ) r �axesr� r r. r� r% r� r� r# r Z_rebuild_blknos_and_blklocs�iget)r� Ztmp_mgrr� r� �msgr r r r � test_duplicate_ref_loc_failuref s z/TestBlockManager.test_duplicate_ref_loc_failurec C sR t �|�}t �t|�t|�� t|d�s,t�t|d�s:t�|jrDt�|jrNt�d S )N�_is_consolidated�_known_consolidated)r r� �assert_frame_equalr �hasattrr# r� r� )r� r �mgr2r r r r� { s z TestBlockManager.test_pickle� mgr_stringza,a,a:f8z a: f8; a: i8c C s* t |�}t�|�}t�t|�t|�� d S )N)r r r� r� r )r� r� r r� r r r �test_non_unique_pickle� s z'TestBlockManager.test_non_unique_picklec C sP t d�}t�|�}t�t|�t|�� td�}t�|�}t�t|�t|�� d S )Nz a: categoryrR )r r r� r� r r{ Zassert_series_equalr )r� r r� ZsmgrZsmgr2r r r � test_categorical_block_pickle� s z.TestBlockManager.test_categorical_block_picklec C s� t td��}tj�dd�}t|�� t�d�|jd�}t |f|t t�d��gd�}t � |� d�� � |d � t � |� d�� � |d � t � |� d�� � |d � d S )N�abcrT )r! r` rO )r� r� r r* rS )r r� r. r� Zrandr r3 r/ rO r r �assert_almost_equalr� �internal_values)r� �colsr! �blockr r r r � test_iget� s zTestBlockManager.test_igetc C s� t ddd�}|�t|j�dt�dgd �� |�dt�dgd �� t�|� d �� � t�d gd �� t�|� d�� � tjdgd tj d �� t�|� d �� � t�d gd �� t�|� d�� � tjdgd tj d �� d S ) Nz a,b,c: int)rT )rr rG �foorT r* �barr )r" rS ) r �insertr2 r� r. r� �isetr r r� r� �object_)r� r r r r �test_set� s "$"zTestBlockManager.test_setc C s |� t|j�dtjttd�� |�|j�d�t� dt�� |j�d�}|� |�j tj ks\t �|�� }|�|j�d�t� dt�� |j�d�}|� |�j tj ks�t �|� t|j�dtj�t��t�� |j�d�}|� |�j tjks�t �|�|j�d�tj�t�� |� |�j tjk�st �d S )N�baz)r" r� Zquux)r� r2 r� r. �zerosra rL r� �get_loc�repeatr� r" r� r# � consolidater� r� rb r: �int_Zfloat_)r� r �idxr� r r r �test_set_change_dtype� s " z&TestBlockManager.test_set_change_dtypec C sN |j dd�}xft|j|j�D ]T\}}t�|j|j� t|jtj�rX|jj |jj kspt �q |jj j |jj j ks t �q W |� � |j dd�}x�t|j|j�D ]�\}}|j}|j}t�||� t|tj�r�|j }|j }n|j j }|j j }t|t��r|d k�r|d k�sF||k �sFt �q�t|tj��s0||k �sFt �q�|d k�rB|d ks�t �q�W d S )NF)�deepT)r3 r� r� r Z assert_equalr! r$ r. �ndarray�baser# rk Z_consolidate_inplaceZ_ndarrayr ) r� r �cpr� Zcp_blkZbvalsZcpvalsZlbaseZrbaser r r r� � s, $zTestBlockManager.test_copyc C s t d�}|�� jtjks t�d S )Nza: sparse-1; b: sparse-2)r r& r" r. �float64r# )r� r r r r � test_sparse� s z TestBlockManager.test_sparsec C s, t d�}t|j�dkst�t|t�s(t�d S )Nz a: sparse-1; b: sparse-2; c: f8rT )r r2 r� r# r$ r )r� r r r r �test_sparse_mixed� s z"TestBlockManager.test_sparse_mixedzmgr_string, dtypez c: f4; d: f2zc: f4; d: f2; e: f8c C s t |�}|�� j|kst�d S )N)r r& r" r# )r� r� r" r r r r �test_as_array_float� s z$TestBlockManager.test_as_array_floatza: bool-1; b: bool-2z%a: i8-1; b: i8-2; c: i4; d: i2; e: u1zc: i4; d: i2; e: u1c C s t |�}|�� j|kst�d S )N)r r& r" r# )r� r� r" r r r r �test_as_array_int_bool� s z'TestBlockManager.test_as_array_int_boolc C s t d�}|�� jdkst�d S )Nz h: datetime-1; g: datetime-2zM8[ns])r r& r" r# )r� r r r r �test_as_array_datetime s z'TestBlockManager.test_as_array_datetimec C sF t d�}|�d�jdks t�|�d�jdks0t�|�� jdksBt�d S )Nz%h: M8[ns, US/Eastern]; g: M8[ns, CET]r zdatetime64[ns, US/Eastern]r* zdatetime64[ns, CET]rC )r r� r" r# r& )r� r r r r �test_as_array_datetime_tz s z*TestBlockManager.test_as_array_datetime_tz�t�float16�float32r� r� �int64c C st t d�}|dkrtnd }t�|�}|�|�}|�d�jj|ksBt�|�d�jj|ksXt�|�d�jj|ksnt�t d�}t�|�}t� |�� |j|dd�}W d Q R X |�d�jj|ks�t�|�d �jj|ks�t�|�d �jj|ks�t�|�d �jj|ks�t�|�d�jjtj k�st�|�d�jjtj k�s0t�|tj k�rX|�d �jjtj k�spt�n|�d �jj|k�spt�d S ) Nzc: f4; d: f2; e: f8r� r r* rS z6a,b: object; c: bool; d: datetime; e: f4; f: f2; g: f8�ignore)�errorsrU r\ r] rT ) r � FutureWarningr. r" rb r� �typer# r r� r� r� � datetime64)r� r� r �warnZtmgrr r r � test_astype s( z TestBlockManager.test_astypec C sJ dd� }t d�}|�� }|||� t d�}|�dtjdgt tjd�� |�dtjd gt tjd�� |�d tjd gt tjd�� |jd d �}|�d�jtj ks�t �|�d�jtj ks�t �|�d �jtjks�t �|�d�jtj ks�t �|�d�jtj ks�t �t d�}|�dtjdgt tjd�� |�dtjd gt tjd�� |�d tjd gt tjd�� |jd d �}|�d�jtj k�s�t �|�d�jtj k�s�t �|�d �jtjk�s�t �|�d�jtj k�s�t �|�d�jtj k�s�t �|�d�jj�s�t tj��|�d�jtj k�st �|�d�jtj k�s.t �|�d�jtjk�sFt �d S )Nc S s� t | j�}t |j�}t|�t|�ks(t�x<|D ]4}d}x"|D ]}|j|jk�� r<d}P q<W |s.t�q.W x<|D ]4}d}x"|D ]}|j|jk�� rzd}P qzW |slt�qlW dS )z4compare the blocks, numeric compare ==, object don'tFTN)�setr� r2 r# r! rl )Zold_mgr�new_mgrZ old_blocksZ new_blocksrK �found�nb�obr r r �_compare6 s" z/TestBlockManager.test_convert.<locals>._comparez f: i8; g: f8z a,b,foo: object; f: i8; g: f8r �1)r" r* z2.rS zfoo.T)�numericrT rU zEa,b,foo: object; f: i4; bool: bool; dt: datetime; i: i8; g: f8; h: f2r\ r] r� � )r �convertr� r. r� ra r� r� r" r� r# r� r� rf r� r� r� )r� r� r r� r r r � test_convert5 s: z TestBlockManager.test_convertc C sH t jtdd�� td� W d Q R X t jtdd�� td� W d Q R X d S )Nz need to split)r� za: category; b: categoryza: category2; b: category2)r� r� rp r )r� r r r �test_invalid_ea_blocko s z&TestBlockManager.test_invalid_ea_blockc C sX xRdD ]J}t d|� ��}|�� j|ks*t�t d|� d|� ��}|�� j|kst�qW d S )N)r8 r; rC rL rB zM8[ns]zm8[ns]za: z; b: )r r& r" r# )r� r" r r r r �test_interleavev s z TestBlockManager.test_interleave)z a: categoryr; )za: category; b: categoryr; )za: category; b: category2rC )z a: category2rC )za: category2; b: category2rC )za: f8r8 )z a: f8; b: i8r8 )z a: f4; b: i8r8 )za: f4; b: i8; d: objectrC )za: bool; b: i8rC )z a: complexrB )za: f8; b: categoryrC )za: M8[ns]; b: categoryrC )za: M8[ns]; b: boolrC )za: M8[ns]; b: i8rC )za: m8[ns]; b: boolrC )za: m8[ns]; b: i8rC )za: M8[ns]; b: m8[ns]rC c C s� t d�}|�� jdkst�t d�}|�� jdks4t�t d�}|�� jdksNt�t d�}|�� jdksht�t d�}|�� jdks�t�t d �}|�� jdks�t�t d �}|�� jdks�t�t d �}|�� jdks�t�t d �}|�� jd ks�t�t d�}|�� jdk�st�t d�}|�� jdk�s"t�t d�}|�� jdk�s>t�t d�}|�� jdk�sZt�t d�}|�� jdk�svt�t d�}|�� jdk�s�t�t d�}|�� jdk�s�t�d S )Nz a: categoryr; za: category; b: category2rC z a: category2za: f8r8 z a: f8; b: i8z a: f4; b: i8za: f4; b: i8; d: objectza: bool; b: i8z a: complexrB za: f8; b: categoryza: M8[ns]; b: categoryza: M8[ns]; b: boolza: M8[ns]; b: i8za: m8[ns]; b: boolza: m8[ns]; b: i8za: M8[ns]; b: m8[ns])r r& r" r# )r� r� r" r r r r �test_interleave_dtype~ s@ z&TestBlockManager.test_interleave_dtypec C s� |� |j�d�tj�t�� |� |j�d�tj�t�� |� |j�d�tj�t�� |� |j�d�tj�t�� |� |j�d�tj�t�� |�� }|jdks�t �|�� � � }|jdks�t �t |j d j t�s�t �t�|j d j jtjt|j�tjd �� d S ) N�frG rK �g�hrU r* r )r" )r� r� r� r. r� r� ra r� r� r# �get_numeric_datar$ r� r% r r r r& r/ r2 r� )r� r Zconsr r r � test_consolidate_ordering_issues� s z1TestBlockManager.test_consolidate_ordering_issuesc C s� t d�}|jddddgdd�}|jdks,t�t�|jtddddg�� t�|� d �� � |� d�� � � t�|� d�� � |� d �� � � t�|� d�� � |� d�� � � t�|� d �� � |� d �� � � d S ) Nz3a: f8; b: i8; c: f8; d: i8; e: f8; f: bool; g: f8-2r rW rV rG r )�axisrS r] r* rT ) r � reindex_axisr� r# r �assert_index_equalr� r r� r� r� )r� r � reindexedr r r �test_reindex_items� s z#TestBlockManager.test_reindex_itemsc C s: t ddd�}|�dtjdddgtjd�� |�� }t�|jt d d d d g�� t� |� |j� d ��� � |� |j� d ��� � � |�|j� d �t�d ddg�� t� |� |j� d ��� � t�d ddg�� |jdd�}t�|jt d d d d g�� |�|j� d �t�dddg�� t� |� |j� d ��� � t�d ddg�� d S )Nz[int: int; float: float; complex: complex;str: object; bool: bool; obj: object; dt: datetime)rT )rr r\ r* rS rT )r" r: r7 rB rL g Y@g i@g �r@T)r3 g @�@g @�@g p�@)r r� r. r� r� r r r r� r r� r� r� r� )r� r r� Znumeric2r r r �test_get_numeric_data� s( z&TestBlockManager.test_get_numeric_datac C s t ddd�}|�dtjdddgtjd�� |�� }t�|jt dd g�� t� |� |j� d��� � |� |j� d��� � � |�d t�dddg�� t�|� |j� d��� � t�dddg�� |jdd �}|�d t�dddg�� t�|� |j� d��� � t�dddg�� d S ) Nz[int: int; float: float; complex: complex;str: object; bool: bool; obj: object; dt: datetime)rT )rr r] TF)r" rL rM r )r3 )r r� r. r� r� Z get_bool_datar r r� r r� r� r� r� r )r� r ZboolsZbools2r r r �test_get_bool_data� s$ z#TestBlockManager.test_get_bool_datac C s t td�� d S )Nu b,א: object)�reprr )r� r r r � test_unicode_repr_doesnt_raise s z/TestBlockManager.test_unicode_repr_doesnt_raiseza,b,c: i8-1; d,e,f: i8-2za,a,a: i8-1; b,b,b: i8-2c C s2 t |�}t|jd d d� |j�}|�|�s.t�d S )NrY )r r r� r� �equalsr# )r� r� �bm1Zbm2r r r � test_equals s z TestBlockManager.test_equalsz a:i8;b:f8za:i8;b:f8;c:c8;d:bza:i8;e:dt;f:td;g:stringza:i8;b:category;c:category2zc:sparse;d:sparse_na;b:f8c C sR t |�}t�|j�}x8|D ]0}tt|�|j�}|�|�s<t�|�|�st�qW d S )N) r � itertools� permutationsr� r r� r� r r# )r� r� ZbmZ block_permsZbm_permZbm_thisr r r �(test_equals_block_order_different_dtypes( s z9TestBlockManager.test_equals_block_order_different_dtypesc C s. t ddd�}|�� �� dddddgks*t�d S ) Nr8 r\ )rz g g �?g @g @g @)r{ �external_values�tolistr# )r� r r r r �test_single_mgr_ctor; s z%TestBlockManager.test_single_mgr_ctor�valuer* �TruerS rT g @c C sL t d�}dt|�j� d�}tjt|d�� |jdgdg|d� W d Q R X d S )Nza,b,c: i8-1; d,e,f: i8-2z9For argument "inplace" expected type bool, received type �.)r� r* rS )�inplace)r r� r� r� r� rp Z replace_list)r� r r r� r r r �test_validate_bool_args? s z(TestBlockManager.test_validate_bool_argsN))r� r� r� r� r� r� r� �mark� parametrizer� r� r� r� r� r� r� r� r. r� r� r� rf r� r� r� r� r� r� r r r r r r r r r r r r r! r r r r r� ` sz ' :( r� c C s | j dkr| �� S | �� S )Nr* )rO r r& )r r r r � _as_arrayK s r$ c @ s� e Zd Zede�ede�edefd�edefd�edefd�edefd�gZej� de�d d � �Z ej� de�d d � �Z ej� de�ej� d de j dg�dd� ��Zej� de�ej� d de j dg�dd� ��ZdS )� TestIndexingr8 r; za,b,c,d,e,f: f8)rr za,b,c,d,e,f: i8z a,b: f8; c,d: i8; e,f: stringza,b: f8; c,d: i8; e,f: f8r c C s� dd� }|j dks t|j ���xnt|j �D �]^}|||td �� |||td�� |||td�� |||tdd�� |||tddd �� |j dk r*|||tjg tjd �� |||tj|j| tjd �� |||tj |j| tjd �� |j| dk�r2|||t� |j| �d dk� |||tjd d d gtjd �� |||g � |||t t|j| ��� |j| dkr*|||dddg� |||d d dg� q*W d S )Nc S s� t | �}t|tj�rb| j| }t|�rbt|�rbt|�t|�krbt�|tjt|�t|� td�g�}t|t �r|| j ||d�}n"| j dkr�|dkr�| � |�}nd S t d �f| |f }t j|| t |�dd� t �| j| | |j| � d S )N)r" )r r* r F)� check_dtype)r$ r$ r. r� r� r2 Z concatenater� rL ry Z get_slicerO Z getitem_mgrr r r )r r Zslobjrv �axZslicedZ mat_slobjr r r �assert_slice_okf s z4TestIndexing.test_get_slice.<locals>.assert_slice_okrS rT �d r* rU r ���)r" TFrY ���) rO r# �rangery r. r� rf re r, r� r/ r� )r� r r( r' r r r �test_get_sliced s* zTestIndexing.test_get_slicec C s� dd� }x�t |j�D ]x}|||g d� |||dddgd� |||tt |j| ��d� |j| dkr|||dddgd� |||dd d gd� qW d S ) Nc S sT t | �}| �||�}tjt�|||�t |�dd� t�| j| �|�|j| � d S )NF)r& )r$ Ztaker r r. r r� )r r �indexerrv Ztakenr r r �assert_take_ok� s z.TestIndexing.test_take.<locals>.assert_take_ok)r. r rT r* rS rY r* r+ )r, rO r� r, )r� r r/ r' r r r � test_take� s zTestIndexing.test_taker^ Ng Y@c C s dd� }x�t |j�D ]�}|||tg �|� ||||j| |� ||||j| dddg |� |||tdddg�|� |||td|j| d dg�|� |j| dkr||||j| d d� |� ||||j| dd d � |� ||||j| dd d dd d g |� qW d S ) Nc S s` t | �}| j| �|�}| j|||d�}tjtj||||d�t |�dd� t�|j| |� d S )N)r^ F)r& ) r$ r� Zget_indexer_forr r r �algos�take_ndr )r r � new_labelsr^ rv r. r r r r �assert_reindex_axis_is_ok� s zATestIndexing.test_reindex_axis.<locals>.assert_reindex_axis_is_okr r� r� r� rT r+ rY r* rS )r, rO r r� r, )r� r^ r r4 r' r r r �test_reindex_axis� s z TestIndexing.test_reindex_axisc C s� dd� }�x�t |j�D �]z}|||tg �tjg tjd�|� ||||j| t�|j| �|� |||tdg|j| �t�|j| �|� ||||j| d d d� t�|j| �|� ||||j| t�|j| �d d d� |� |||tdddg�t�dddg�|� |||tdddg�t�dddg�|� |||td|j| d dg�t�dddg�|� |j| d kr|||tdddg�t�dd d g�|� qW d S ) Nc S sV t | �}tj||||d�}| j||||d�}tj|t |�dd� t�|j| |� d S )N)r^ F)r& )r$ r1 r2 Zreindex_indexerr r r r� )r r r3 r. r^ rv Z reindexed_matr r r r � assert_reindex_indexer_is_ok� s zGTestIndexing.test_reindex_indexer.<locals>.assert_reindex_indexer_is_ok)r" r� rY r� r� r rT r* rS ) r, rO r r. r� r� r� r/ r, )r� r^ r r6 r' r r r �test_reindex_indexer� sB **$$ z!TestIndexing.test_reindex_indexer)r� r� r� r{ ra r ZMANAGERSr� r" r# r- r0 r. rn r5 r7 r r r r r% Q s < r% c @ s� e Zd Zej�dedd�dfeddd�dfeddd�dfeddd�dfeddd�dfg�dd � �Zej�d eddd�eddd�g�d d � �Zd d� Z ej�d edd�edd�eddd�eddd�edd�edd�edd�eddd�eddd�eddd�g �dd� �Z ej�d edd�edd�edd�eddd�eddd�g�dd� �Z ej�ddgeddd�fdgeddd�fdddgeddd�fdddgeddd�fddgeddd�fddgeddd�fg�dd � �Z ej�d g dgdd d gd gdgddddgd dddgdddgdddgg �d!d"� �Z ej�dedd�dddgfedd�g fedd�g fg�d#d$� �Zej�d%edd�dddgfedd�g fedd�g feddd�dddgfg�d&d'� �Zd(d)� Zej�d*edd�dg fedd�ddddgfeddd�ddddgfdddgddddgfedd�dg fedd�dd+d,d-gfeddd�dd-d,d+gfdddgdd+d,d.gfedd�dg fedd�ddddgfdddgddddgfg �d/d0� �Zej�d1edd�dddgg�d2d3� �ZdS )4�TestBlockPlacementz slc, expectedr rU rS rT r* rY c C s t t|��|kst�d S )N)r2 r r# )r� �slcr� r r r �test_slice_len s z!TestBlockPlacement.test_slice_lenr9 c C s* d}t jt|d�� t|� W d Q R X d S )Nzslice step cannot be zero)r� )r� r� rp r )r� r9 r� r r r �test_zero_step_raises s z(TestBlockPlacement.test_zero_step_raisesc C s. t ddd�}t|�}|jt dd d�ks*t�d S )NrT rY r* )ry r r. r# )r� r9 �bpr r r �!test_slice_canonize_negative_stop! s z4TestBlockPlacement.test_slice_canonize_negative_stopNr6 c C s* d}t jt|d�� t|� W d Q R X d S )Nzunbounded slice)r� )r� r� rp r )r� r9 r� r r r �test_unbounded_slice_raises( s z.TestBlockPlacement.test_unbounded_slice_raisesr) c C s t |�jrt�d S )N)r � is_slice_liker# )r� r9 r r r �test_not_slice_like_slices> s z-TestBlockPlacement.test_not_slice_like_sliceszarr, slc�e r\ � �� c C s t |�j|kst�d S )N)r �as_slicer# )r� r4 r9 r r r � test_array_to_slice_conversionK s z1TestBlockPlacement.test_array_to_slice_conversionr4 r* r+ i����c C s t |�jrt�d S )N)r r? r# )r� r4 r r r �test_not_slice_like_arraysY s z-TestBlockPlacement.test_not_slice_like_arraysc C s t t|��|kst�d S )N)r� r r# )r� r9 r� r r r �test_slice_iterj s z"TestBlockPlacement.test_slice_iterzslc, arrc C s" t �t|�jtj|tjd�� d S )N)r" )r r r r& r. �asarrayr� )r� r9 r4 r r r � test_slice_to_array_conversionq s z1TestBlockPlacement.test_slice_to_array_conversionc C sz t tdd��}|�d�jtddd�ks*t�|�t�d��jtddd�ksLt�t|�t�ddd���dddddgksvt�d S )Nr r\ r* r] r6 rS rY )r ry �addrD r# r. r/ r� )r� Zbplr r r �test_blockplacement_add s "z*TestBlockPlacement.test_blockplacement_addzval, inc, expected� � � � c C s t t|��|��|kst�d S )N)r� r rJ r# )r� �val�incr� r r r �test_blockplacement_add_int� s z.TestBlockPlacement.test_blockplacement_add_intrP c C s0 d}t jt|d�� t|��d� W d Q R X d S )Nziadd causes length change)r� i����)r� r� rp r rJ )r� rP r� r r r �"test_blockplacement_add_int_raises� s z5TestBlockPlacement.test_blockplacement_add_int_raises)r� r� r� r� r" r# ry r: r; r= r> r@ rE rF rG rI rK rR rS r r r r r8 s� * 8 ""r8 c @ s� e Zd Zejdd� dd� dd� dd� dd� dd� dd� d d� gd �d d � �Zd d� Zej�de j e j e j g�dd� �Z ej�de j e j e j g�dd� �Zdd� Zdd� Zeed�dd�Zed�d d �Zeed�d d �Zeed�d d!�Zd"S )#�TestCanHoldElementc C s | S )Nr )r+ r r r r� � r� zTestCanHoldElement.<lambda>c C s | � � S )N)Z to_series)r+ r r r r� � r� c C s | j S )N)rk )r+ r r r r� � r� c C s t | �S )N)r� )r+ r r r r� � r� c C s | � t�S )N)rb rC )r+ r r r r� � r� c C s t �| �S )N)r. rH )r+ r r r r� � r� c C s | d S )Nr r )r+ r r r r� � r� c C s | d d� S )Nr r )r+ r r r r� � r� )r c C s |j S )z� Functions that take an Index and return an element that should have blk._can_hold_element(element) for a Block with this index's dtype. )r )r� r r r r �element� s zTestCanHoldElement.elementc C s� t ddg�}|�g �st�t�|j�� �}|�d �s8t�d |d< |d tjksRt�t� d�t ddd�g}x"|D ]}|�|�s�t�||d< qnW t ddd�}|�|�r�t�d}t j t|d�� ||d< W d Q R X d S )Nr r z 2010-10-10i� r6 zLvalue should be a 'Timestamp', 'NaT', or array of those. Got 'date' instead.)r� )rx �_can_hold_elementr# �pdr� r! rd ZNaTr. r� r r r� r� � TypeError)r� r� r4 �valsrP r� r r r �$test_datetime_block_can_hold_element� s z7TestCanHoldElement.test_datetime_block_can_hold_elementr" c C sB t jdddg|d�}t�|�}t|jdgdd�}|�g �s>t�d S )Nr* rT rU )r" rS )rO )r. r� r � from_breaksr rk rV r# )r� r" rU r4 �iir� r r r �(test_interval_can_hold_element_emptylist� s z;TestCanHoldElement.test_interval_can_hold_element_emptylistc C s t jddddg|d�}t�|�}t|jdgdd�}||�}| �||d� |�|�sVt�tj|d d � d d �}||�}| �||d � |�|�r�t�t�t d�t d�t d�g�}||�}| �||d � |�|�r�t�t�t d�t d�t d�g�} || �}| �||d � |�|��rt�d S ) Nr* rT rU � )r" rS )rO TrY Zneither)�closedF) r. r� r r[ r rk �check_series_setitemrV r# r r ) r� r" rU r4 r\ r� �elemZii2Zii3Zii4r r r � test_interval_can_hold_element� s$ z1TestCanHoldElement.test_interval_can_hold_elementc C s2 t dddd�}t|jdgdd�}|�g �s.t�d S )N�2016rT rF )�periods�freqr* rS )rO )r r rk rV r# )r� �pir� r r r �&test_period_can_hold_element_emptylist� s z9TestCanHoldElement.test_period_can_hold_element_emptylistc C sx t dddd�}||�}| �||d� |�d�d d� }||�}| �||d� |�d �d d� }||�}| �||d� d S ) Nrc rT rF )rd re T�DrY F�S)r r` ZasfreqZ to_timestamp)r� rU rf ra Zpi2Zdtir r r � test_period_can_hold_element� s z/TestCanHoldElement.test_period_can_hold_element)�indexr c C s | � |||� | �|||� d S )N)r` �check_frame_setitem)r� ra rk r r r r � check_setting s z TestCanHoldElement.check_setting)r c C s2 |j jd }|r |�|�s.t�n|�|�r.t�d S )Nr )r� r� rV r# )r� �objra r r� r r r �check_can_hold_element s z)TestCanHoldElement.check_can_hold_elementc C sh |j �� }t|�}| �|||� t|�r2||d< n||d t|��< |rV|j|ksdt�n|jt ksdt�d S )Nr ) rk r3 r ro r r2 r� r# r" rC )r� ra rk r r4 �serr r r r` s z'TestCanHoldElement.check_series_setitemc C s| |j �� }t|�}| �|||� t|�r4||jd< n||jd t|��df< |rf|jjd j |ksxt �n|j d t ksxt �d S )N)r r r ) rk r3 r ro r Zilocr2 r� r� r! r# ZdtypesrC )r� ra rk r r4 r� r r r rl % s z&TestCanHoldElement.check_frame_setitemN)r� r� r� r� �fixturerU rZ r" r# r. r� Zuint64r� r] rb rg rj r rL rm ro r` rl r r r r rT � s$ " "rT c @ s e Zd Zdd� ZdS )�TestShouldStorec C sp t dddg�}t|�}|jjd }|�|�s0t�|�|d d� �sFt�|�|�� �rXt�|�t�|��rlt�d S )NrF �B�Cr rY ) r r r� r� Z should_storer# Z as_orderedr. rH )r� �catr� r� r r r � test_should_store_categorical; s z-TestShouldStore.test_should_store_categoricalN)r� r� r� rv r r r r rr : s rr c C sF t �ddg�}td�}d}tjt|d�� | ||dd� W d Q R X d S )Ng �?g @rS z:Wrong number of dimensions. values.ndim != ndim \[1 != 2\])r� )rO )r. r� ry r� r� rp )r r! r` r� r r r �test_validate_ndimK s rw c C sf t dddddg�} tdddg��| �}ttdddg���| �}|jjd jj|jjd jjksbt�d S )Nr r* rS rT rU ) r r Zreindexr r� r� r% r. r# )r� rV rK r r r �test_block_shapeT s rx c C s� t j�t�ddg��}| |tt|��|jd�}|jj dks>t �|j dksLt �| t kr�| |tt|��|j|jd�}|jj dks�t �|j dks�t �| |� � tt|��|j|jd�}|jj dks�t �|j dks�t �d S )Nr* rS )rO )rI �uF)r" rO )rW ZarraysZ PandasArrayr. r� ry r2 rO r" �kindr# Z is_extensionr Zto_numpy)r r4 r� r r r � test_make_block_no_pandas_array\ s r{ c C sB t td��} | jjd }t�t�� t|| jdd� W d Q R X d S )NrT r T)Zfastpath) r r, rk r� r r� r� r rk )rp r� r r r �-test_single_block_manager_fastpath_deprecatedt s r| )N)N)Br r r rh �numpyr. r� Zpandas._libs.internalsr Z pandas.util._test_decorators�utilZ_test_decoratorsrQ Zpandas.core.dtypes.commonr ZpandasrW r r r r r r r r r Zpandas._testingZ_testingr Zpandas.core.algorithms�coreZ algorithmsr1 Zpandas.core.arraysr r r Zpandas.core.internalsr r r Z pandas.core.internals.blocksr r Z skip_array_manager_invalid_testZ pytestmarkrq r r r) r5 ra rx r{ r r� r� r$ r% r8 rT rr rw rx r{ r| r r r r �<module> sP , V ?l n =
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка