Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/frame/__pycache__/test_block_internals.cpython-37.pyc
Назад
B �5�g$9 � @ s� d dl m Z mZ d dlmZ d dlZd dlZd dlZd dlm Z d dl m m Z d dlZd dlmZmZmZmZmZmZmZ d dlmZ d dlmZm Z e j Z G dd� d�Z d d � Z d d � Z!dS ) � )�datetime� timedelta)�StringION)�PerformanceWarning)� Categorical� DataFrame�Series� Timestamp�compat� date_range�option_context)� NumericBlock� ObjectBlockc @ 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"� Zd#d$� Zd%S )&�TestDataFrameBlockInternalsc C sx t dddd�}|d }td|i�}|d jjd ks6t�tj|jd< |d jjd ksVt�|jdksdt�|d |kstt�d S ) N�20130101� z US/Eastern)�periods�tz� �B)r r �D)r r Z_values�freq�AssertionError�pdZNaTZiloc)�selfZdti�ts�df� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/frame/test_block_internals.py�,test_setitem_invalidates_datetime_index_freq( s zHTestDataFrameBlockInternals.test_setitem_invalidates_datetime_index_freqc C sX t |jtd�}t |jtd�}t�||� t |jtjd�}t |jtjd�}t�||� d S )N)�dtype)r �_mgr�intZ_series�tm�assert_frame_equal�np�int32)r � float_frameZcasted�expectedr r r �test_cast_internals9 s z/TestDataFrameBlockInternals.test_cast_internalsc C s� d|d< |� � }t|jj�dks$t�|� � }||k s8t�t�||� d|d< t|jj�dks`t�|�� }|d kstt�t|jj�dks�t�d S )Ng @�Er g @�Fr )Z _consolidate�lenr! �blocksr r# r$ Z_consolidate_inplace)r r' Z consolidatedZreconsZ return_valuer r r �test_consolidateB s z,TestDataFrameBlockInternals.test_consolidatec C s8 |� � }x*ttd�td��D ]}t|�|t|�< q W d S )N�A�Z)�copy�range�ord�chr)r r' �frame�letterr r r �test_consolidate_inplaceS s z4TestDataFrameBlockInternals.test_consolidate_inplacec C s. d|d< |j �� rt�|j}|j �� s*t�d S )Ng @r* )r! Zis_consolidatedr �values)r r' �_r r r �test_values_consolidateZ s z3TestDataFrameBlockInternals.test_values_consolidatec C st d|j d< |j d dk�� s t�d|d< |d }d|j d< |j d dk�� sPt�|d |k s`t�|dk�� spt�d S )N� g @r* � � )r8 �allr )r r' �colr r r �test_modify_values` s z.TestDataFrameBlockInternals.test_modify_valuesc C s< d|d< |j �� }d||dk< d||dk< t�||j � d S )Ng @r* � r )r8 r1 r# Zassert_almost_equal)r r' r( r r r �test_boolean_set_unconsn s z3TestDataFrameBlockInternals.test_boolean_set_unconsc C s t ddgi�}|d }tt�dgtj�dd�}t�||� t ddgi�}|d }tt�dgtj�dd�}t�||� t dtddd�dgi�}|d }tt�tddd�dgtj �dd�}t�||� t dd dgi�}|d }tt�tj dgtj �dd�}t�||� t ddd gi�}|d }tt�dd gtj �dd�}t�||� t dd d gi�}|d }tt�d d gtj �dd�}t�||� t dd d gi�}|d }tt�d d gtj �dd�}t�||� t dd dgi�}|d }tt�d dgtj �dd�}t�||� t ddd gi�}|d }tt�dtj gtj �dd�}t�||� t dd d gi�}|d }tt�d tj gtj �dd�}t�||� t dd ddd gi�}|d }tt�d ddd gtj �dd�}t�||� t dd dtddd�d gi�}|d }tt�d dtddd�d gtj �dd�}t�||� d S )Nr/ l ���� )�namel i� r Tg �?rA y �? @r g @g @i� ) r r r% Zasarray�int64r# �assert_series_equalZuint64r Zobject_�nanZfloat_Zcomplex_)r r �resultr( r r r � test_constructor_with_convertw sd $z9TestDataFrameBlockInternals.test_constructor_with_convertc C s� t ddd�tjt ddd�gt ddd�t ddd�t ddd�gg}t|�}|j}tddi�}t �� |d< tddd �|d < |d jd ks�t �|d jd ks�t �|j}tt�d �gd t�d�t�d�t�d�g t d�ddd g d�}t � ||� d S )Ni� r r; rA i� r zdatetime64[ns]r )�days�secondsr zM8[ns]zm8[ns]�float64� �objectztimedelta64[ns]ZABCDZfoo)�index) r r% rF r Zdtypesr �nowr r r �listr# rE )r �float_string_frame�datar rG r( r r r � test_construction_with_mixed� s" $ z8TestDataFrameBlockInternals.test_construction_with_mixedc C s� t jdddgdd�}ttd�d�}||d< tdtjddd d �itd�d�}t�||� ttd �t d dd �d �td�d�}ttd�d�}t � d�|d< t jdddgdd�|d< t�||� d S )Nr rA r ztimedelta64[s])r )rN r/ z00:00:01�s)r r r )r )�dt1�dt2z 2013-01-01rU z 2013-01-02z 2013-01-03z datetime64[D]rV ) r% �arrayr r2 r Ztimedelta_ranger# r$ r r Z datetime64)r �arrr r( r r r �"test_construction_with_conversions� s z>TestDataFrameBlockInternals.test_construction_with_conversionsc C s� dd� }d}t jt|d�� |dddg� W d Q R X d}tjt|d�� |d � W d Q R X tjt|d�� |d � W d Q R X t�� s�|d � d S ) Nc S s2 t t�tddd�ddfd��}t|dddg| d �S ) Ni� r Zaa� � r/ r �C)rR �columnsr )rP � itertools�repeatr r )r rR r r r �f� s zGTestDataFrameBlockInternals.test_constructor_compound_dtypes.<locals>.fz@compound dtypes are not implemented in the DataFrame constructor)�match)r/ z datetime64[h])r �str)r\ r& zBeither all columns will be cast to that dtype, or a TypeError willrD rK zM8[ns])�pytestZraises�NotImplementedErrorr# �assert_produces_warning� FutureWarningr Zis_platform_windows)r r` �msgZdepr_msgr r r � test_constructor_compound_dtypes� s z<TestDataFrameBlockInternals.test_constructor_compound_dtypesc C sP t � }t�|�}t�||� |jj t�|�}t|� t�|�}t�||� d S )N)r r# Zround_trip_pickler$ r! �ndim�repr)r rQ Ztimezone_frameZ empty_frameZ unpickledr r r � test_pickle s z'TestDataFrameBlockInternals.test_picklec C s� d}t jt|�ddgd�}|j}|j|_|�d�}|�d�}d|j_|j }|j|_|�d�}|�d�}d|j_|j|_|j|_ t � t � |j�|j� t � t � |j �|j� d S ) Nz�starting,ending,measure 2012-06-21 00:00,2012-06-23 07:00,77 2012-06-23 07:00,2012-06-23 16:30,65 2012-06-23 16:30,2012-06-25 08:00,77 2012-06-25 08:00,2012-06-26 12:00,0 2012-06-26 12:00,2012-06-27 08:00,77 r r )Z parse_datesz US/Eastern�UTC�starting�ending) r Zread_csvr rm r8 rN Z tz_localizeZ tz_convertrC rn r# Zassert_index_equalZ DatetimeIndex)r rR r Z ser_startingZ ser_endingr r r �test_consolidate_datetime64! s z7TestDataFrameBlockInternals.test_consolidate_datetime64c C s |j r t�|j st�d S )N)Z_is_mixed_typer )r r' rQ r r r �test_is_mixed_type@ s z.TestDataFrameBlockInternals.test_is_mixed_typec C s� t dd ��v ttj�d�ddd�}t|� |d �d�|d< tj|d d < t|� |�� }|d �� }t� |d d �s|t �W d Q R X d S ) NZchained_assignment)rL rL )�a�b�c�d)�er` �g�h)rN r] ru rM rv rs ) r r r% �randomrj Zastype�NaN�sumr �isnar )r �YrG �expr r r � test_stale_cached_series_bug_473D s z<TestDataFrameBlockInternals.test_stale_cached_series_bug_473c C s� t ddgd�}tj|d< i }t�t��X xPt|j�D ]B\}}x8tdd�D ]*}||krfd||< tj||< ||| |< qHW q4W W d Q R X d}t |j t � || �|gf �}t |j t � || �|gf �}|| kr�dks�n t �d S )Nr r )rN �d �� )r r% rF r# re r � enumeraterN r2 r, �locr r{ r ) r r ZwasCol�i�dtr? Zmyid�first�secondr r r �$test_strange_column_corruption_issueU s z@TestDataFrameBlockInternals.test_strange_column_corruption_issuec C sT t dddg�j}td|i�}tddddgi�}t�||� t|jjd t�sPt �d S )Nr rA r r/ r ) r rW r r# r$ � isinstancer! r- r r )r rX rG r( r r r � test_constructor_no_pandas_arrayj s z<TestDataFrameBlockInternals.test_constructor_no_pandas_arrayc C s� t ddddgddddgd ��}tj�tjddd dgtd ��|d< t ddddgddddgtj�tjddd dgtd ��d ��}t|d jj d �t ks�t �t|d jj d �t ks�t �t � ||� d S ) Nr rA r rL rq rr rs rt )rq rr )r )rq rr rs r )r r ZarraysZ PandasArrayr% rW rM �typer! r- r r r# r$ )r r Zdf2r r r �!test_add_column_with_pandas_arrays s " $ z=TestDataFrameBlockInternals.test_add_column_with_pandas_arrayN)�__name__� __module__� __qualname__r r) r. r7 r: r@ rB rH rS rY rh rk ro rp r~ r� r� r� r r r r r ' s$ C r c C s` t dtddd gdd�i�} | d jddd� t| jjd jt�sDt�| � � � � � � dks\t�d S ) Nrq r rA �category)r T)Zinplacer ) r r Zfillnar� r! r- r8 r r Zisnullrz )r r r r �+test_update_inplace_sets_valid_block_values� s r� c C s� t � } tdgtd�| d< tdgtd�| d< | d dk | | d dk d| jd< t dgdgd�td�}t�| |� | jd dks�t�d S ) Nrq )r �col1r �col2r/ )r r� )r� r� )r r rM �atr# r$ r )r r( r r r �$test_nonconsolidated_item_cache_take� s r� )"r r �ior r^ �numpyr% rc Z pandas.errorsr Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr r r r r r r r Zpandas._testingZ_testingr# Zpandas.core.internalsr r Z skip_array_manager_invalid_testZ pytestmarkr r� r� r r r r �<module> s $ ^
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка