Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/extension/base/__pycache__/constructors.cpython-37.pyc
Назад
B �5�g � @ sT d dl Zd dlZd dlZd dlmZ d dlmZm Z d dl mZ G dd� de�ZdS )� N)�ExtensionArray)�DatetimeTZBlock�ExtensionBlock)�BaseExtensionTestsc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zdd� Ze j �d ddg�dd� �Zdd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd S )!�BaseConstructorsTestsc C sP t |�j||jd�}| �||� |d d� }t |�j||jd�}| �||� d S )N)�dtyper )�type�_from_sequencer �assert_extension_array_equal)�self�data�result� r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/extension/base/constructors.py�test_from_sequence_from_cls s z1BaseConstructorsTests.test_from_sequence_from_clsc C s6 |d |d |d g}|� |�}t|t|��s2t�d S )Nr � � )r � isinstancer �AssertionError)r r Zscalarsr r r r �test_array_from_scalars s z-BaseConstructorsTests.test_array_from_scalarsc C s� t �|�}|j|jkst�t|�t|�ks.t�t|jd�rTt|jjd t t f�sTt�|jj|ksdt�t �|�}|j|jks~t�t|jd�r�t|jjd t t f�s�t�d S )N�blocksr )�pd�Seriesr r �len�hasattr�_mgrr r r r �array)r r r Zresult2r r r �test_series_constructor s z-BaseConstructorsTests.test_series_constructorc C st t jdddg|d�}t j|gd dddg|d�}| �||� t jg |d�}t jg t jg dd�|d�}| �||� d S )Nr r � )�indexr �object)r )r r �assert_series_equal�Index)r r �na_valuer �expectedr r r �*test_series_constructor_no_data_with_index) s z@BaseConstructorsTests.test_series_constructor_no_data_with_indexc C sB t j|dddg|d�}t j|gd dddg|d�}| �||� d S )Nr r r )r r )r r r! )r r r# r r$ r r r �,test_series_constructor_scalar_na_with_index3 s zBBaseConstructorsTests.test_series_constructor_scalar_na_with_indexc C s| |d }t j|dddg|d�}t j|gd dddg|d�}| �||� t j|dg|d�}t j|gdg|d�}| �||� d S )Nr r r r )r r Zfoo)r r r! )r r r Zscalarr r$ r r r �)test_series_constructor_scalar_with_index8 s z?BaseConstructorsTests.test_series_constructor_scalar_with_index�from_seriesTFc C s� |rt �|�}t �d|i�}|jd |jks0t�|jt|�dfksFt�t|j d�rlt |j jd tt f�slt�t |j jd t�s�t�d S )N�Ar r r )r r � DataFrame�dtypesr r �shaper r r r r r r �arraysr )r r r( r r r r �$test_dataframe_constructor_from_dictB s z:BaseConstructorsTests.test_dataframe_constructor_from_dictc C sz t �t �|��}|jd |jks$t�|jt|�dfks:t�t|j d�r`t |j jd tt f�s`t�t |j jd t�svt�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 �test_dataframe_from_seriesM s z0BaseConstructorsTests.test_dataframe_from_seriesc C sB d}t jt|d��& tj|d d� dddddgd� W d Q R X d S ) Nz;Length of values \(3\) does not match length of index \(5\))�matchr r r r � )r )�pytestZraises� ValueErrorr r )r r �msgr r r �)test_series_given_mismatched_index_raisesU s z?BaseConstructorsTests.test_series_given_mismatched_index_raisesc C s� |j }t�|�}tjt|�|d�}| �||� tjt|�t|�d�}| �||� t�|��|�}tjt|�|d�}| �||� tjt|�t|�d�}| �||� d S )N)r ) r r r �listr! �strr* Zastype�assert_frame_equal)r r r r$ r r r r �test_from_dtypeZ s z%BaseConstructorsTests.test_from_dtypec C s t �|�}| �||� d S )N)r r r )r r r r r r �test_pandas_arrayn s z'BaseConstructorsTests.test_pandas_arrayc C s: t j|t�t�d�}t j�tj|td��}| �||� d S )N)r ) r r �npr r r- ZPandasArrayZasarrayZassert_equal)r r r r$ r r r �test_pandas_array_dtypes s z-BaseConstructorsTests.test_pandas_array_dtypec C sF t jdg|d�}t jdt jg |d�it jg dd�d�}| �||� d S )N�a)�columnsr )r r )r )r r* r r"