3
���hl* � @ s d dl Zd dlZd dlZd dlmZmZmZmZ d dl j
Z d dl m
Z
d dlmZmZ ejedddgd dd �edd
d gd d d �ed
ddgd dd �edddgd �edddgd �eejddd��eejdddd��eejddd��gd �d d � �Zejddgd �d d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zejjd)ej dd*d+g�ej dd*d+ge d �g�d,d-� �Z ejjd.d/d0ej d�g�d1d2� �Z d3d4� Z!ejjd.d/d0ej d�g�d5d6� �Z"d7d8� Z#d9d:� Z$ejjd;edddg�ed<d=d>g�ed<d=ej%g�ed<d=d>gd=d?d@gdA�ed
ddg�ed
ej%dg�ed
ddg�edddg�ee d �edddg�edddg�ed
ddgdddgdB��e� e j&� e j'� e j(� e j)� e j*� e j+� ee j+� �eejdddd��ej,e-d/�d0dCdDgejddd�g�ej,ej.e/dE��e-d�g�g�dFdG� �Z0dHdI� Z1ejjd;eg dJd �eg dKd �eg �g�dLdM� �Z2ejjdNed
dddg�edOdPdQdRg�eejd d*d��g�ejjdSddg�dTdU� ��Z3dVdW� Z4ejjd;ej d�g�dXdY� �Z5dZd[� Z6d\d]� Z7d^d_� Z8d`da� Z9ejjdbe-dc��ejjddd dg�dedf� ��Z:dgdh� Z;didj� Z<dkdl� Z=dS )m� N)� DataFrame�Index�
MultiIndex�Series)� hash_tuples)�
hash_array�hash_pandas_object� � � Zint32)�dtypeg @g @Zfloat32�a�b�c�category�d�e�fTFZ20130101� )�periodsz
US/Eastern)r �tz�2000)�paramsc C s | j S )N)�param)�request� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/util/test_hashing.py�series s
r c C s | j S )N)r )r r r r �index s r c K s( t | f|�}t | f|�}tj||� dS )z�
Check that hashing an objects produces the same value each time.
Parameters
----------
obj : object
The object to hash.
kwargs : kwargs
Keyword arguments to pass to the hashing function.
N)r �tm�assert_series_equal)�obj�kwargsr
r r r r � _check_equal s r# c C s@ t | t�s<t| dd�}t| dd�}t| �r<||kj� s<t�dS )z�
Check the hash of an object with and without its index is not the same.
Parameters
----------
obj : object
The object to hash.
T)r FN)�
isinstancer r �len�all�AssertionError)r! r
r r r r �_check_not_equal_with_index0 s
r( c C sD t tdddg��} ttjdddgdd�dddgd �}tj| |� d S )
N�foo�bar�bazl �7�* � l 6jz�d� l n �*5�uint64)r )r )r r r �np�arrayr r )�result�expectedr r r �test_consistencyA s r1 c C s | j }tjt|�t|�� d S )N)�valuesr �assert_numpy_array_equalr )r Zarrr r r �test_hash_arrayO s r4 �arr2� �Allc C s, t tjdddg��}t | �}tj||� d S )N�3�4r7 )r r- r. r r3 )r5 �result1�result2r r r �test_hash_array_mixedT s r< �val� r) c C s* d}t jt|d�� t| � W d Q R X d S )Nzmust pass a ndarray-like)�match)�pytest�raises� TypeErrorr )r= �msgr r r �test_hash_array_errors^ s rD c C sN dddg} t | �}ttj| ��j}tj||� t | d �}||d ksJt�d S ) Nr �one�twor
r )r rE )r rF )r
rE )r r r � from_tuplesr2 r r3 r' )Ztuplesr/ r0 r r r �test_hash_tuplese s
rH c C s* d}t jt|d�� t| � W d Q R X d S )Nz'must be convertible to a list-of-tuples)r? )r@ rA rB r )r= rC r r r �test_hash_tuples_errp s rI c C s: t jdd d
d g�} | jdks t�t| �}|jdks6t�d S ) N�v �� �� �3 �� �f T)rJ rK )rL rJ )rM rN )rO rM )r rG Z is_uniquer' r )�mir/ r r r �test_multiindex_uniquew s rQ c C sj t dddgdddggddddgddddggdd gd
�} | j� }| j|�sLt�t| j�jt|j��sft�d S ) Nr r r
r r
r r Zcol1Zcol2)Zlevels�codes�names)r Z_sort_levels_monotonic�equalsr' r r2 )rP Zreconsr r r �test_multiindex_objects s rU r! g �?g �?g������ @g�������?gffffff
@)r )�x�yr* r+ Zaabcc C s t | |d� t| � d S )N)r )r# r( )r! r r r r �test_hash_pandas_object� s rX c C s t | |d� t| � d S )N)r )r# r( )r r r r r �test_hash_pandas_object2� s rY Zfloat64�objectc C s t | |d� d S )N)r )r# )r! r r r r � test_hash_pandas_empty_object� s r[ �s1i� i� i� i� �
categorizec C sf | j d�jj| �}|jjtt| ���}t| |d�}t||d�}t||d�}tj||� tj||� d S )Nr )r] )Zastype�catZset_categories�list�reversedr r r )r\ r] �s2Zs3�h1�h2�h3r r r � test_categorical_consistency� s re c C s� t jjd
dddddgt jdddd �d
�} t| d d �}t jjddgt jd�gd
�} t| d d �}|d |kslt�|d |ks|t�d S )Nr r r
r r6 z
2012-01-01r> �B)r �name)�
categoriesF)r] ���ri )�pdZ CategoricalZ
from_codes�
date_ranger � Timestampr' )r r0 r/ r r r �%test_categorical_with_nan_consistency� s " rm c C s* d}t jt|d�� t| � W d Q R X d S )NzUnexpected type for hashing)r? )r@ rA rB r )r! rC r r r �test_pandas_errors� s rn c C s8 t td��} t| dd�}t| dd�}||kj� s4t�d S )N�abcZ9876543210123456)�hash_keyZ9876543210123465)r r_ r r& r' )r! r
r r r r �test_hash_keys� s rq c C s6 d} t jt| d�� tttd��dd� W d Q R X d S )Nz&key should be a 16-byte string encoded)r? ro r) )rp )r@ rA �
ValueErrorr r r_ )rC r r r �test_invalid_key� s rs c C s$ t td��jjd�}t|| d� d S )Nro �utf8)r )r r_ �str�encoder# )r r! r r r �test_already_encoded s rw c C s t td��}t|| dd� d S )Nro �ascii)r �encoding)r r_ r# )r r! r r r �test_alternate_encoding s rz �l_exp� �l_addc C s@ d| d | }t j|d�}t|d�}|d |d k s<t�d S )Nr
r| rt r r )r Z rands_arrayr r' )r{ r} �length�sr/ r r r � test_same_len_hash_collisions s
r� c C s� ddg} t tj| dd� td�d�}tjdgtjd�}tj||� t tj| dd� td�d�}tjd gtjd�}tj||� t tj| td�d�}tj|tj||gdd
�� d S ) Na Ingrid-9Z9fKIZmkO7i7Cn51Li34pJm44fgX6DYGBNj3VPlOH50m7HnBlPxfIwFMrcNJNMP6PSgLmwWnInciMWrCSAlLEvt7JkJl4IxiMrVbXSa8ZQoVaq5xoQPjltuJEfwdNlO6jo8qRRHvD8sBEBMQASrRa6TsdaPTPCBo3nwIBpE7YzzmyH0vMBhjQZLx1aCT7faSEx7PgFxQhHdKFWROcysamgy9iVj8DO2Fmwg1NNl93rIAqC3mdqfrCxrzfvIY8aJdzin2cHVzy3QUJxZgHvtUtOLxoqnUHsYbNTeq0xcLXpTZEZCxD4PGubIuCNf32c33M7HFsnjWSEjE2yVdWKhmSVodyF8hFYVmhYnMCztQnJrt3O8ZvVRXd5IKwlLexiSp4h888w7SzAIcKgc3g5XQJf6MlSMftDXm9lIsE1mJNiJEv6uY6pgvC3fUPhatlR5JPpVAHNSbSEE73MBzJrhCAbOLXQumyOXigZuPoME7QgJcBalliQol7YZ9a Tim-b9MddTxOWW2AT1Py6vtVbZwGAmYCjbp89p8mxsiFoVX4FyDOF3wFiAkyQTUgwg9sVqVYOZo09Dh1AzhFHbgij52ylF0SEwgzjzHH8TGY8Lypart4p4onnDoDvVMBa0kdthVGKl6K0BDVGzyOXPXKpmnMF1H6rJzqHJ0HywfwS4XYpVwlAkoeNsiicHkJUFdUAhG229INzvIAiJuAHeJDUoyO4DCBqtoZ5TDend6TK7Y914yHlfH3g1WZu5LksKv68VQHJriWFYusW5e6ZZ6dKaMjTwEGuRgdT66iU5nqWTHRH8WSzpXoCFwGcTOwyuqPSe0fTe21DVtJn1FKj9F9nEnR9xOvJUO7E0piCIF4Ad9yAIDY4DBimpsTfKXCu1vdHpKYerzbndfuFe5AhfMduLYZJi5iAw8qKSwR5h86ttXV0Mc0QmXz8dsRvDgxjXSmupPxBggdlqUlC828hXiTPD7am0yETBV0F3bEtvPiNJfremszcV8NcqAoARMer r )r rt l ��rIU} r
l pX�^�� )Zaxis) r r- ZasarrayrZ r. r, r r3 Z concatenate)�hashesr: Z expected1r; Z expected2r/ r r r �test_hash_collisions s r� c C s� t jdtd�td�gi�} t| �}t jddgtjd�}tj||� t jdtdg�tdg�gi�}t|�}t jd d
gtjd�}tj||� t jdtdg g�tdi g�gi�}t j
t d d �� t|� W d Q R X d S )
N�data�1�2l �.C7�j�| l 93�B
� )r r r
l pH-\h+� l �1�"�R�k zunhashable type: 'list')r? ) rj r �tupler r r- r, r r r@ rA rB )Zdfr/ r0 Zdf2Zdf3r r r �test_hash_with_tuple+ s "r� c C s< t jjt jddg�d d�} t jddgdd�}tj| |� d S )Nr
r )rp l A�!� M| l �Z4^� � r, )r )rj �utilr r r r )r/ r0 r r r �test_hash_object_none_key@ s r� )>Znumpyr- r@ Zpandasrj r r r r Zpandas._testingZ_testingr Zpandas.core.util.hashingr Z pandas.utilr r Zfixturerk Ztimedelta_ranger r r# r( r1 r4 �markZ parametrizer. rZ r< rl rD rH rI rQ rU �nanZmakeMissingDataframeZmakeMixedDataFrameZmakeTimeDataFrameZmakeTimeSeriesZmakeTimedeltaIndexZmakePeriodIndexZ from_product�rangeZCategoricalIndexr_ rX rY r[ re rm rn rq rs rw rz r� r� r� r� r r r r �<module> s�
0 " "
(*
|