U
|
ý°d¼ ã @ s^ d dl Zd dlZd dlmZmZ d dlmZ G dd dZ G dd de Z
|
G dd de Z dS ) é N)Ú DataFrameÚSeriesc @ s\ e Zd Zejdd Zdd Zdd Zdd Zej
|
d d
|
d g¡d d Z dd Z dd Z dS )ÚSharedSetAxisTestsc C s t dd S )NzImplemented by subclasses)ÚNotImplementedError)Úself© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_set_axis.pyÚobj s zSharedSetAxisTests.objc C s@ t dd t|
}| ¡ }||_|j|dd}t ||¡ d S )NÚabcdr ©Úaxis©Ú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_axisc s0 t dd t
} jd d
¡ }||_ j|dddt |¡ k sXt|s j dkrxt
|
¡rtn&t fddt j d D rt j|dddt |¡ k sÆt j dkrât
|
¡sàtn(t fd dt j d D s
|
t j|dd
|
t |¡ k s2t|r j dkrXt
|
¡stn(t fd dt j d D sÊtnH j dkr¢t
|
¡rÊtn(t fd dt j d D rÊt j|dd t |¡ j dkrt
|
¡s,tn(tfddt j d D s,td S )Nr
|
r T)r r é c 3 s6 | ].}t jd d
|f jd d
|f ¡V qd S ©N©r Ú shares_memoryÚiloc©Ú.0Úi©r r r r Ú <genexpr>( s ÿz8SharedSetAxisTests.test_set_axis_copy.<locals>.<genexpr>Fc 3 s6 | ].}t jd d
|f jd d
|f ¡V qd S r r r r" r r r# 4 s ÿr c 3 s6 | ].}t jd d
|f jd d
|f ¡V qd S r r r r" r r r# B s ÿc 3 s6 | ].}t jd d
|f jd d
|f ¡V qd S r r r r" r r r# K s ÿ)r c 3 s6 | ].}t jd d
|f jd d
|f ¡V qd S r r r )ÚorigÚresr r r# V s ÿ)r r r r r r r r ÚAssertionErrorÚndimr ÚanyÚrangeÚshapeÚall)r r Zusing_copy_on_writer r r )r r$ r% r r Útest_set_axis_copy sT
|
þ
|
þ þ þ þz%SharedSetAxisTests.test_set_axis_copyc C s< t dd t|
}| ¡ }||_| |¡}t ||¡ d S )Nr
|
r r r r r Ú!test_set_axis_unnamed_kwarg_warns[ s
|
|
z4SharedSetAxisTests.test_set_axis_unnamed_kwarg_warnsr é Zfooc C s0 t jtdd |jtd|d W 5 Q R X d S )Nz No axis named©ÚmatchÚabcr )ÚpytestÚraisesÚ
|
ValueErrorr r )r r r r r r Ú test_set_axis_invalid_axis_namee s z2SharedSetAxisTests.test_set_axis_invalid_axis_namec C s( d}t jt|d d |_W 5 Q R X d S )NzNIndex\(\.\.\.\) must be called with a collection of some kind, None was passedr/ )r2 r3 Ú TypeErrorr ©r r Úmsgr r r Ú*test_set_axis_setattr_index_not_collectionk s ÿz=SharedSetAxisTests.test_set_axis_setattr_index_not_collectionc C s dt | dt |d d}tjt|d t t |d ¡|_W 5 Q R X |jdkrtjtdd |jd d d
|_W 5 Q R X d S )Nz#Length mismatch: Expected axis has z elements, new values have r z elementsr/ é zLength mismatch) r r2 r3 r4 ÚnpÚaranger r' Úcolumnsr7 r r r Ú(test_set_axis_setattr_index_wrong_lengtht s ÿ
|
z;SharedSetAxisTests.test_set_axis_setattr_index_wrong_lengthN)Ú__name__Ú
|
__module__Ú __qualname__r2 Úfixturer r r, r- ÚmarkZ parametrizer5 r9 r> r r r r r s
|
C
|
|
r c @ s e Zd Zejdd ZdS )ÚTestDataFrameSetAxisc C s0 t dddgdddgddd gd
|
d d d gd}|S )Ngñ?g@gffffff
|
@g @gffffff@gÍÌÌÌÌÌ @g@g @gffffff@)ÚAÚBÚCiÚ iÛ iÜ )r )r )r Zdfr r r r s
|
þzTestDataFrameSetAxis.objN©r? r@ rA r2 rB r r r r r rD s rD c @ s e Zd Zejdd ZdS )ÚTestSeriesSetAxisc C s t t d¡ddddgdd}|S )Né r r. é é Úint64)r Zdtype)r r; r< )r Zserr r r r s zTestSeriesSetAxis.objNrH r r r r rI s rI ) Únumpyr; r2 Zpandasr r Zpandas._testingZ_testingr r rD rI r r r r Ú<module> s w
|
