3
���h � @ sf d dl Zd dlZd dlmZmZ d dlZd dlm Z d dl
j Z G dd� d�Z
G dd� d�Zdd � ZdS )
� N)�lib�writers)�Indexc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestMiscc
C s� t jddt jgdd� }}tj|�dks,t�|jd�jt�}tj|�dksNt�|jd�jt�}tj|�dkspt�d}tj t
|d �� tj|jd�� W d Q R X d S )
N�foo�b�object)�dtype� �U�SzNo matching signature found)�match) �np�array�nan�
libwritersZmax_len_string_array�AssertionErrorZastyper �pytest�raises� TypeError)�self�arr�a�msg� r �7/tmp/pip-build-5_djhm0z/pandas/pandas/tests/test_lib.py�test_max_len_string_array s z"TestMisc.test_max_len_string_arrayc C s� ddgddgddgg}dd� |D �}t jdddddg�}tj|dd �}tjt j|�|� d
d� |D �}t jdddddg�}tj|d d �}tjt j|�|� d S ) N�pr �n�d�sc s s | ]
}|V qd S )Nr )�.0�keyr r r � <genexpr>! s zCTestMisc.test_fast_unique_multiple_list_gen_sort.<locals>.<genexpr>T)�sortc s s | ]
}|V qd S )Nr )r! r" r r r r# & s F)r r r Z fast_unique_multiple_list_gen�tm�assert_numpy_array_equal)r �keys�gen�expected�outr r r �'test_fast_unique_multiple_list_gen_sort s z0TestMisc.test_fast_unique_multiple_list_gen_sortc C s> t jdtjd�g�g}tjt�� tj|d d� W d Q R X d S )Nr �2000)r$ ) r r �pdZ Timestampr% Zassert_produces_warning�RuntimeWarningr Zfast_unique_multiple)r r r r r �3test_fast_unique_multiple_unsortable_runtimewarning+ s z<TestMisc.test_fast_unique_multiple_unsortable_runtimewarningN)�__name__�
__module__� __qualname__r r+ r/ r r r r r s
r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� TestIndexingc C s� t jd�}t jg t jd�}tj|t|��}t|t�s8t �t
j || || � x�d
D ]�}x�d D ]�}t jd ||t jd�}tj|t|��}t|t�s�t �t
j || || � |d d d � }tj|t|��}t|t�s�t �t
j || || � q\W qRW x�dddd gdddd gd dddgd
d dgdd dggD ]X}t j|t jd�}tj|t|��}t|t� �sXt �t
j ||� t
j || || � �q"W d S )N�d )r � � � � �c � r )r5 r6 r7 r8 r9 )r5 r6 r: ������r< ) r �aranger �intpr �maybe_indices_to_slice�len�
isinstance�slicer r% r& )r �target�indices� maybe_slice�end�step�caser r r �%test_maybe_indices_to_slice_left_edge2 s*
8 z2TestIndexing.test_maybe_indices_to_slice_left_edgec " C sF t jd�}x�dD ]�}x�dD ]�}t j|d
|t jd �}tj|t|��}t|t�sPt�t j
|| || � |d d d� }tj|t|��}t|t�s�t�t j
|| || � qW qW t j ddd
dgt jd �}tj|t|��}t|t� s�t�t j
||� d }t j
t|d
�� || W d Q R X t j
t|d
�� || W d Q R X t j dd
ddgt jd �}tj|t|��}t|t� �srt�t j
||� t j
t|d
�� || W d Q R X t j
t|d
�� || W d Q R X x~d
dd
dgd
d
ddgddddggD ]X}t j |t jd �}tj|t|��}t|t� �s t�t j
||� t j
|| || � �q�W d S )Nr4 r r6 r7 r8 �a �b r5 r: r9 )r z7index 100 is out of bounds for axis (0|1) with size 100)r
�` )r r6 r7 r8 rJ rK )r5 r6 r: r; )r r= r> r r? r@ rA rB r r% r& r r r �
IndexError)r rC �startrG rD rE r rH r r r �&test_maybe_indices_to_slice_right_edgeT sD
( z3TestIndexing.test_maybe_indices_to_slice_right_edgec C s t jd�}x�d
D ]�}t jdd|t jd �}tj|t|��}t|t�sFt�t j
|| || � |d d d � }tj|t|��}t|t�s�t�t j
|| || � qW x|dddd gddddgddddggD ]V}t j |t jd �}tj|t|��}t|t� � s�t�t j
||� t j
|| || � q�W d S )
N�
r5 r6 r: r7 � � r )r )r5 r6 r: r7 rQ rR r; r< ) r r= r> r r? r@ rA rB r r% r& r )r rC rG rD rE rH r r r �&test_maybe_indices_to_slice_both_edges� s
( z3TestIndexing.test_maybe_indices_to_slice_both_edgesc C s2 t jd�}x�dD ]�\}}x�dD ]�}t j|||t jd �}tj|t|��}t|t�sTt�t j
|| || � |d d d� }tj|t|��}t|t�s�t�t j
|| || � q W qW x|d d
dd
gd
d
ddgddd
dggD ]V}t j |t jd �}tj|t|��}t|t� �s
t�t j
||� t j
|| || � q�W d S )Nr4 r6 rP r7 � �A rJ r5 r: r8 )r � � � �r6 rP �r7 rT �rU rJ )rY rZ r[ )r5 r6 r: r8 r; ) r r= r> r r? r@ rA rB r r% r& r )r rC rN rF rG rD rE rH r r r �"test_maybe_indices_to_slice_middle� s"
( z/TestIndexing.test_maybe_indices_to_slice_middlec C s` t jdddddddgt jd�}tj|�}|jt jks8t�tj|d d� �}|tdd�ks\t�d S )Nr r5 )r ) r r Zuint8r Zmaybe_booleans_to_slicer Zbool_r rB )r r �resultr r r � test_maybe_booleans_to_slice� s
z)TestIndexing.test_maybe_booleans_to_slicec C sV t jd
d dddd ddgt jd�}tj|d�}t jddddd gt jd�}tj||� d S )
Nr5 r6 r r
r: )r r7 � � r; r; r; )r r Zint64r Zget_reverse_indexerr% r& )r Zindexerr] r) r r r �test_get_reverse_indexer� s z%TestIndexing.test_get_reverse_indexerN) r0 r1 r2 rI rO rS r\ r^ ra r r r r r3 1 s "4 r3 c C s t jjd k st�d S )N)r Zhasnans�__doc__r r r r r �'test_cache_readonly_preserve_docstrings� s rc )Znumpyr r Z pandas._libsr r r Zpandasr- r Zpandas._testingZ_testingr% r r3 rc r r r r �<module> s
& |