Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/libs/__pycache__/test_hashtable.cpython-37.pyc
Назад
B �5�g3H � @ sx d dl mZ d dlZd dlZd dlZd dlmZ d dl Z d dl m Z d dlmZ edd� �Zdd� Zej�d ejejfejejfejejfejejfej ej fej ej fej ej!fej"ej#fej$ej%fej&ej'fej(ej)fej*ej+fej,ej-fg �G d d � d ��Z.G d d � d �Z/dd� Z0dd� Z1dd� Z2dd� Z3dd� Z4ej�d ej ej fej$ej%fejejfej ej fg�G dd� d��Z5dd� Z6d d � Z7d d � Z8d d!� Z9ej�d"ejd#fejd$fejd%fejd&fej d'fej d(fej!d)fej#d*fej%d+fej'd,fej)d-fej+d.fej-d/fg �G d0d1� d1��Z:d2d3� Z;ej�d"ej d'fej%d+fejd$fej d(fg�G d4d5� d5��Z<d6d7� Z=d8d9� Z>dS ):� )�contextmanagerN)� hashtable)�isinc c s" t �� z d V W d t �� X d S )N)� tracemalloc�start�stop� r r �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/libs/test_hashtable.py�activated_tracemalloc s r c C s4 t �� } | �t �dt�� �f�} ttdd� | j��S )NTc S s | j S )N)�size)�xr r r �<lambda> � z,get_allocated_khash_memory.<locals>.<lambda>) r Z take_snapshotZ filter_tracesZ DomainFilter�htZget_hashtable_trace_domain�sum�mapZtraces)Zsnapshotr r r �get_allocated_khash_memory s r ztable_type, dtypec @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� TestHashTablec C sp d}|d�}t |�dks t�||ks(t�|�|d� t |�dksDt�||ksPt�|�|�dksbt�|�|d d� ||ks~t�|d |ks�t�t |�dks�t�|�|�dks�t�|�|d �dks�t�|�|d� ||ks�t�|d |ks�t�t |�dks�t�|�|�dk�st�|�|d �dk�s*t�|d |k�s<t�tjtt|d �d �� |�|d � W d Q R X d S ) N� �7 r �* � �) � � )�match)�len�AssertionError�set_item�get_item�pytest�raises�KeyError�str)�self� table_type�dtype�index�tabler r r �test_get_set_contains_len2 s. z'TestHashTable.test_get_set_contains_lenc C s� |t jkr�d}|� }t�|��|�}t�|��tj�| }||j_||j_|�||� x*t |�D ] }|� || �|| ks`t �q`W d S )N�M ) r �PyObjectHashTable�np�arange�astype�int64�flags� writeabler �ranger r ) r$ r% r& �writable�Nr( �keys�vals�ir r r �test_mapO s zTestHashTable.test_mapc C s\ d}|� }t �|�| �|�}||j_|�|� x&t|�D ]}|�|| �|ks:t�q:W d S )N� ) r, r- r. r0 r1 � map_locationsr2 r r )r$ r% r& r3 r4 r( r5 r7 r r r �test_map_locations\ s z TestHashTable.test_map_locationsc C sd d}|� }t �|�| �|�}||j_|�|� |�|�}t �|�}t�|�t j �|�t j �� d S )N� ) r, r- r. r0 r1 r: �lookup�tm�assert_numpy_array_equalr/ ) r$ r% r& r3 r4 r( r5 �result�expectedr r r � test_lookupe s zTestHashTable.test_lookupc C sn |t jt jfkrd}nd}|� }t �|�| �|�}|�|� t �|��|�}|�|�}t �|dk�sjt�d S )N�d i ���) r, �int8�uint8r- r. r: r= �allr )r$ r% r& r4 r( r5 Z wrong_keysr@ r r r �test_lookup_wrongo s z TestHashTable.test_lookup_wrongc C sb |t jt jfkrd}nd}|� }t �|�| �|�}t �|d�}||j_|�|�}t � ||� d S )N�X i� r ) r, rE rF r- r. �repeatr0 r1 �uniquer> r? ) r$ r% r& r3 r4 r( rA r5 rK r r r � test_unique{ s zTestHashTable.test_uniquec C sz |t jt jfkrd}nd}t �|��|�}t� �@ |� }|�|� t� }|�� }||ks\t �~t� dkslt �W d Q R X d S )N� i0u r ) r, rE rF r- r. r r: r �sizeofr )r$ r% r& r4 r5 r( �used�my_sizer r r �test_tracemalloc_works� s z$TestHashTable.test_tracemalloc_worksc C sF t � �6 |� }t� }|�� }||ks(t�~t� dks8t�W d Q R X d S )Nr )r r rN r )r$ r% r& r( rO rP r r r �test_tracemalloc_for_empty� s z(TestHashTable.test_tracemalloc_for_emptyc C sL |d�}|� � }|d dks t�|d dks0t�d|ks<t�d|ksHt�d S )Ni� r r Z n_occupied� n_buckets� upper_bound)� get_stater )r$ r% r& r( �stater r r �test_get_state� s z TestHashTable.test_get_statec C s� x|t dd�D ]n}t�|��|�}||�}|�� d }|�|� |�� d }||ksVt�|� }|�|� ||�� d ks t�q W d S )Nr �n rS )r2 r, r- r. rU r: r ) r$ r% r& r4 r5 �preallocated_table�n_buckets_start� n_buckets_end� clean_tabler r r �test_no_reallocation� s z"TestHashTable.test_no_reallocationN) �__name__� __module__� __qualname__r) r8 r; rB rH rL rQ rR rW r] r r r r r s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� TestPyObjectHashTableWithNansc C sF t d�}t d�}||k s t�t�� }|�|d� |�|�dksBt�d S )N�nanr )�floatr r r+ r r )r$ �nan1�nan2r( r r r �test_nan_float� s z,TestPyObjectHashTableWithNans.test_nan_floatc C sZ t td�td��}t td�td��}||k s0t�t�� }|�|d� |�|�dksVt�d S )Nrb r )�complexrc r r r+ r r )r$ rd re r( r r r �test_nan_complex_both� s z3TestPyObjectHashTableWithNans.test_nan_complex_bothc C s� t td�d�}t td�d�}t td�d�}||k s6t�t�� }|�|d� |�|�dks\t�tjt d d��}|�|� W d Q R X t |j �t |�ks�t�d S )Nrb r r r )r ) rg rc r r r+ r r r r! r" r# �value)r$ rd re �otherr( �errorr r r �test_nan_complex_real� s z3TestPyObjectHashTableWithNans.test_nan_complex_realc C s� t dtd��}t dtd��}t dtd��}||k s6t�t�� }|�|d� |�|�dks\t�tjt d d��}|�|� W d Q R X t |j �t |�ks�t�d S )Nr rb r r )r ) rg rc r r r+ r r r r! r" r# ri )r$ rd re rj r( rk r r r �test_nan_complex_imag� s z3TestPyObjectHashTableWithNans.test_nan_complex_imagc C sR t d�f}t d�f}|d |d k s(t�t�� }|�|d� |�|�dksNt�d S )Nrb r r )rc r r r+ r r )r$ rd re r( r r r �test_nan_in_tuple� s z/TestPyObjectHashTableWithNans.test_nan_in_tuplec C s� ddt d�fff}ddt d�fff}d}t�� }|�|d� |�|�dksNt�tjtd d��}|�|� W d Q R X t |j �t |�ks�t�d S )Nr r rb )r r r )r ) rc r r+ r r r r r! r" r# ri )r$ rd re rj r( rk r r r �test_nan_in_nested_tuple� s z6TestPyObjectHashTableWithNans.test_nan_in_nested_tupleN) r^ r_ r` rf rh rl rm rn ro r r r r ra � s ra c C s\ t d�t d�t d�ff} t d�t d�t d�ff}t�| �t�|�ksHt�t�| |�sXt�d S )Nrb )rc r Z object_hashr Zobjects_are_equal)�a�br r r � test_hash_equal_tuple_with_nans� s rr c C s� t �� }tjddddddgtjd�}| |j_|�|�\}}tjddddddgtjd�}tjddgtjd�}t� |� tj�|� t� ||� d S )Nr r rD )r& r ) r �Int64HashTabler, �arrayr/ r0 r1 Zget_labels_groupbyr> r? r. )r3 r( r6 �arrrK Z expected_arrZexpected_uniquer r r �!test_get_labels_groupby_for_Int64� s rv c C sr d} t �| ��t jj��t j�}t� �B t�� }|� |� t � }|� � }||ksTt �~t � dksdt �W d Q R X d S )Ni� r ) r, r- r. �compat�unicode�object_r r �StringHashTabler: r rN r )r4 r5 r( rO rP r r r �*test_tracemalloc_works_for_StringHashTable s r{ c C sH t � �8 t�� } t� }| �� }||ks*t�~ t� dks:t�W d Q R X d S )Nr )r r rz r rN r )r( rO rP r r r �*test_tracemalloc_for_empty_StringHashTable s r| c C s� x�t dd�D ]~} t�| ��tjj��tj�}t�| �}|� � d }|� |� |� � d }||ksdt �t�� }|� |� ||� � d ks t �q W d S )Nr rX rS ) r2 r, r- r. rw rx ry r rz rU r: r )r4 r5 rY rZ r[ r\ r r r �$test_no_reallocation_StringHashTable s r} c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestHashTableWithNansc C s� t d�}|� }||kst�|�|d� t|�dks6t�||ksBt�|�|�dksTt�|�|d� t|�dkspt�||ks|t�|�|�dks�t�d S )Nrb r r r )rc r r r r )r$ r% r& r' r( r r r r) 5 s z/TestHashTableWithNans.test_get_set_contains_lenc C sn d}|� }t j|t j|d�}t �|�| �t j�}|�||� t|�dksNt�|� t j�d| d ksjt�d S )NiL )r& r r ) r, �fullrb r- r. r/ r r r r )r$ r% r& r4 r( r5 r6 r r r r8 D s z TestHashTableWithNans.test_mapc C sR d}|� }t j|t j|d�}|�|� t|�dks6t�|�t j�|d ksNt�d S )N� )r& r )r, r rb r: r r r )r$ r% r& r4 r( r5 r r r r; M s z(TestHashTableWithNans.test_map_locationsc C sJ d}|� }t j|t j|d�}|�|�}t �t �|��rBt|�dksFt�d S )Ni� )r& r )r, r rb rK rG �isnanr r )r$ r% r& r4 r( r5 rK r r r rL U s z!TestHashTableWithNans.test_uniqueN)r^ r_ r` r) r8 r; rL r r r r r~ + s r~ c C sD t �� } tjdd� td�D �tjd�}| �|�}t|�dks@t�d S )Nc S s g | ] }t d ��qS )rb )rc )�.0r7 r r r � <listcomp>_ s z6test_unique_for_nan_objects_floats.<locals>.<listcomp>�2 )r& r ) r r+ r, rt r2 ry rK r r )r( r5 rK r r r �"test_unique_for_nan_objects_floats] s r� c C sD t �� } tjdd� td�D �tjd�}| �|�}t|�dks@t�d S )Nc S s g | ]}t td �d��qS )rb g �?)rg rc )r� r7 r r r r� f s z7test_unique_for_nan_objects_complex.<locals>.<listcomp>r� )r& r ) r r+ r, rt r2 ry rK r r )r( r5 rK r r r �#test_unique_for_nan_objects_complexd s r� c C sJ t �� } tjdgdd� td�D � tjd�}| �|�}t|�dksFt�d S )Nr c S s g | ]}d t d�d ff�qS )g �?rb )rc )r� r7 r r r r� n s z5test_unique_for_nan_objects_tuple.<locals>.<listcomp>r� )r& r ) r r+ r, rt r2 ry rK r r )r( r5 rK r r r �!test_unique_for_nan_objects_tuplek s r� c C s t t| �S )N)�getattrr )Zfun_name� type_suffixr r r �get_ht_functiont s r� zdtype, type_suffix�object� complex128r/ �uint64�float64� complex64�int32�uint32�float32�int16�uint16rE rF c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestHelpFunctionsc C sl d}t d|�}t�|�| �|�}t�|d�}||j_||d�\}} t�t� |�|� t� | dk�sht �d S )N�+ � value_countr F) r� r, r- r. rJ r0 r1 r> r? �sortrG r ) r$ r& r� r3 r4 r� rA �valuesr5 �countsr r r �test_value_count� s z"TestHelpFunctions.test_value_countc C s` t d|�}t�dddddddg��|�}||j_||d �\}}t�||� t�|dk�s\t �d S ) Nr� r r r � r< rD r9 F) r� r, rt r. r0 r1 r> r? rG r )r$ r& r� r3 r� r� r5 r� r r r �test_value_count_stable� s z)TestHelpFunctions.test_value_count_stablec C sd d}t d|�}t�t�|��|�d�}||j_||�}tj|tjd�}d|d d d�< t � ||� d S )NrC � duplicatedr )r& F) r� r, rJ r- r. r0 r1 � ones_like�bool_r> r? ) r$ r& r� r3 r4 r� r� r@ rA r r r �test_duplicated_first� s z'TestHelpFunctions.test_duplicated_firstc C sh d}t d|�}t�|��|�}t�|��|�}||j_||j_|||�}tj|tjd�} t� || � d S )N� �ismember)r& ) r� r, r- r. r0 r1 r� r� r> r? ) r$ r& r� r3 r4 r� ru r� r@ rA r r r �test_ismember_yes� s z#TestHelpFunctions.test_ismember_yesc C s\ d}t d|�}t�|��|�}t�|�| �|�}|||�}tj|tjd�}t�||� d S )N� r� )r& )r� r, r- r. Z zeros_liker� r> r? ) r$ r& r� r4 r� ru r� r@ rA r r r �test_ismember_no� s z"TestHelpFunctions.test_ismember_noc C sf |t jt jfkrd}nd}td|�}t �t �|��|�d�}d|d< ||j_||d�}|dksbt �d S )N�5 ig+ �moder r r F) r, rE rF r� rJ r- r. r0 r1 r )r$ r& r� r3 r4 r� r� r@ r r r � test_mode� s zTestHelpFunctions.test_modec C sJ t d|�}t�dddddddg��|�}||j_||d �}t�||� d S ) Nr� r r r r� r< rD r9 F)r� r, rt r. r0 r1 r> r? )r$ r& r� r3 r� r� r5 r r r �test_mode_stable� s z"TestHelpFunctions.test_mode_stableN) r^ r_ r` r� r� r� r� r� r� r� r r r r r� x s r� c C sJ t jdtjt jgt jd�} t�| d�}|jdks4t �t � |d �sFt �d S )NT)r& Fr r ) r, rt �pdZNArb ry r r� r r r� )r� �modesr r r �test_modes_with_nans� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestHelpFunctionsWithNansc C s� t d|�}tjtjtjtjg|d�}||d�\}}t|�dksBt�||d�\}}t|�dkrlt�t�|��spt�|d dks�t�d S )Nr� )r& Tr Fr r< )r� r, rt rb r r rG r� )r$ r& r� r� r� r5 r� r r r r� � s z*TestHelpFunctionsWithNans.test_value_countc C sL t d|�}tjtjtjtjg|d�}||�}t�dddg�}t�||� d S )Nr� )r& FT)r� r, rt rb r> r? )r$ r& r� r� r� r@ rA r r r r� � s z/TestHelpFunctionsWithNans.test_duplicated_firstc C sj t d|�}tjtjtjtjg|d�}tjtjtjg|d�}|||�}tjdddgtjd�}t�||� d S )Nr� )r& T)r� r, rt rb r� r> r? )r$ r& r� r� ru r� r@ rA r r r r� � s z+TestHelpFunctionsWithNans.test_ismember_yesc C sd t d|�}tjtjtjtjg|d�}tjdg|d�}|||�}tjdddgtjd�}t�||� d S )Nr� )r& r F)r� r, rt rb r� r> r? )r$ r& r� r� ru r� r@ rA r r r r� � s z*TestHelpFunctionsWithNans.test_ismember_noc C sP t d|�}tjdtjtjtjg|d�}||d�dks8t�t�||d��sLt�d S )Nr� r )r& TF)r� r, rt rb r r� )r$ r& r� r� r� r r r r� s z#TestHelpFunctionsWithNans.test_modeN)r^ r_ r` r� r� r� r� r� r r r r r� � s r� c C sL dt d�fdg} dt d�fg}t| |�}tjddgtjd�}t�||� d S )Nrp rb )rq r TF)r& )rc r r, rt r� r> r? )r� �compsr@ rA r r r � test_ismember_tuple_with_nans s r� c C sJ ddddg} t td��}t| |�}tjddddgtjd�}t�||� d S ) Nrp r g @y @ � FT)r& )�listr2 r r, rt r� r> r? )r� r� r@ rA r r r �+test_float_complex_int_are_equal_as_objects s r� )?� contextlibr r �numpyr, r Z pandas._libsr r Zpandasr� Zpandas._testingZ_testingr> Zpandas.core.algorithmsr r r �markZ parametrizer+ ry ZComplex128HashTabler� rs r/ ZUInt64HashTabler� ZFloat64HashTabler� ZComplex64HashTabler� ZInt32HashTabler� ZUInt32HashTabler� ZFloat32HashTabler� ZInt16HashTabler� ZUInt16HashTabler� Z Int8HashTablerE ZUInt8HashTablerF r ra rr rv r{ r| r} r~ r� r� r� r� r� r� r� r� r� r r r r �<module> s� = , I +
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка