Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/__pycache__/test_algos.cpython-37.pyc
Назад
B �5�g�K � @ s d dl m Z d dlmZ d dlZd dlZd dlZd dlmZ m Z d dl m Z d dlm mZ d dlmZmZmZmZmZ d dlmZ d dlZ d dlm Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- d dl.m/Z0 d dl1m2 m3Z d d l4m5Z5 d dl6m2 m7Z8 G d d � d �Z9G d d � d �Z:G dd� d�Z;G dd� d�Z<G dd� d�Z=G dd� d�Z>dd� Z?dd� Z@G dd� d�ZAd d � ZBG d d � d �ZCd d!� ZDd"d#� ZEd$d%� ZFd&d'� ZGd(d)� ZHd*d+� ZIG d,d-� d-�ZJG d.d/� d/�ZKejL�Md0ejNe jNg�d1d2� �ZOdS )3� )�datetime)� permutationsN)�algos� hashtable)�np_array_datetime64_compat)� is_bool_dtype�is_complex_dtype�is_float_dtype�is_integer_dtype�is_object_dtype)�CategoricalDtype)� Categorical�CategoricalIndex� DataFrame� DatetimeIndex�Index� IntervalIndex� MultiIndex�NaT�Period� PeriodIndex�Series� Timedelta� Timestamp� date_range�timedelta_range� to_datetime� to_timedelta)� DatetimeArrayc @ s� e Zd Zej�dddg�dd� �Zdd� Zdd � Zd d � Z d d � Z dd� Z ej�ddddddgdddddgddddgfdddddgdddddgddddgfddddgddddgdddgfg�dd� �Z d d � Z d d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zej�dddg�d,d-� �Zej�dddg�d.d/� �Zd0d1� Zej�d2ejdddgd3d4�ejd5dd5gd6d4�ejd7d8d7gd9d4�g�d:d;� �Zej�d<ejddddgd3d4�dfejddddgd3d4�dfejd5dd5dgd6d4�d5fejdd5ddgd6d4�dfejd=d>d=d?ged4�d=fejd@dAd@dBged4�d@fejdAd@dAdBged4�dAfg�dCdD� �Z ej�dddg�ej�dEdFdGdHg�ejjdIejd?d=dJd?ged4�ejd?d=ged4�fe jddej dgdKd4�e jddgdKd4�fgdLdMgdN�dOdP� ���Z ej�dd=dJd?d=gejddddge� dQ�d4�ejd=d?ej ged4�fd=ej d?d=gejddddge� dQ�d4�ejd=d?ej ged4�fg�dRdS� �Z!ej�dddJddgejddddge� dQ�d4�ejddej gdTd4�fdej ddgejddddge� dQ�d4�ejddej gej"d4�fg�dUdV� �Z#dJS )W� TestFactorize�sortTFc s� |}|j |d�\}}t}t|t�r(tj}||�� �}|r@|�� }t|�� � fdd�|D �}tj |tj d�}t � ||� t � ||� d S )N)r c s g | ]}� � |��qS � )�index)�.0�val)�expected_uniques_listr! �M/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/test_algos.py� <listcomp>C s z0TestFactorize.test_factorize.<locals>.<listcomp>)�dtype)� factorizer � isinstancer � from_tuples�uniqueZ sort_values�list�np�asarray�intp�tm�assert_numpy_array_equal�assert_index_equal) �selfZindex_or_series_objr �objZ result_codesZresult_uniques� constructor�expected_uniques�expected_codesr! )r% r& �test_factorize3 s z TestFactorize.test_factorizec C sp t �dddt jg�}t|�}|jd d�\}}t jddddgt jd�}tddt jg�}t�||� t� ||� d S )N� � )� na_sentinelr )r( g �?g @) r. �array�nanr r) r0 r r1 r2 r3 )r4 �values�ser�codes�uniquesr8 r7 r! r! r& �&test_series_factorize_na_sentinel_noneI s z4TestFactorize.test_series_factorize_na_sentinel_nonec C sJ t �ddddddddg�\}}t�|tjdddgtd�� t jddddddddgdd�\}}tjdddddd d d gtjd�}t�||� tjdddgtd�}t�||� tjd tjd�d d d � }t �|�\}}tjddd d d gtjd�}t�||� tjd d d ddg|j d�}t�||� t j|dd�\}}tjd d d ddgtjd�}t�||� tjddd d d g|j d�}t�||� t�d�d d d � }t �|�\}}tjddd d d gtjd�}t�||� tjdddddg|j d�}t�||� t j|dd�\}}tjd d d ddgtjd�}t�||� tjdddddg|j d�}t�||� d S )N�a�b�c)r( T)r r r: r; � ���� � g @g @g @g @g �?g ) r r) r1 r2 r. r= �objectr0 �aranger( )r4 rA rB �exp�arrr! r! r& � test_basicU s< zTestFactorize.test_basicc C s� t ddtjddtjg�}t�|�\}}tjddddddgtjd �}t� ||� t dddtjg�}t� ||� tj|d d �\}}tjddddddgtjd �}t� ||� t dtjddg�}t� ||� d S ) N�A�Bg ��Q� @r rH r: r; rI )r( T)r ) r r. r> �infr r) r= r0 r1 r2 r r3 )r4 �xrA rB rM r! r! r& � test_mixed~ s zTestFactorize.test_mixedc C sN t d�}t d�}t||||||g�}t�|�\}}tjddddddgtjd�}t�||� t ||g�}t� ||� tj|dd�\}}tjddddddgtjd�}t�||� t ||g�}t� ||� t dd d �}t d d d �}t||||||g�}t�|�\}}tjddddddgtjd�}t�||� t� |t ||g�� tj|dd�\}}tjddddddgtjd�}t�||� t� |t ||g�� t d �}t d �}t|||||||g�}t�|�\}}tjdddddddgtjd�}t�||� t� |t ||g�� tj|dd�\}}tjdddddddgtjd�}t�||� t� |t ||g�� d S )Nz20130101 09:00:00.00004�20130101r r: )r( T)r Z201302�M)�freqZ201303z 1 day 1 minz1 day)r r r r) r. r= r0 r1 r2 r r3 r r r )r4 Zv1Zv2rS rA rB rM r! r! r& � test_datelike� sF zTestFactorize.test_datelikec C s t jdddt jgdd�}t�t|��}xddD ]\}|j|d|d�}t jddd|gd d�}tt|��tt|��ksrt�t � t � |�||k� q,W t jdt jdgdd�}d }|j|d |d�}t jdd dgd d�}tt|��tt|��ks�t�t � t � |�||k� d S ) Nr: r; �O)r( )rH � T)r r< r �int32rH F) r. r= r> �htZObjectFactorizer�lenr) �set�AssertionErrorr1 r2 �pd�isna)r4 �keyZrizerr< �ids�expectedr! r! r& �test_factorize_nan� s z TestFactorize.test_factorize_nanz&data, expected_codes, expected_uniques)r: r: )r: r; )r r Znonsenser r: r; rI )r: r; rI c C sD t �|�\}}t�|tj|tjd�� tj|t d�}t�||� d S )N)r( ) r` r) r1 r2 r. r= r0 �comZasarray_tuplesaferK )r4 �datar8 r7 rA rB Zexpected_uniques_arrayr! r! r& �test_factorize_tuple_list� s z'TestFactorize.test_factorize_tuple_listc C sV t jdd� td�D �td�}d}tjt|d�� tj|d d d� dd � W d Q R X d S ) Nc S s g | ] }t |��qS r! )�complex)r# �ir! r! r& r' � s z6TestFactorize.test_complex_sorting.<locals>.<listcomp>� )r( zJunorderable types: .* [<>] .*|'[<>]' not supported between instances of .*)�matchrH T)r ) r. r= �rangerK �pytest�raises� TypeErrorr r) )r4 Zx17�msgr! r! r& �test_complex_sorting� s z"TestFactorize.test_complex_sortingc C sn |}t jddddg|d�}t jddddgt jd�}t jddg|d�}t�|�\}}t�||� t�||� d S )Nr: r; )r( r )r. r= r0 r r) r1 r2 )r4 Zany_real_dtyper( rg r8 r7 rA rB r! r! r& � test_numeric_dtype_factorize� s z*TestFactorize.test_numeric_dtype_factorizec C s� t jddddddgt jd�}|j|d� t jddddddgt jd�}t jdddgt jd�}t�|�\}}t�||� t�||� d S ) Ng �?g �חAg:�0�yE>)r( )�writer r: r; ) r. r= �float64�setflagsr0 r r) r1 r2 )r4 �writablerg r8 r7 rA rB r! r! r& �test_float64_factorize s z$TestFactorize.test_float64_factorizec C sv t jdddgt jd�}|j|d� t jdddgt jd�}t jddgt jd�}t�|�\}}t�||� t�||� d S )Nl ���� r: )r( )rt r ) r. r= �uint64rv r0 r r) r1 r2 )r4 rw rg r8 r7 rA rB r! r! r& �test_uint64_factorize s z#TestFactorize.test_uint64_factorizec C sv t jdddgt jd�}|j|d� t jdddgt jd�}t jddgt jd�}t�|�\}}t�||� t�||� d S )Nl ���� l���� )r( )rt r r: ) r. r= �int64rv r0 r r) r1 r2 )r4 rw rg r8 r7 rA rB r! r! r& �test_int64_factorize s z"TestFactorize.test_int64_factorizec C s| t jdddddgtd�}|j|d� t jdddddgt jd�}t jdddgtd�}t�|�\}}t�||� t�||� d S ) NrD rF rE )r( )rt r r: r; ) r. r= rK rv r0 r r) r1 r2 )r4 rw rg r8 r7 rA rB r! r! r& �test_string_factorize% s z#TestFactorize.test_string_factorizec C s� t jddd t jddtdgtd�}|j|d� t jdddddd ddgt jd�}t jdddgtd�}t�|�\}}t � ||� t � ||� d S ) NrD rF rE )r( )rt r r: rH r; ) r. r= r> r rK rv r0 r r) r1 r2 )r4 rw rg r8 r7 rA rB r! r! r& �test_object_factorize/ s z#TestFactorize.test_object_factorizec C sj t �t �d�g�}|j|d� t jdgt jd�}t jdgdd�}t�|�\}}t�||� t�||� d S )Nz2020-01-01T00:00:00.000)rt r )r( z 2020-01-01T00:00:00.000000000zdatetime64[ns]) r. r= � datetime64rv r0 r` r) r1 r2 )r4 rw rg r8 r7 rA rB r! r! r& �test_datetime64_factorize9 s z'TestFactorize.test_datetime64_factorizec C s� t j�td��}tjdtjd�|f}tj||d�}t � |d |d � t j |d |d dd� |j|d�}t � |d |d � t j |d |d dd� d S )N� )r( )r r r: T)�exact) r` � RangeIndex� from_rangerm r. rL r0 r r) r1 r2 r3 )r4 r �rird �resultr! r! r& �test_factorize_rangeindexF s z'TestFactorize.test_factorize_rangeindexc C s� t j�td��}tjdtjd�|f}|d d d� }|d |f}|rf|d d d d� |d d d d� f}tj||d�}t � |d |d � t j |d |d dd� |j|d�}t � |d |d � t j |d |d dd� d S ) Nr� )r( rH r r: )r T)r� ) r` r� r� rm r. rL r0 r r) r1 r2 r3 )r4 r r� rd Zri2r� r! r! r& �$test_factorize_rangeindex_decreasingT s $ z2TestFactorize.test_factorize_rangeindex_decreasingc C sb t jdddgt jd�}tjtdd�� tj|dd� W d Q R X t� d�� t�|� W d Q R X d S ) Nl r: )r( zgot an unexpected keyword)rl T)�orderF) r. r= ry rn ro rp r r) r1 �assert_produces_warning)r4 rg r! r! r& �test_deprecate_orderg s z"TestFactorize.test_deprecate_orderrg �u8)r( l���� �i8Z__nan__�foorK c C sL t �|�\}}|ddg }tjdddgtjd�}t�||� t�||� d S )Nr r: )r( )r r) r. r= r0 r1 r2 )r4 rg rA rB r7 r8 r! r! r& �,test_parametrized_factorize_na_value_defaultp s z:TestFactorize.test_parametrized_factorize_na_value_defaultzdata, na_valuerD � rE r! )rD r: )rD r; c C sR t j||d�\}}|ddg }tjddddgtjd�}t�||� t�||� d S )N)�na_valuer: rI rH r )r( )r Zfactorize_arrayr. r= r0 r1 r2 )r4 rg r� rA rB r7 r8 r! r! r& �$test_parametrized_factorize_na_value� s z2TestFactorize.test_parametrized_factorize_na_valuer< rH i�����d z data, uniquesNZInt64Z numpy_arrayZextension_array)rc c C s� t j|||d�\}}|r<tjdd|dgtjd�}t �|�}n tjdd|dgtjd�}|}t�||� t|tj �r~t�||� n t� ||� d S )N)r r< r: r )r( ) r r) r. r= r0 Z safe_sortr1 r2 r* �ndarray� assert_extension_array_equal)r4 r r< rg rB rA r8 r7 r! r! r& �test_factorize_na_sentinel� s z(TestFactorize.test_factorize_na_sentinelr0 c C s. t j|d d�\}}t�||� t�||� d S )N)r< )r r) r1 r2 )r4 rg r8 r7 rA rB r! r! r& �&test_object_factorize_na_sentinel_none� s z4TestFactorize.test_object_factorize_na_sentinel_nonerY c C s. t j|d d�\}}t�||� t�||� d S )N)r< )r r) r1 r2 )r4 rg r8 r7 rA rB r! r! r& �#test_int_factorize_na_sentinel_none� s z1TestFactorize.test_int_factorize_na_sentinel_none)$�__name__� __module__� __qualname__rn �mark� parametrizer9 rC rO rT rX re rh rr rs rx rz r| r} r~ r� r� r� r� r. r= r� rK r� r` r> r� r( r� ru r� r! r! r! r& r 2 s� )1 , & $ r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zej�d d ejd d ged!�fd"ejd#ged!�fg�d$d%� �Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0S )1� TestUniquec C s0 t jjdddd�}t�|�}t|t j�s,t�d S )Nr r� �2 )�size)r. �random�randintr r, r* r� r_ )r4 rN r� r! r! r& � test_ints� s zTestUnique.test_intsc C s6 t jjdddd��d�}t�|�}t|t j�s2t�d S )Nr r� r� )r� rY ) r. r� r� �astyper r, r* r� r_ )r4 rN r� r! r! r& � test_objects� s zTestUnique.test_objectsc C s2 dddddg}x t d�D ]}tt�|�� qW d S )NrP rQ �C�D�Ei� )rm r] r r, )r4 �lstrj r! r! r& �test_object_refcount_bug� s z#TestUnique.test_object_refcount_bugc C sb t �t�d��d�t�t�d�d�g�}|j}|�� |�d�}t� |�}|�� t � ||� d S )NrG r; ) r Z from_arraysr. rL �repeat�tiler? r r` r, r1 �assert_almost_equal)r4 Zmindexrd r� r! r! r& �test_on_index_object� s $ z TestUnique.test_on_index_objectc C s |t jt j kr t�d� n�t|�r8dddg}ddg}n�t|�rTdddg}ddg}n�t|�r�tdd�tdd�tdd�g}tdd�tdd�g}nJt |�r�dddg}ddg}n.t |�r�d d d g}d d g}ndddg}ddg}t ||d �� � }t j||d �}t �||� d S ) Nzskip string dtyper: r; g �?g @r TFrP rQ )r( )r1 Z BYTES_DTYPESZ STRING_DTYPESrn �skipr r r ri r r r r, r. r= r2 )r4 Zany_numpy_dtyperg rB r� rd r! r! r& �test_dtype_preservation s, z"TestUnique.test_dtype_preservationc C s� t ddgdd�}tdddg�}t�|�}t�||� |j|jksDt�t|�}t�|�}t�||� |j|jksrt�|j }t�|�}t�||� |j|jks�t�d S )Nz"2015-01-03T00:00:00.000000000+0000z"2015-01-01T00:00:00.000000000+0000zM8[ns])r( z 2015-01-03T00:00:00.000000000z 2015-01-01T00:00:00.000000000) r r r r, r1 r2 r( r_ r r? )r4 rd Zdt_indexr� �srN r! r! r& �$test_datetime64_dtype_array_returned s& z/TestUnique.test_datetime64_dtype_array_returnedc C s@ t jdddgdd�}t�|�}t jddgdd�}t�||� d S )N�2000�2001z datetime64[s])r( zdatetime64[ns])r. r= r` r, r1 r2 )r4 rD r� rd r! r! r& �test_datetime_non_ns= s z TestUnique.test_datetime_non_nsc C s@ t jdddgdd�}t�|�}t jddgdd�}t�||� d S )Nr� r� ztimedelta64[s])r( l �RFl j�IGztimedelta64[ns])r. r= r` r, r1 r2 )r4 rD r� rd r! r! r& �test_timedelta_non_nsC s z TestUnique.test_timedelta_non_nsc C s� t jdddgdd�}tdddddg�}t�|�}t�||� |j|jksLt�t |�}t�|�}t�||� |j|jkszt�|j }t�|�}t�||� |j|jks�t�d S )Ni�y in� i' zm8[ns])r( ) r. r= r r r, r1 r2 r( r_ r r? )r4 rd Ztd_indexr� r� rN r! r! r& �%test_timedelta64_dtype_array_returnedI s z0TestUnique.test_timedelta64_dtype_array_returnedc C sB t ddddgtjd�}tjdddgtjd�}t�t�|�|� d S )Nr: r; l )r( )r r. ry r= r1 r2 r r, )r4 r� rM r! r! r& �test_uint64_overflow\ s z TestUnique.test_uint64_overflowc C s> dt jddg}t�|�}t jdt jdgtd�}t�||� d S )NrD rF )r( )r. r> r` r, r= rK r1 r2 )r4 Zduplicated_itemsr� rd r! r! r& �test_nan_in_object_arraya s z#TestUnique.test_nan_in_object_arrayc C s t td��}t td�td�dd�}t td��}|�� }t�||� t�|�}t�||� t td�dd�}|�� }t�||� t�|�}t�||� tt td��dd�}|�� }t�||� t�|�}t�||� tt td�td�d ��}t|�}|�� }t� ||� t�|�}t� ||� d S ) NZbac�abcT)� categories�orderedZbaabc)r� r� )�name)r� ) r r- r, r1 �assert_categorical_equalr r r` r r3 )r4 rd Z expected_orF r� r� �cir! r! r& �test_categoricalg s. zTestUnique.test_categoricalc C s t ttddd�tddd�g���� }t�t�tddd�g��}t� ||� ttddd�tddd�g��� }t dgdd d�}t� ||� t �t ttddd�tddd�g���}t�t�tddd�g��}t� ||� t �ttddd�tddd�g��}t dgdd d�}t� ||� d S ) N�20160101z US/Eastern)�tzz2016-01-01 00:00:00-0500z2016-01-01 00:00:00zdatetime64[ns, US/Eastern])r( rW z 2016-01-01) r r r r, r Z_from_sequencer. r= r1 r� r r3 r` )r4 r� rd r! r! r& �test_datetime64tz_aware� s8 z"TestUnique.test_datetime64tz_awarec C s: t �tddddg��}t�|tjdddgdd�� t �tdgdgd ��}t�|tjddgdd�� t �ttd�td�g��}tjdgd d�}t�||� t �ttdd d �tdd d �g��}t d gd d d�}t� ||� t �t d��}tjdddgt d�}t�||� t �tt t d����}t t d��}t�||� d S )Nr; r: rI r{ )r( rG r� z 2016-01-01T00:00:00.000000000zdatetime64[ns]z US/Eastern)r� z2016-01-01 00:00:00zdatetime64[ns, US/Eastern])r( rW ZaabcrD rE rF r� )r` r, r r1 r2 r. r= r r r r3 r- rK r r� )r4 r� rd r! r! r& �test_order_of_appearance� s( z#TestUnique.test_order_of_appearancez arg ,expected)�1r� �2r� r� )r( )r� r� c C s t �|�}t�||� d S )N)r` r, r1 r2 )r4 �argrd r� r! r! r& �test_tuple_with_strings� s z"TestUnique.test_tuple_with_stringsc C sB t jdd gtd�}t�|�}t jdd gtd�}tj||dd� d S )Nr� )r( T)Z strict_nan)r. r= rK r` r, r1 r2 )r4 rN r� rd r! r! r& �test_obj_none_preservation� s z%TestUnique.test_obj_none_preservationc C s4 t �ddg�}t�|�}t �dg�}t�||� d S )Ng �g )r. r= r` r, r1 r2 )r4 rD r� rd r! r! r& �test_signed_zero� s zTestUnique.test_signed_zeroc C s~ t �dt �dd��d }t �dt �dd��d }||ks<t�||ksHt�t�||g�}t�|�}t�tjg�}t � ||� d S )N�dz=Ql � r l � ) �struct�unpack�packr_ r. r= r` r, r> r1 r2 )r4 �NAN1�NAN2rD r� rd r! r! r& �test_different_nans s z TestUnique.test_different_nansc C s� d}d}t �dt �d|��d }t �dt �d|��d }||ksDt�||ksPt�xdtjtgD ]V}tj||g|d�}t� |�}|j dks�t�t �dt �d|d ��d }||ks\t�q\W d S )Nl � l � r� z=Qr )r( r: ) r� r� r� r_ r. ru rK r= r` r, r� ) r4 Z bits_for_nan1Z bits_for_nan2r� r� Zel_typerD r� Zresult_nan_bitsr! r! r& �test_first_nan_kept s z TestUnique.test_first_nan_keptc C sZ ||kr d S t j||gtd�}t�|�}|jdks6t�|d |ksFt�|d |ksVt�d S )N)r( r; r r: )r. r= rK r` r, r� r_ )r4 Zunique_nulls_fixtureZunique_nulls_fixture2rD r� r! r! r& � test_do_not_mangle_na_values# s z'TestUnique.test_do_not_mangle_na_valuesN) r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rn r� r� r. r= rK r� r� r� r� r� r� r! r! r! r& r� � s. *7# r� c @ s� e Zd Zdd� Zdd� Zdd� Zej�ddd d d g�ej�d d ddg�dd� ��Z dd� Z dd� Z dd� Z dd� Z dd� Zd d � Zd d � Zd d!� Zd"d#� Zej�d$g eed%�e�g �g�d&d'� �Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2S )3�TestIsinc C s~ d}t jt|d�� t�dd� W d Q R X t jt|d�� t�ddg� W d Q R X t jt|d�� t�dgd� W d Q R X d S )NzQonly list-like objects are allowed to be passed to isin\(\), you passed a \[int\])rl r: )rn ro rp r �isin)r4 rq r! r! r& � test_invalid/ s zTestIsin.test_invalidc C s� t �ddgdg�}t�ddg�}t�||� t �t�ddg�dg�}t�ddg�}t�||� t �tddg�dg�}t�ddg�}t�||� t �tddg�tdg��}t�ddg�}t�||� t �tddg�dh�}t�ddg�}t�||� t �ddgdg�}t�ddg�}t�||� t �tddg�tdg��}t�ddg�}t�||� t �tddg�dh�}t�ddg�}t�||� t �ddgdg�}t�ddg�}t�||� d S )Nr: r; TFrD rE )r r� r. r= r1 r2 r )r4 r� rd r! r! r& rO < s6 zTestIsin.test_basicc C sD t ddd�j}t�||d g�}t�dddg�}t�||� t�||dd� �}t�dddg�}t�||� t�|t|dd� ��}t�dddg�}t�||� t ddd�j}t�||d g�}t�dddg�}t�||� t�||dd� �}t�dddg�}t�||� t�|t|dd� ��}t�dddg�}t�||� d S ) NrU rI )�periodsr TFr; z1 day) r r? r r� r. r= r1 r2 r^ r )r4 rN r� rd r! r! r& �test_i8b s( zTestIsin.test_i8�dtype1zm8[ns]zM8[ns]z M8[ns, UTC]z period[D]r( r� Zf8r� c C s� t ddd�j}|dkr"|�d�}n*|dkr6|�d�}nt|�d���|�j}|�d��|�}t�||�}t j |j t d �}t �||� d S ) Nz 2013-01-01rI )r� z period[D]r� z M8[ns, UTC]�UTCr� )r( )r Z_valuesZ to_periodZ tz_localizer �viewr� r r� r. �zeros�shape�boolr1 r2 )r4 r( r� �dtarN �compsr� rd r! r! r& �+test_isin_datetimelike_values_numeric_comps~ s z4TestIsin.test_isin_datetimelike_values_numeric_compsc C sV t dddd�j}t�||dd� �}tjt|�td�}d|d< d|d < t� ||� d S ) NZ20000101i�� r� )r� rW r r; )r( Tr: ) r r? r r� r. r� r] r� r1 r2 )r4 r� r� rd r! r! r& � test_large� s zTestIsin.test_largec C s� t �ddddg�}dddg}ttdg��||��}ttdg��t �ddg�|��}t �ddddg�}t�||�}t�||� d S ) Nr r: r; rD rE rF TF) r. r= r r � from_codesr r� r1 r2 )r4 �vals�catsZSdZStrd r� r! r! r& �test_categorical_from_codes� s z$TestIsin.test_categorical_from_codesc C sx t �ddddg�}dddg}tdg��||�}tdg��t �ddg�|�}t �ddddg�}t�||�}t�||� d S ) Nr r: r; rD rE rF TF)r. r= r r� r r� r1 r2 )r4 r� r� �cat�otherrd r� r! r! r& �test_categorical_isin� s z TestIsin.test_categorical_isinc C s8 t jg}t jg}t �dg�}t�||�}t�||� d S )NT)r. r> r= r r� r1 r2 )r4 r� r? rd r� r! r! r& �test_same_nan_is_in� s z TestIsin.test_same_nan_is_inc C sJ t �dd�}t j|d< t�|t jdg�}t jt|�td�}t� ||� d S )Ng �?iAB r r: )r( ) r. r� r> r r� �onesr] r� r1 r2 )r4 r� r� rd r! r! r& �test_same_nan_is_in_large� s z"TestIsin.test_same_nan_is_in_largec C sT t �dd�}t|�}t j|d< |�t jdg�}tt jt|�td��}t� ||� d S )Ng �?iAB r r: )r( ) r. r� r r> r� r� r] r� r1 �assert_series_equal)r4 r� Zseriesr� rd r! r! r& � test_same_nan_is_in_large_series� s z)TestIsin.test_same_nan_is_in_large_seriesc C s` G dd� d�}|� |� }}t �t�|g|g�t�dg�� t �t�|g|g�t�dg�� d S )Nc @ s" e Zd Zed�dd�Zdd� ZdS )z0TestIsin.test_same_object_is_in.<locals>.LikeNan)�returnc S s dS )NFr! )r4 r� r! r! r& �__eq__� s z7TestIsin.test_same_object_is_in.<locals>.LikeNan.__eq__c S s dS )Nr r! )r4 r! r! r& �__hash__� s z9TestIsin.test_same_object_is_in.<locals>.LikeNan.__hash__N)r� r� r� r� r� r� r! r! r! r& �LikeNan� s r� TF)r1 r2 r r� r. r= )r4 r� rD rE r! r! r& �test_same_object_is_in� s z TestIsin.test_same_object_is_inc C s� t d�g}t d�g}|d |d k s(t�t�||�}t�t�dg�|� t�tj|t d�tj|t d��}t�t�dg�|� t�tj|tj d�tj|tj d��}t�t�dg�|� d S )Nr> r T)r( ) �floatr_ r r� r1 r2 r. r= r/ rK ru )r4 r� r? r� r! r! r& r� � s z TestIsin.test_different_nansc C s8 ddg}dg}t �ddg�}t�||�}t�||� d S )N�ss�* Z42F)r. r= r r� r1 r2 )r4 r� r? rd r� r! r! r& � test_no_cast� s zTestIsin.test_no_cast�empty)r( c C s6 t ddg�}t�ddg�}t�||�}t�||� d S )NrD rE F)r r. r= r r� r1 r2 )r4 r� r� rd r� r! r! r& � test_empty s zTestIsin.test_emptyc C s^ t jdt jd td�gtd�}t jtd�gtd�}t �dddg�}t�||�}t�||� d S )Nr> y �?)r( FT) r. r= r> r� rK r r� r1 r2 )r4 r� r� rd r� r! r! r& �test_different_nan_objects s z#TestIsin.test_different_nan_objectsc C s� t �dt �dd��d }t �dt �dd��d }||ks<t�||ksHt�tj||gtjd�}tj|gtjd�}t�||�}t�ddg�}t � ||� tj|gtjd�}t�||�}t�ddg�}t � ||� d S )Nr� z=Ql � r l � )r( T) r� r� r� r_ r. r= ru r r� r1 r2 )r4 r� r� rN Zlookup1r� rd Zlookup2r! r! r& � test_different_nans_as_float64 s z'TestIsin.test_different_nans_as_float64c C s< t dddgi�}|�dg�}t dddgi�}t�||� dS )zComparing df with int`s (1,2) with a string at isin() ("1") -> should not match values because int 1 is not equal str 1r? r: r; r� FN)r r� r1 �assert_frame_equal)r4 �dfr� �expected_falser! r! r& � test_isin_int_df_string_search' s z'TestIsin.test_isin_int_df_string_searchc C s> t dtjdgi�}|�dg�}t dddgi�}t�||� dS )z�Comparing df with nan value (np.nan,2) with a string at isin() ("NaN") -> should not match values because np.nan is not equal str NaNr? r; �NaNFN)r r. r> r� r1 r� )r4 r� r� r� r! r! r& � test_isin_nan_df_string_search/ s z'TestIsin.test_isin_nan_df_string_searchc C s< t dddgi�}|�dg�}t dddgi�}t�||� dS )z�Comparing df with floats (1.4245,2.32441) with a string at isin() ("1.4245") -> should not match values because float 1.4245 is not equal str 1.4245r? g�n���?g�#Ed�@z1.4245FN)r r� r1 r� )r4 r� r� r� r! r! r& � test_isin_float_df_string_search7 s z)TestIsin.test_isin_float_df_string_searchN) r� r� r� r� rO r� rn r� r� r� r� r� r� r� r� r� r� r� r� r rK r. r= r� r� r� r� r r r! r! r! r& r� . s( & ( r� c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )�TestValueCountsc C s� t j�d� ddlm} t j�d�}||d�}t�|�}ddddd g}t� |�� t d d ��}t d d d d g|d �}t �|�� |�� � d S )Ni� r )�cutrJ g��v���g ��Q� �g`��"��?gw��/��?g!�rh��?T)r� r: )r" )r. r� �seedZpandas.core.reshape.tiler �randnr � value_countsr Z from_breaksr� �CDTr r1 r� Z sort_index)r4 r rN �factorr� Zbreaksr" rd r! r! r& �test_value_countsA s z!TestValueCounts.test_value_countsc C sv ddddg}t j|dd�}tdgt�dg�d�}t�||� t j|ddd �}tddgt�d d g�d�}t�||� d S ) Nr: r; rI rJ )�bins)g� Zd;��?g @)r" F)r r )g� Zd;��?g @)g @g @)r r r r r+ r1 r� )r4 r� r� rd r! r! r& �test_value_counts_binsO s z&TestValueCounts.test_value_counts_binsc C s� t �ddg�}t|�dks t�t jddgdd�}t|�dks@t�t �tdddg��}t|�dksdt�d}tjt|d�� t jddgdd� W d Q R X d S )Nr: g �?)r r� r; z*bins argument only works with numeric data)rl )r r r] r_ r rn ro rp )r4 r� rq r! r! r& �test_value_counts_dtypes[ s z(TestValueCounts.test_value_counts_dtypesc C s� t t�d�tgdd�}tddg�}xH||gD ]<}t�|�}tj|dd�}t|�dksXt�t|�d ks,t�q,W t t d �di�}t � t�|�|� d S ) Ni' ztimedelta64[ns])r( r z 2014-01-01F)�dropnar: r; z2014-01-01 00:00:00) r r. � timedelta64r r r r r] r_ r r1 r� )r4 �td�dtr� �vcZ vc_with_naZexp_dtr! r! r& �test_value_counts_nati s z%TestValueCounts.test_value_counts_natc C s� t tddd�tddd�tddd�tddd�tddd�tddd�g�}|�� }ttddd�tddd�tddd�gtd�}t dddg|d�}t�||� tt d tj g�d d �}t d tj gtd�}t�||� d S ) Ni� r: i� ip )r( rI r; )r" z 2362-01-01�ignore)�errors) r r r r rK r1 r� r r. r> )r4 r� �resZ exp_indexrM r! r! r& �&test_value_counts_datetime_outofboundsw s z6TestValueCounts.test_value_counts_datetime_outofboundsc C sv t ttd���}|�� }t dddgtdddg�d�}tj||d d � |j�� }|�� }|j �� |_ tj||d d � d S ) NZaaabbcrI r; r: rD rE rF )r" T)�check_index_type) r r r- r r r1 r� r� Z as_orderedr" )r4 r� r� rd r! r! r& r� � s z TestValueCounts.test_categoricalc C sN t ttd���}tj|jd< |�� }t dddgtdddgdddgd �d �}tj ||d d � |jd d�}t ddddgtdddtjg�d �}tj ||d d � t ttd�d dddgd��}tj|jd< |�� }t dddgtdddgdddgd d�d �}tj ||d d � |jd d�}t ddddgtdddtjgdddgd d�d �}tj ||d d � d S )NZ aaaaabbbccr: rJ rI r; rD rE rF )r� )r" T)r F)r )r� r� )r� r� ) r r r- r. r> Zilocr r r1 r� )r4 r� r� rd r! r! r& �test_categorical_nans� s2 " z%TestValueCounts.test_categorical_nansc C s` t ttd�td�dd��}|�� }t ddddgtd d d d gtd�dd�d �}tj||dd� d S )NZbbbaac�abcdT)r� r� rI r; r: r rE rD rF r� )r" )r )r r r- r r1 r� )r4 r� r� rd r! r! r& �test_categorical_zeroes� s z'TestValueCounts.test_categorical_zeroesc C s� t �tdddg�jdd�tddgddgd�� t �tdddg�jdd�tddgddgd�� t �tdgd dgd d gd �jdd�tddgddgd�� t �tdgd dgd d gd �jdd�tdddgddtjgd�� t �td d d g�jdd�tddgd d gd�� t �td d d g�jdd�tddgd d gd�� t �td d d d g�jdd�tddgd d gd�� td d d d d d g�jdd�}tdddgd d tjgd�}t �||� d S ) NTF)r r; r: )r" rI rG g������$@g @)r1 r� r r r. r> )r4 r� rd r! r! r& � test_dropna� s0 (( zTestValueCounts.test_dropnac C s� t dgd dgd tjgd �}tjtdf}x�|D ]�}|�|�}|jddd�}t d d d gt tjd d g|d�d�}t�||� |jddd�}t ddgt d d g|d�d�}t�||� q6W d S )Nr: r; rI rG zM8[ns]TF)� normalizer g �?g333333�?g�������?g @g �?)r( )r" g333333�?g�������?) r r. r> ru rK r� r r1 r� )r4 r� Zdtypes�tZs_typedr� rd r! r! r& � test_value_counts_normalized� s $ z,TestValueCounts.test_value_counts_normalizedc C sx t jdgt jd�}tdgdgd�}t�|�}t�||� t jddgtd�}tddgddgd�}t�|�}t�||� d S )Nl )r( r: )r" rH ) r. r= ry r r r r1 r� rK )r4 rN rd r� r! r! r& �test_value_counts_uint64 s z(TestValueCounts.test_value_counts_uint64N)r� r� r� r r r r r r� r r r r r r! r! r! r& r @ s $ &r c @ s. e Zd Zdd� Zej�de�dddddddddd g �e�d d d ej d d d d ej dg �e�ddddddddddg �ejddddddddddg e d�ejdd dd d d d dd d g ej d�g�d!d"� �Z d#d$� Z d%d&� Zej�d'd(d)d*d+d(d)d*d+gd(d)d*d+gfd,d-d-d,gd,d-gfd.d/d0d.gd.d/d0gfg�d1d2� �Zd3S )4�TestDuplicatedc C s� t jddt jddt jgtd�}t�|�}t �ddddddg�}t�||� tj|dd�}t �ddddddg�}t�||� tj|d d�}t �ddddddg�}t�||� tj|dd�}t �ddddddg�}t�||� t jd td�}xDt t ddt jt jgd dt jdt jgd ��D ]\}}|||< �qW t�|�}dgd }dgd }t �|| �}t�||� tj|d d�}t �|| �}t�||� tj|dd�}t �|| �}t�||� d S ) Nr r: r; )r( FT�first)�keep�last� rJ ) r. r= r> rK r � duplicatedr1 r2 r� � enumerate�zip)r4 �keysr� rd rj r ZfalsesZtruesr! r! r& �test_duplicated_with_nas s8 4 z'TestDuplicated.test_duplicated_with_nas�caser: r; rG rI rJ � g�������?g������@gffffff @g������@gffffff@y �? �?y @ @y @ @y @ @y @ @y @ @rD rE �erF r� �f)r( l �� r� �' � c C sp t �ddddddddddg �}t �ddddddddddg �}||B }tj|dd�}t�||� tj|dd�}t�||� tj|dd�}t�||� xdt|�t|dd�gD ]L}|jdd�}t�||� |jdd�}t�||� |jdd�}t�||� q�W xrt|�t|dd�gD ]Z} | jdd�}t�|t|�� | jdd�}t�|t|�� | jdd�}t�|t|�� �qW d S )NFTr" )r# r$ �category)r( ) r. r= r r& r1 r2 r r r� ) r4 r+ � exp_first�exp_last� exp_false� res_first�res_last� res_false�idxr� r! r! r& �test_numeric_object_likes: s2 z(TestDuplicated.test_numeric_object_likesc C s& ddddddddddg }ddddd dd ddd g }t �d d � |D ��t �dd � |D ��t �dd � |D ��t �dd � |D ��t �dd � |D ��g}t �ddddddddddg �}t �ddddddddddg �}||B }�xL|D �]B}tj|dd�}t�||� tj|dd�} t�| |� tj|dd�} t�| |� xpt|�t|dd�t|td�gD ]N} | jdd�}t�||� | jdd�} t�| |� | jdd�} t�| |� �qNW x|t|�t|dd�t|td�gD ]Z} | jdd�}t� |t|�� | jdd�} t� | t|�� | jdd�} t� | t|�� �q�W q�W d S )Nz 2011-01-01z 2011-01-02r z 2011-01-03z 2011-01-04z 2011-01-06z1 daysz2 daysz3 daysz4 daysz6 daysc S s g | ] }t |��qS r! )r )r# r� r! r! r& r' � s z6TestDuplicated.test_datetime_likes.<locals>.<listcomp>c S s g | ]}t |d d��qS )z US/Eastern)r� )r )r# r� r! r! r& r' � s c S s g | ]}t |d d��qS )r� )rW )r )r# r� r! r! r& r' � s c S s g | ]}t �|��qS r! )r. r )r# r� r! r! r& r' � s c S s g | ] }t |��qS r! )r )r# r� r! r! r& r' � s FTr" )r# r$ r2 )r( ) r. r= r r& r1 r2 r rK r r� ) r4 r r �casesr3 r4 r5 r+ r6 r7 r8 r9 r� r! r! r& �test_datetime_likes{ sr z"TestDuplicated.test_datetime_likesc C sV t dddg�t�dd�g}x6|D ].}|jdks2t�t�|�� t� dddg�� q W d S )Nr: r; rI r TF) r r` r� Z is_uniquer_ r1 r2 r& r. r= )r4 r; r+ r! r! r& �test_unique_index� s z TestDuplicated.test_unique_indexz arr, uniques)r r )r r: )r: r )r: r: )rE rF )rD rE )rD r: )rE r; )rD rI c C s8 t jt|�td�}||d d �< t�|�}t�||� d S )N)r( )r. r� r] rK r` r, r1 r2 )r4 rN rB rd r� r! r! r& �test_unique_tuples� s z!TestDuplicated.test_unique_tuplesN)r� r� r� r* rn r� r� r. r= r> rK ry r: r< r= r> r! r! r! r& r! s8 ' ,,S r! c @ sb e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j � d ddg�e j � de j e jddfe je jddfe je jddfe je jddfe je jddfe je jddfg�dd� ��Ze j � de j dfe jdfe jd fe jd fe jd fg�d d � �Ze j � de j dfe jdfe jd fe jd fe jd fg�d!d"� �Ze j � d#e j e je je je je jg�d$d%� �Zd&S )'� TestHashTablec C st t �� }|�dd� |�d�dks&t�tjtdd�� |�dd� W d Q R X tjtdd�� |�d� W d Q R X d S )Nrb r: z'key' has incorrect type)rl rJ r, z'val' has incorrect type)r\ �StringHashTable�set_itemZget_itemr_ rn ro rp )r4 �tblr! r! r& �(test_string_hashtable_set_item_signature� s z6TestHashTable.test_string_hashtable_set_item_signaturec C s^ t �ddt jddddg�}|j|d� t�� }|�|� t�|� |�t j t |�t j d�� d S ) NgX9��v�@g ��Q� @i����rG r; rI )rt )r( ) r. r= r> rv r\ �Float64HashTable� map_locationsr1 r2 �lookuprL r] r0 )r4 rw �xs�mr! r! r& �test_lookup_nan� s z TestHashTable.test_lookup_nanc C s: d}t �|�}|�dd� |�dd� t|�dks6t�d S )NrJ g r g �r: )r\ rD rA r] r_ )r4 �NrH r! r! r& �test_add_signed_zeros s z#TestHashTable.test_add_signed_zerosc C s| t �dt �dd��d }t �dt �dd��d }||ks<t�||ksHt�t�� }|�|d� |�|d� t|�dksxt�d S )Nr� z=Ql � r l � r: )r� r� r� r_ r\ rD rA r] )r4 r� r� rH r! r! r& �test_add_different_nans s z%TestHashTable.test_add_different_nansc C sZ t jdddgt jd�}|j|d� t�� }|�|� t�|� |�t j t |�t j d�� d S )Nr: r; l )r( )rt ) r. r= ry rv r\ �UInt64HashTablerE r1 r2 rF rL r] r0 )r4 rw rG rH r! r! r& �test_lookup_overflow s z"TestHashTable.test_lookup_overflowc C s@ t ddddgtjd�}tjdddgtjd�}t�|�� |� d S )Nr: r; l )r( )r r. ry r= r1 r2 r, )r4 r� rM r! r! r& �test_get_unique$ s z TestHashTable.test_get_unique�nvalsr r� z&htable, uniques, dtype, safely_resizesrK FTru r{ r[ ry c C s� t jt j�d�|d�}|j|d� |� }|� }|�|d |� |dd� |�� }|j} |rj|�||dd� n*tj t dd�� |�||dd� W d Q R X |�� |j| ks�t �d S )Ni� )r( )rt r rH zexternal reference.*)rl ) r. r= r� r rv Z get_labelsZto_arrayr� rn ro � ValueErrorr_ ) r4 rw �htablerB r( Zsafely_resizesrP r� �tmpZoldshaper! r! r& �test_vector_resize) s z TestHashTable.test_vector_resizezhtable, tm_dtype�String�FloatZIntZUIntc C s� t td| d �}t|d��}|tjkr6tj|jd< n |tjkrVtjd t g|jdd�< |j ddd�j dd �}|j j |d � |jd d �j }|� �|j �}t�||� |� j|j dd �\}} t�||� || } t�| |j � d S )N�maker i� i� i� rI T)�frac�replace)�drop)rt r" )r# )Zreturn_inverse)�getattrr1 r r\ rD r. r> �loc�PyObjectHashTabler �sample� reset_indexr? rv �drop_duplicatesr, r2 ) r4 rR �tm_dtyperw �makerr� � s_duplicated�expected_unique� result_unique�result_inverseZreconstrr! r! r& �test_hashtable_uniqueV s z#TestHashTable.test_hashtable_uniquec C s� t td| d �}t|d��}|tjkr6tj|jd< n |tjkrVtjd t g|jdd�< |j ddd�j dd �}|j j |d � |�� j }|� �|j �\}} |�� �� j } t�|| � || | } |�� j } t�| | � d S ) NrW r i� i� i� rI T)rX rY )rZ )rt )r[ r1 r r\ rD r. r> r\ r] r r^ r_ r? rv ra r) r r` r2 ) r4 rR ra rw rb r� rc Zna_maskre rf rd Zresult_reconstructZexpected_reconstructr! r! r& �test_hashtable_factorize~ s z&TestHashTable.test_hashtable_factorizer c C s t �t j�jd }||d�}d S )Nr: )� size_hint)r. �iinfoZuint32�max)r4 r ri rB r! r! r& � test_hashtable_large_sizehint� s z+TestHashTable.test_hashtable_large_sizehintN) r� r� r� rC rI rK rL rN rO rn r� r� r\ r] Z ObjectVectorr@ rD Z Float64VectorZInt64HashTableZ Int64VectorZInt32HashTableZ Int32VectorrM Z UInt64VectorrT rg rh rl r! r! r! r& r? � sJ $!!r? c C sN t tj�d��} t�| dddddg�}t�| jdddddg�}t�||� d S )Nr� r g �?g �?g �?g �?) r r. r� r r Zquantiler? r1 r� )r� r� rd r! r! r& � test_quantile� s rm c C s� t j�ddd��d�} t�| �}t j| dd�d }tj||dd� d | t j� t | �d �< t�| �}t j| dd�d dd � }tj||dd� d S ) Nr: i i � r{ T)Z return_indexF)Z check_dtyperH r� ) r. r� r� r� r\ Zunique_label_indicesr, r1 r2 �choicer] )rD �left�rightr! r! r& �test_unique_label_indices� s rq c @ s� e Zd Zejdd� �Zej�de j d �dd� �Z dd� Z d d � Z ejjejjejjd e �d �e �d ��d d�gddgd�dd� ���ZdS )�TestRankc sn ddl m� � fdd�}|t�tjtjdddtjdddtjg �� |t�d tjdddtjddd tjg �� d S ) Nr )�rankdatac s^ t �| � }| �� } tj| t jt| �t jd�d�}t j| |< � | �}t j ||< t � ||� d S )N)r( )�labels) r. �isfinite�copy�libalgosZrank_1dr� r] r0 rR r> r1 r� )rN �maskr� rM )rs r! r& �_check� s z*TestRank.test_scipy_compat.<locals>._checkg @r: r; rI g @)Z scipy.statsrs r. r= r> )r4 ry r! )rs r& �test_scipy_compat� s *zTestRank.test_scipy_compatr( � AllIntegerc C sT t jddgt jd�}t jddg|d�}|j|d� t|�}t�|�}t�||� d S )Nr: r; )r( r� )rt ) r. r= ru rv r r �rankr1 r2 )r4 rw r( rM rg r@ r� r! r! r& rO � s zTestRank.test_basicc C sP t jddgt jd�}x6t jt jgD ]&}tddg|d�}t�t�|�|� q"W d S )Nr: r; )r( l ) r. r= ru ry r r1 r2 r r| )r4 rM r( r� r! r! r& r� � s z TestRank.test_uint64_overflowc C sP t �dddgdddgddd ggg�}d }tjt|d �� t�|� W d Q R X d S ) Nr: r; rI rJ rG r, r1 r% � z%Array with ndim > 2 are not supported)rl )r. r= rn ro rp r r| )r4 rN rq r! r! r& �test_too_many_ndims� s $z TestRank.test_too_many_ndimsr? i i r; Z1dZ2d)rc c C s" t j|dd��� }|dks t�d S )NT)Zpctr: )r r| rk r_ )r4 r? r� r! r! r& �test_pct_max_many_rows� s z TestRank.test_pct_max_many_rowsN)r� r� r� r Zskip_if_no_scipyrz rn r� r� r. � typecodesrO r� r~ ZsingleZ high_memoryrL �reshaper r! r! r! r& rr � s rr c C s� t jg dd�} t jtddd�gdd�}tjd | |�}t jdgt jd�}t�||� tjd || �}t jg t jd�}t�||� tjd | |�}t jdgt jd�}t�||� tjd || �}t jg t jd�}t�||� d S )NrY )r( i� � � rK rH ) r. r= r rw �padr0 r1 r2 �backfill)�old�newr� rd r! r! r& �!test_pad_backfill_object_segfault s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�TestTseriesUtilc C s d S )Nr! )r4 r! r! r& �test_combineFunc s z TestTseriesUtil.test_combineFuncc C s d S )Nr! )r4 r! r! r& � test_reindex s z TestTseriesUtil.test_reindexc C s d S )Nr! )r4 r! r! r& � test_isna! s zTestTseriesUtil.test_isnac C s d S )Nr! )r4 r! r! r& � test_groupby$ s z TestTseriesUtil.test_groupbyc C s d S )Nr! )r4 r! r! r& �test_groupby_withnull' s z%TestTseriesUtil.test_groupby_withnullc C s� t dddg�}t ttd���}tjd |j|j�}tjddddddddddddg tjd �}t � ||� t dd g�}t ttdd���}tjd |j|j�}tjdddddgtjd �}t � ||� d S ) Nr: rG r� r� �int64_tr r; rH )r( rJ ) r r- rm rw r� r? r. r= r0 r1 r2 )r4 r� r� �filler� expect_fillerr! r! r& � test_backfill* s ( z TestTseriesUtil.test_backfillc C s� t dddg�}t ttd���}tjd |j|j�}tjddddddddddddg tjd �}t � ||� t ddg�}t t� d��}tjd |j|j�}tjdddddgtjd �}t � ||� d S ) Nr: rG r� r� r� rH r r; )r( ) r r- rm rw r� r? r. r= r0 r1 r2 rL )r4 r� r� r� r� r! r! r& �test_pad; s ( zTestTseriesUtil.test_padN) r� r� r� r� r� r� r� r� r� r� r! r! r! r& r� s r� c ~ C s t jddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddg|dd�t jddd d d d d ddddddddddddddd d d d d d!ddddddd d d d d ddddddddddddddd d d d d d!ddddddd d d d d ddddddddddddddd d d d d d!ddddddd d d d d ddddddddddddddd d d d d d!ddddg|dd�g} t�| ��r t�d S )"NrI r; r: r r{ )r( � � � � � � � � � � rZ � � rk � � � � r� � r� r} r% r1 r, rG rJ )r. r= rw Z is_lexsortedr_ )Zfailurer! r! r& �test_is_lexsortedL s� r� c C s� t j�ddd��t j�} t j�ddd��t j�}t�| d�d }t j| dd�}|�t j�}t� ||� | d | }t�|d�d }t � || f�}|�t j�}t� ||� d S )Nr i� r� Z mergesort)�kindi@B ) r. r� r� r� r0 rw Zgroupsort_indexer�argsortr1 r2 Zlexsort)rD rE r� rd rb r! r! r& �test_groupsort_indexerU s r� c s� t �� � t �� ��td�dddtd�� g} t� fdd�| D ��sDt�t� fdd�| D ��s^t�� � krn� � ksrt�� � k s�� � kr�t�t �� t �� ks�t�t �� t �� kr�t�t�fd d�| D ��s�t�t�fd d�| D ��s�t���kr��ks�t���k �s ��k�rt�t �� t �� k�s$t�t �� t �� k�r:t�x$t| �D ]}t|�| k�sDt��qDW t� t �� gd �� � t� t �� gd �� � d S ) Nz-infg}Ô%�I��r g}Ô%�I�TrR c 3 s | ]}� |kV qd S )Nr! )r# rS )�Infr! r& � <genexpr>y s z%test_infinity_sort.<locals>.<genexpr>c 3 s | ]}� |kp|� kV qd S )Nr! )r# rS )r� r! r& r� z s c 3 s | ]}� |kV qd S )Nr! )r# rS )�NegInfr! r& r� � s c 3 s | ]}� |k p|� kV qd S )Nr! )r# rS )r� r! r& r� � s � ) rw �Infinity� NegInfinityr� �allr_ r �sortedr. r= r� )Zref_numsZpermr! )r� r� r& �test_infinity_sortn s& r� c C s� t �� } t �� }| tjkr t�| tjkr,t�| tjk r:t�| tjkrHt�| tjkrVt�| tjksdt�|tjkrrt�|tjkr�t�|tjk r�t�|tjkr�t�|tjkr�t�|tjks�t�d S )N)rw r� r� r. r> r_ )r� r� r! r! r& �test_infinity_against_nan� s r� c C s* t jdt jd�} t�| �}|| ks&t�d S )Nr� )r( )r. rL r0 rw Zensure_platform_intr_ )rN r� r! r! r& �test_ensure_platform_int� s r� c C s� d} t �t j�j}t �t j�j}tjt| d�� t� t � ||g�|� W d Q R X tjt| d��& t� t � ||g�t � ||g�� W d Q R X tjt| d�� t� t � ||g�|� W d Q R X tjt| d��& t� t � ||g�t � ||g�� W d Q R X tjt| d��& t� t � ||g�t � ||g�� W d Q R X tjt| d��4 tj t � ||g�t � ||g�t � ddg�d� W d Q R X tjt| d��4 tj t � ||g�t � ||g�t � ddg�d� W d Q R X tjt| d��@ tj t � ||g�t � ||g�t � ddg�t � ddg�d� W d Q R X tjt| d��> t � t ��( t� t � ||g�t � t j|g�� W d Q R X W d Q R X tj t � ||g�t � ||g�t � ddg�d� tj t � ||g�t � ||g�t � ddg�d� tj t � ||g�t � ||g�t � ddg�t � ddg�d� d S )NzOverflow in int64 addition)rl FT)�arr_mask)�b_mask)r� r� )r. rj r{ rk �minrn ro � OverflowErrorr Zchecked_add_with_arrr= r1 r� �RuntimeWarningr> )rq rH �nr! r! r& �test_int64_add_overflow� sJ * **44 6** r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�TestModec C s. t g tjtg td�d�}t�t�g �|� d S )N)r( )r( r" ) r r. ru r �intr1 r� r �mode)r4 rM r! r! r& � test_no_mode� s zTestMode.test_no_modec C s� dg}dg}dg}ddg}xpt jd t jd D ]X}t||d�}t||d�}t�t�|�|� t||d�}t||d�}t�t�|�|� q0W tdgtd�}t�t�dg�|� tdddgtd�}t�t�dddg�|� d S )Nr: r{ rV )r( rD rE rF ) r. r� r r1 r� r r� r� rK )r4 � exp_single� data_single� exp_multi� data_multir r� rM r! r! r& �test_mode_single� s zTestMode.test_mode_singlec C s� dg}dgd dgd }ddg}dgd dgd dgd }xpt jd t jd D ]X}t||d�}t||d�}t�t�|�|� t||d�}t||d�}t�t�|�|� qVW d S )Nr: rG r; rI r{ rV )r( )r. r� r r1 r� r r� )r4 r� r� r� r� r r� rM r! r! r& �test_number_mode� s zTestMode.test_number_modec C s� dg}dgd dgd }t |dd�}t |dd�}t�t�|�|� dg}dgd dgd }x:ttgD ].}t ||d�}t ||d�}t�t�|�|� qhW d S ) NrE rD r; rI rF )r( �barr� )r r1 r� r r� �strrK )r4 rM rg r� r r! r! r& �test_strobj_mode s zTestMode.test_strobj_modec C sr t dddgdd�}t dddgdd�}t�t�|�|� t ddgdd�}t dddddgdd�}t�t�|�|� d S )Nz 1900-05-03z 2011-01-03z 2013-01-02zM8[ns])r( )r r1 r� r r� )r4 rM r� r! r! r& �test_datelike_mode s zTestMode.test_datelike_modec C st t dddgdd�}t dddgdd�}t�t�|�|� t ddgdd�}t dddd ddgdd�}t�t�|�|� d S ) Nz-1 daysz0 daysz1 daysztimedelta64[ns])r( z2 minz1 dayz-1 dayz -1 day 2 min)r r1 r� r r� )r4 rM r� r! r! r& �test_timedelta_mode% s z TestMode.test_timedelta_modec C s. t dg�}t dddg�}t�t�|�|� d S )Nr� r: )r r1 r� r r� )r4 rM r� r! r! r& �test_mixed_dtype1 s zTestMode.test_mixed_dtypec C sp t dgtjd�}t dddgtjd�}t�t�|�|� t ddgtjd�}t ddgtjd�}t�t�|�|� d S )Nl )r( r: )r r. ry r1 r� r r� )r4 rM r� r! r! r& r� 6 s z TestMode.test_uint64_overflowc C s� t ddg�}|}t�t�|�|� t�|�� |� t dddg�}t dgddgd�}t�t�|�|� t�|�� |� t dddddg�}t ddgdddgd�}t�t�|�|� t�|�� |� d S )Nr: r; rD )r� rI )r r1 r� r r� )r4 rF rM r! r! r& r� ? s zTestMode.test_categoricalc C s� t dddg�}tdddgtjd�}t�t�|�|� t dddg�}tdgtd�}t�t�|�|� t dddddg�}tddgtjd�}t�t�|�|� tddgdd�}t ddd d ddgdd�}t�t�|�|� d S ) Nr: r; rI )r( rD z2 minz1 dayztimedelta64[ns]z-1 dayz -1 day 2 min) r r r. r{ r1 r� r r� rK )r4 r9 rM r! r! r& � test_indexO s zTestMode.test_indexN) r� r� r� r� r� r� r� r� r� r� r� r� r� r! r! r! r& r� � s r� c @ sH e Zd Zej�dddg�dd� �Zdd� Zej�ddd g�d d � �Zd S ) �TestDiffr( zM8[ns]zm8[ns]c C s� t �d��t j��|��dd�}|j�dd�|d d �df< tj |ddd �}t j |j d d �d }t � dd�|d d �df< t � dd�|dd d �f< t �||� tj |jddd �}t �||j� d S ) Nr� rI rJ r �nsr; r: r )�axisztimedelta64[ns])r( )r. rL r� r{ r� r� r( �typer �diffr� r� r r1 r2 �T)r4 r( rN r� rd r! r! r& �test_diff_datetimelike_nate s z#TestDiff.test_diff_datetimelike_natc C sB t dddd�j}d}tjt|d�� tj|ddd� W d Q R X d S ) Nz 2016-01-01rI z US/Pacific)r� r� z#cannot diff DatetimeArray on axis=1)rl r: )r� )r �_datarn ro rQ r r� )r4 r� rq r! r! r& �test_diff_ea_axisu s zTestDiff.test_diff_ea_axisZint8Zint16c C sN t jdddddg|d�}t�|d�}t jt jddddgdd�}t�||� d S )Nr r: )r( rH Zfloat32)r. r= r r� r> r1 r2 )r4 r( rN r� rd r! r! r& �test_diff_low_precision_int| s z$TestDiff.test_diff_low_precision_intN) r� r� r� rn r� r� r� r� r� r! r! r! r& r� d s r� �opc C s� | ddddg�}| ddddg�}| ddddddg�}t |tj�rZt�||�}t�||� n.t�t�� t�||�}W d Q R X t� ||� d S )NrI r: rJ r; ) r* r. r� r Zunion_with_duplicatesr1 r2 r� r� r� )r� ZlvalsZrvalsrd r� r! r! r& �test_union_with_duplicates� s r� )Pr � itertoolsr r� �numpyr. rn Z pandas._libsr rw r r\ Z pandas.compatr Z pandas.util._test_decorators�utilZ_test_decoratorsr Zpandas.core.dtypes.commonr r r r r Zpandas.core.dtypes.dtypesr r Zpandasr` r r r r r r r r r r r r r r r r r Zpandas._testingZ_testingr1 Zpandas.core.algorithms�coreZ algorithmsZpandas.core.arraysr Zpandas.core.common�commonrf r r� r� r r! r? rm rq rr r� r� r� r� r� r� r� r� r� r� r� r� r= r� r! r! r! r& �<module> s` L 1 P S \ K62 !4
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка