3
Ud                 @   s\   d dl Zd dlZd dlmZmZ d dljZG dd dZ	G dd de	Z
G dd de	ZdS )	    N)	DataFrameSeriesc               @   sb   e Zd Zejdd Zdd Zejjddddd	gd
d Z	dd Z
ejjdddgdd ZdS )SharedSetAxisTestsc             C   s   t dd S )NzImplemented by subclasses)NotImplementedError)self r   ^/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/generic/methods/test_set_axis.pyobj	   s    zSharedSetAxisTests.objc             C   sB   t dd t| }|j }||_|j|ddd}tj|| d S )Nabcdr   F)axisinplace)listlencopyindexset_axistmassert_equal)r   r	   	new_indexexpectedresultr   r   r   test_set_axis   s
    z SharedSetAxisTests.test_set_axisr   r   r      columnsc             C   sp   |j dkr|dkrd S tdd t| }|j }|d	krB||_n||_|j }|j||dd tj|| d S )
Nr   r   r
   r   r   T)r   r   )r   r   )r   r   )	ndimr   r   r   r   r   r   r   r   )r   r   r	   r   r   r   r   r   r   test_set_axis_inplace_axis   s    z-SharedSetAxisTests.test_set_axis_inplace_axisc             C   sV   t dd t| }|j }||_tjd  |j|dd}W d Q R X tj|| d S )Nr
   F)r   )r   r   r   r   r   Zassert_produces_warningr   r   )r   r	   r   r   r   r   r   r   !test_set_axis_unnamed_kwarg_warns+   s    z4SharedSetAxisTests.test_set_axis_unnamed_kwarg_warns   Zfooc             C   s0   t jtdd |jtd|d W d Q R X d S )NzNo axis named)matchabc)r   )pytestZraises
ValueErrorr   r   )r   r   r	   r   r   r   test_set_axis_invalid_axis_name6   s    z2SharedSetAxisTests.test_set_axis_invalid_axis_nameN)__name__
__module____qualname__r    fixturer	   r   markZparametrizer   r   r"   r   r   r   r   r      s
   r   c               @   s   e Zd Zejdd ZdS )TestDataFrameSetAxisc             C   s0   t dddgdddgddd	gd
dddgd}|S )Ng?g@gffffff
@g      @gffffff@g@g@g      @gffffff@)ABCi  i  i  )r   )r   )r   dfr   r   r   r	   >   s    zTestDataFrameSetAxis.objN)r#   r$   r%   r    r&   r	   r   r   r   r   r(   =   s   r(   c               @   s   e Zd Zejdd ZdS )TestSeriesSetAxisc             C   s    t tjdddddgdd}|S )N   r   r         Zint64)r   Zdtype)r   npZarange)r   Zserr   r   r   r	   H   s    zTestSeriesSetAxis.objN)r#   r$   r%   r    r&   r	   r   r   r   r   r-   G   s   r-   )Znumpyr1   r    Zpandasr   r   Zpandas._testingZ_testingr   r   r(   r-   r   r   r   r   <module>   s   
5
