Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/reshape/__pycache__/test_get_dummies.cpython-37.pyc
Назад
B �5�g�\ � @ sv d dl Z d dlZd dlZd dlmZ d dlZd dlmZm Z m Z m Z m Z d dl mZ d dlmZmZ G dd� d�ZdS )� N)�is_integer_dtype)� Categorical�CategoricalIndex� DataFrame�Series� get_dummies)� SparseArray� SparseDtypec @ s( e Zd Zejdd� �Zejddejedgd�dd� �Z ejd d gd�d d � �Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z ej � d1d2e d3d4gi�ie d5d6giej d7�fd2e d8d3gi�ie d9d6giej d7�fe d8d4gi�d3d:�e d5d6giej d7�fe d8d4gi�d3d;�e d<d6giej d7�fg�d=d>� �Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)ej � dOdPdQg�dRdS� �Z*ej � d dPdQg�dTdU� �Z+dVdW� Z,dXdY� Z-ej � dZd[g�d\d]� �Z.dS )^�TestGetDummiesc C s" t dddgdddgdddgd��S )N�a�b�c� � � )�A�B�C)r )�self� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/reshape/test_get_dummies.py�df s zTestGetDummies.df�uint8�i8N)�paramsc C s t �|j�S )N)�np�dtype�param)r �requestr r r r s zTestGetDummies.dtypeZdense�sparsec C s |j dkS )Nr )r )r r r r r r s zTestGetDummies.sparsec C s |d krt jS |S )N)r r )r r r r r �effective_dtype& s z TestGetDummies.effective_dtypec C s. d}t jt|d�� t|dd� W d Q R X d S )Nz1dtype=object is not a valid dtype for get_dummies)�match�object)r )�pytest�raises� ValueErrorr )r r �msgr r r �'test_get_dummies_raises_on_dtype_object+ s z6TestGetDummies.test_get_dummies_raises_on_dtype_objectc C s� t d�}t|�}t|t d��}tdddgdddgdddgd�| �|�d�}|r\|jtdd�}t|||d �}t�||� t|||d �}t�||� t d�|_ t|||d �}t�||� d S ) N�abc�ABCr r )r r r )r g )� fill_value)r r ) �listr r r �applyr r �tm�assert_frame_equal�index)r r r �s_list�s_series�s_series_index�expected�resultr r r �test_get_dummies_basic0 s z%TestGetDummies.test_get_dummies_basicc C s� t d�}t|�}tdddddgdddddgdddddgd ��}tdddgdddgdddgd �| �|�t d�d �}|r�t|�r�d}n|tkr�d }nd }|jt|d �}t|||d�}t � ||� t|||d�}t � ||� t||j ||d�}|�r d| �|�j � d|� d�} n | �|�j } t| di�}|j �� }dd� |jD �|_t �||� t|dg||d�}ddd�} d| �| d� | | < t| ��� }|j �� }dd� |jD �|_|�� }t �||� d S )Nr( r r r r r r r )r r r )r �columnsFg )r* )r r )r6 r r zSparse[z, �]� c S s g | ] }t |��qS r )�str)�.0�ir r r � <listcomp>i s z?TestGetDummies.test_get_dummies_basic_types.<locals>.<listcomp>r )�int64r" c S s g | ] }t |��qS r )r9 )r: r; r r r r< s s )r+ r r r r �boolr, r r r- r. r6 �nameZdtypesZ value_countsr/ Zassert_series_equal�get� sort_index) r r r r0 r1 Zs_dfr3 r* r4 Z dtype_nameZexpected_countsr r r � test_get_dummies_basic_typesE sF , z+TestGetDummies.test_get_dummies_basic_typesc C s� t jg}t|�}t|dgd�}t||d�}t||d�}t||d�}|jsLt�|jsVt�|js`t�|j�� dgkstt�|j�� dgks�t�|j�� dgks�t�d S )Nr )r/ )r r )r �nanr r �empty�AssertionErrorr/ �tolist)r r Z just_na_listZjust_na_seriesZjust_na_series_indexZres_listZ res_seriesZres_series_indexr r r �test_get_dummies_just_naw s z'TestGetDummies.test_get_dummies_just_nac C s$ ddt jg}t|||d�}tdddgdddgd�| �|�d�}|rP|jtdd �}t�||� t|d ||d �}tt jdddgddddgddddgi| �|�d�}|j ddt jgdd �}|j |_ |r�|jtdd �}t�||� tt jgd ||d �}tt ddgd �t jg| �|�d�} t� |j | j � d S )Nr r )r r r r )r r )r g )r* T)�dummy_nar r )�axis)r/ )r6 r )r rC r r r r, r r- r. �reindexr6 r Zassert_numpy_array_equal�values) r r r �s�res�exp�res_na�exp_na� res_just_na� exp_just_nar r r �test_get_dummies_include_na� s( " " z*TestGetDummies.test_get_dummies_include_nac C sx dd l }d}|�d�}|||g}t|d|d�}tddddgd|� �dddgitjd �}|rh|jtdd �}t� ||� d S ) Nr �ez LATIN SMALL LETTER E WITH ACUTE�letter)�prefixr Zletter_er Zletter_)r )r* ) � unicodedata�lookupr r r r r, r r- r. )r r rW rT �eacuterL rM rN r r r �test_get_dummies_unicode� s &z'TestGetDummies.test_get_dummies_unicodec C s� |ddg }t ||d�}tdddgdddgdddgdddgd�tjd�}|r�ttdddgdd�tdddgdd�tdddgdd�tdddgdd�d��}t�||� d S ) Nr r )r r r )�A_a�A_b�B_b�B_c)r r )r r r r r r- r. )r r r r4 r3 r r r � test_dataframe_dummies_all_obj� s $ z-TestGetDummies.test_dataframe_dummies_all_objc C s� t |||d�}|r"t}t|d�}n tj}|}tdddg|dddg|d�|dddg|d�|dddg|d�|dddg|d�d��}|dd d d d g }t�||� d S ) N)r r r r r r )r )r r[ r\ r] r^ r r[ r\ r] r^ )r r r r �arrayr r- r. )r r r r r4 �arr�typr3 r r r �"test_dataframe_dummies_mix_default� s z1TestGetDummies.test_dataframe_dummies_mix_defaultc s� ddg}t |||d�}tdddgdddgdddgdddgdddgd�tjd �}|d g |d g< d d d dg}|d g| }|r�tnt� || �� fdd��||< t�||� d S )N�from_A�from_B)rV r r r r r )r �from_A_a�from_A_b�from_B_b�from_B_c)r r rf rg rh ri c s � | �S )Nr )�x)rb r r �<lambda>� � zCTestGetDummies.test_dataframe_dummies_prefix_list.<locals>.<lambda>) r r r r r r r, r- r. )r r r �prefixesr4 r3 �colsr )rb r �"test_dataframe_dummies_prefix_list� s z1TestGetDummies.test_dataframe_dummies_prefix_listc C s� t |d|d�}ddddg}tdddddgdddddgd ddddggd g| tjd �}|�d tji�}|r�tjtddd gd d �tdddgdd d�tdddgdd d�tdddgdd d�tdddgdd d�gdd�}t � ||� d S )N�bad)rV r Zbad_aZbad_bZbad_cr r r r r )r6 r )r? z Sparse[uint8])r? r )rI ) r r r r �astyper= �pd�concatr r- r. )r r r r4 Z bad_columnsr3 r r r �!test_dataframe_dummies_prefix_str� s & z0TestGetDummies.test_dataframe_dummies_prefix_strc C s� t |dgdg|d�}tdddgdddgdd dgd dd gd ��}|j}||dd � �tj�||dd � < |d g |d g< |r�d d g}|| �tdd ��||< t�||� d S )Nrd r )rV r6 r r r r r r r )r r rf rg r rf rg r ) r r r6 rq r r r r- r. )r r r r4 r3 rn r r r � test_dataframe_dummies_subset s $z,TestGetDummies.test_dataframe_dummies_subsetc C s� t |d|d�}tdddgdddgdddgdddgdddgd�tjd�}|d g |d g< |d d d d d g }|r�d d d d g}|| �tdd��||< t�||� t |ddg|d�}|jddd�d�}t�||� t |ddd�|d�}t�||� d S )Nz..)� prefix_sepr r r r r )r zA..azA..bzB..bzB..c)r r zA..azA..bzB..bzB..cr �__ZB__bZB__c)zB..bzB..c)r6 )r r ) r r r r rq r r- r. �rename)r r r r4 r3 rn r r r �!test_dataframe_dummies_prefix_sep s& z0TestGetDummies.test_dataframe_dummies_prefix_sepc C s8 t �d�}tjt|d�� t|dg|d� W d Q R X d S )NzPLength of 'prefix' (1) did not match the length of the columns being encoded (2))r! ztoo few)rV r )�re�escaper# r$ r% r )r r r r& r r r �(test_dataframe_dummies_prefix_bad_length7 s z7TestGetDummies.test_dataframe_dummies_prefix_bad_lengthc C s8 t �d�}tjt|d�� t|dg|d� W d Q R X d S )NzTLength of 'prefix_sep' (1) did not match the length of the columns being encoded (2))r! rp )rv r )rz r{ r# r$ r% r )r r r r& r r r �,test_dataframe_dummies_prefix_sep_bad_length? s z;TestGetDummies.test_dataframe_dummies_prefix_sep_bad_lengthc C s� ddd�}t dddgdddgddd gd ��}t|||d �}t dddgdd dgd dd gddd gd d dgd ��}ddddg}|| �tj�||< |r�|| �tdd ��||< t�||� d S )Nrd re )r r r r r r r r )r r r )rV r r )r rf rg rh ri rf rg rh ri r )r r rq r r r r- r. )r r rm r r4 r3 r6 r r r �"test_dataframe_dummies_prefix_dictG s " z1TestGetDummies.test_dataframe_dummies_prefix_dictc C s t jt jt jg|jdd d �f< t|d||d�jdd�}|rJt}t|d�}n t j}|}tdddt jg|ddddg|d�|ddddg|d�|ddddg|d�|ddddg|d�|ddddg|d�|ddddg|d�d ��jdd�}t � ||� t|d ||d�}|d d d ddg }t � ||� d S )Nr T)rH r r r )rI r r )r )r r[ r\ �A_nanr] r^ �B_nanFr r[ r\ r] r^ ) r rC �locr rA r r r` r r- r. )r r r r r4 ra rb r3 r r r � test_dataframe_dummies_with_na] s* z-TestGetDummies.test_dataframe_dummies_with_nac C s� t dddg�|d< t|||d�jdd�}|r<t}t|d�}n tj}|}tddd g|dddg|d �|dddg|d �|dddg|d �|dddg|d �|dddg|d �|dddg|d �d ��jdd�}t� ||� d S ) Nrj �y�cat)r r r )rI r r r )r )r r[ r\ r] r^ Zcat_x�cat_y) r r rA r r r r` r r- r. )r r r r r4 ra rb r3 r r r �'test_dataframe_dummies_with_categorical| s" z6TestGetDummies.test_dataframe_dummies_with_categoricalzget_dummies_kwargs,expected�data� är u ä_ar )r rj u x_ä)r� rV )r� rv u xäac C s t f |�}t�||� d S )N)r r- r. )r Zget_dummies_kwargsr3 r4 r r r � test_dataframe_dummies_unicode� s z-TestGetDummies.test_dataframe_dummies_unicodec C s� t d�}t|�}t|t d��}tdddgdddgd�tjd�}t|d|d�}|r^|jtdd �}t� ||� t|d|d�}t� ||� t d�|_ t|d|d�}t� ||� d S ) Nr( r) r r )r r )r T)� drop_firstr )r* ) r+ r r r r r r, r r- r. r/ )r r r0 r1 r2 r3 r4 r r r �!test_get_dummies_basic_drop_first� s z0TestGetDummies.test_get_dummies_basic_drop_firstc C s� t d�}t|�}t|t d��}tt�d�d�}t|d|d�}t�||� t|d|d�}t�||� tt d�d�}t|d|d�}t�||� d S )NZaaar) r )r/ T)r� r )r+ r r r �aranger r- r. )r r r0 r1 r2 r3 r4 r r r �+test_get_dummies_basic_drop_first_one_level� s z:TestGetDummies.test_get_dummies_basic_drop_first_one_levelc C s� ddt jg}t|d|d�}tddddgit jd�}|rD|jtdd�}t�||� t|dd|d �}tddddgt jdddgit jd�j dt jgdd �}|r�|jtdd�}t�||� tt jgdd|d �}tt � d�d �}t�||� d S ) Nr r T)r� r r r )r )r* )rH r� r )rI )r/ ) r rC r r r r, r r- r. rJ r� ) r r Zs_NArM rN rO rP rQ rR r r r �$test_get_dummies_basic_drop_first_NA� s $ z3TestGetDummies.test_get_dummies_basic_drop_first_NAc C s\ |ddg }t |d|d�}tdddgdddgd�tjd�}|rL|jtdd �}t�||� d S ) Nr r T)r� r r r )r\ r^ )r )r* )r r r r r, r r- r. )r r r r4 r3 r r r �!test_dataframe_dummies_drop_first� s z0TestGetDummies.test_dataframe_dummies_drop_firstc C s� t dddg�|d< t|d|d�}tdddgd dd gd d dgd ddgd ��}d d d g}|| �tj�||< |dd d d g }|r�x |D ]}t|| �||< q�W t�||� d S )Nrj r� r� T)r� r r r r r )r r\ r^ r� r\ r^ r� r ) r r r rq r r r r- r. )r r r r r4 r3 rn �colr r r �2test_dataframe_dummies_drop_first_with_categorical� s ( zATestGetDummies.test_dataframe_dummies_drop_first_with_categoricalc C s� t jt jt jg|jdd d �f< t|dd|d�jdd�}tdddt jgddddgddddgddddgddddgd��}d d d d g}|| �t j�||< |jdd�}|r�x |D ]}t|| �||< q�W t � ||� t|d d|d�}|dd d g }t � ||� d S )Nr T)rH r� r r )rI r r )r r\ r r^ r� r\ r r^ r� Fr ) r rC r� r rA r rq r r r- r. )r r r r4 r3 rn r� r r r �)test_dataframe_dummies_drop_first_with_na s( z8TestGetDummies.test_dataframe_dummies_drop_first_with_nac C s� t dddg�}t|�}tddgddgddggddgtjd�}t�||� t tdddg��}t|�}tddgddgddggtddg�tjd�}t�||� d S )Nr r r )r6 r r r )r r r r r r- r. r )r r� r4 r3 r r r �test_get_dummies_int_int s & (z'TestGetDummies.test_get_dummies_int_intc C s� t dddgtdddg�dddgdddgd��}dd d d d d g}t ddddddgddddddgddddddgg|d�}||dd � �|�||dd � < t|ddg|d�}t�||� d S )Nr r r r g �?g @)r r r �Dr r� �A_1�A_2ZB_ar] r )r6 r r )r6 r )r r rq r r- r. )r r r� r6 r3 r4 r r r �test_get_dummies_int_df, s ,"z&TestGetDummies.test_get_dummies_int_df�orderedTFc C s| t td�td�|d�}t||d�}tjdddgdddgg| �|�d�}t|j|j|d�}t||| �|�d�}t � ||� d S )NZxyZxyz)� categoriesr� )r r r )r6 r ) r r+ r r r` r r r� r r- r. )r r r� r� r4 r� rn r3 r r r �1test_dataframe_dummies_preserve_categorical_dtype> s $z@TestGetDummies.test_dataframe_dummies_preserve_categorical_dtypec C sL t �ddgddgd��}t|dg|d�}|jdgd �}t�|dg |� d S ) Nr r ZABZCD)�GDP�Nationr� )r6 r r� )r6 )r � from_dictr rJ r- r. )r r r Zdf2r r r �*test_get_dummies_dont_sparsify_all_columnsL s z9TestGetDummies.test_get_dummies_dont_sparsify_all_columnsc C s� dddg|_ t|�jdd�}tdddddgdddddgdddddggddddd gtjd �jdd�}|�dtji�}t� ||� d S ) Nr r )rI r r r r[ r\ ZA_c)r6 r ) r6 r rA r r r rq r= r- r. )r r r4 r3 r r r �"test_get_dummies_duplicate_columnsU s & z1TestGetDummies.test_get_dummies_duplicate_columnsc C s` t dddgi�}t|dgdd�}tdd�}t tddg|d�tddg|d�d ��}t�||� d S ) Nr r r T)r6 r r r )r )r� r� )r r r r r- r. )r r r4 r r3 r r r �test_get_dummies_all_sparsed s z*TestGetDummies.test_get_dummies_all_sparserK �bazc C sp t ddddddgddddddgd d d d d d gd d ddddgd��}d}tjt|d�� t||d� W d Q R X d S )Nr r r � � � �one�twor r r rj r� �z�q�w�t)�barZfoor� Zzooz1Input must be a list-like for parameter `columns`)r! )r6 )r r# r$ � TypeErrorr )r rK r r& r r r �#test_get_dummies_with_string_valuesp s z2TestGetDummies.test_get_dummies_with_string_values)/�__name__� __module__� __qualname__r# Zfixturer r Zfloat64r> r r r r' r5 rB rG rS rZ r_ rc ro rt ru ry r| r} r~ r� r� �markZ parametrizer r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r sX 2 " r )rz �numpyr r# Zpandas.core.dtypes.commonr Zpandasrr r r r r r Zpandas._testingZ_testingr- Zpandas.core.arrays.sparser r r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка