3
���h� � @ s< d dl Zd dlmZmZmZmZ d dljZ G dd� d�Z
dS )� N)� DataFrame�Index� PeriodIndex� period_rangec @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestPeriodIndexc C sv t ddd�}ttjjdd�|d�}||d }tj||jd d �df � t|� |d }tj||jd d �df � d S )Nz1/1/2000� )�periods�
)�columnsr ) r r �np�random�randn�tmZassert_series_equalZiloc�repr)�self�rng�df�ts� r �@/tmp/pip-build-5_djhm0z/pandas/pandas/tests/frame/test_period.py�test_as_frame_columns s z%TestPeriodIndex.test_as_frame_columnsc C s� t dddd�}ttjjdd�|d�}||d< t|d �}tj||dd � |jdksVt �|jdksdt �|j
� j d�}t |j
t�s�t �tj|j
|� d S )
Nz1/1/2000r �index)r �name� )r r F)Z check_names)r r r r r
r r Zassert_index_equalr �AssertionErrorZ reset_indexZ set_index�
isinstancer r )r r r �rsr r r �test_frame_setitem s z"TestPeriodIndex.test_frame_setitemc C s4 t dddgdd�}ttjjdd�|d�}|j� d S ) Nz2011-1z2011-2z2011-3�M)�freqr � )r )r r r r r
Z to_string)r r �framer r r �test_frame_index_to_string# s z*TestPeriodIndex.test_frame_index_to_stringN)�__name__�
__module__� __qualname__r r r" r r r r r s
r ) Znumpyr Zpandasr r r r Zpandas._testingZ_testingr r r r r r �<module> s
|