Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/__pycache__/test_combine_concat.cpython-37.pyc
Назад
B �5�g[ � @ s d dl Zd dlZd dlZd dlmZ d dlm Z G dd� d�Z ej�dej dddgd d �e�d d �fej ddd gdd �e�dd �fej dddged �e�ed �fej dddgdd �e�d �fej ddd gdd �e�d�fej dddgdd �e�e�fg�dd� �ZdS )� N)�SparseArrayc @ s@ e Zd Zej�dddg�dd� �Zej�dddg�dd� �ZdS ) �TestSparseArrayConcat�kind�integer�blockc C sn t ddddg|d�}t ddddg|d�}t �||g�}tjdddddgdd�}t�|j|� |j|ksjt�d S )N� r � )r �int64)�dtype) r �_concat_same_type�np�array�tm�assert_numpy_array_equal� sp_valuesr �AssertionError)�selfr �a�b�result�expected� r �d/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/sparse/test_combine_concat.py� test_basic s z TestSparseArrayConcat.test_basicc C s~ |dkrdnd}t ddddg|d�}t ddddg|d�}t �||g�}tjdddddgdd�}t�|j|� |j|kszt�d S ) Nr r r r r )r r )r ) r r r r r r r r r )r r �otherr r r r r r r �test_uses_first_kind s z*TestSparseArrayConcat.test_uses_first_kindN)�__name__� __module__�__qualname__�pytest�mark�parametrizer r r r r r r s r zother, expected_dtype� � � r )r g �?g @g @Zfloat64r r �c�categoryc C s� t jdddgt �dd�d�}t j|| gdd�}t �t|�t| � ��|�}t�||� t j| |gdd�}t �t| �t|� ��|�}t�||� d S )Nr r r r )r T)Zignore_index)�pd�Series�SparseDtype�concat�listZastyper Zassert_series_equal)r Zexpected_dtypeZs_sparser r r r r �test_concat_with_non_sparse# s r, )�numpyr r Zpandasr'