U
|
¬ý°d' ã @ sÆ d dl Z d dlmZ d dlZd dlZd dlmZ d dl m
|
Z
|
dd Z e j ed ddd Z G d
|
d d Zej d ejejjjfejejjjfejejjjfg¡d d Zdd Zdd Zdd Z dS )é N)Ú Generator)Úaccessorc C s: G dd dt j} dd t| D }|ddgks6td S )Nc @ s( e Zd ZU dZeed< ddddZdS )z test_dirname_mixin.<locals>.Xé ÚyN©Úreturnc S s
|
d| _ d S )Né )Úz©Úself© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_register_accessor.pyÚ__init__ s z&test_dirname_mixin.<locals>.X.__init__)Ú__name__Ú
|
__module__Ú __qualname__ÚxÚintÚ__annotations__r r r r r ÚX s
|
r c S s g | ]}| d ¡s|qS )Ú_)Ú
|
startswith)Ú.0Ú attr_namer r r Ú
|
<listcomp> s
|
z&test_dirname_mixin.<locals>.<listcomp>r r )r Z DirNamesMixinÚdirÚAssertionError)r Úresultr r r Útest_dirname_mixin s r )NNNr c c sB z
|
dV W 5 zt | | W n tk
|
r. Y nX | j |¡ X dS )z[Ensure that an attribute added to 'obj' during the test is
|
removed when we're done
|
N)ÚdelattrÚAttributeErrorÚ
|
_accessorsÚdiscard)ÚobjÚattrr r r Úensure_removed s
|
r% c @ s. e Zd ZddddZedd Zdd ZdS ) Ú
|
MyAccessorNr c C s || _ d| _d S )NÚitem)r# r' )r r# r r r r * s zMyAccessor.__init__c C s | j S ©N©r' r
|
r r r Úprop. s zMyAccessor.propc C s | j S r( r) r
|
r r r Úmethod2 s zMyAccessor.method)r r r r Úpropertyr* r+ r r r r r& ) s
|
r& zobj, registrarc C s t | dx tt| }|dt | tjk r6| g n
|
| g td}|jjdksRt tt| }||A dhkspt d| j
|
ks~t W 5 Q R X d S )NÚmine©Zdtyper' ) r% Úsetr r& ÚpdÚSeriesÚobjectr- r* r r! )r# Z registrarZbeforeÚoÚafterr r r Ú test_register6 s r5 c C sn t tjdX tjj d¡t t ddg¡} | jj| ks>t | jj
|
dksNt | j ¡ dks`t W 5 Q R X d S )Nr- r é r' ) r% r0 r1 ÚapiÚ
|
extensionsÚregister_series_accessorr& r- r# r r* r+ )Úsr r r Útest_accessor_worksI s r; c C s t jj} z|t t¡6}t jj d¡t t ddg¡}|jj
|
dksFt W 5 Q R X t |d j }d|ksjt d|ksvt d|kst W 5 | t j_X d S )NÚmeanr r6 r' r r& r1 )r0 r1 r< ÚtmZassert_produces_warningÚ UserWarningr7 r8 r9 r& r* r ÚstrÚmessage)r< Úwr: Úmsgr r r Útest_overwrite_warnsT s rC c
|
C sb t tjdL tjj d¡G dd d} tjtdd tjg t dj
|
W 5 Q R X W 5 Q R X d S )NÚbadc @ s e Zd ZddddZdS )z(test_raises_attribute_error.<locals>.BadNr c S s t dd S )NÚwhoops)r )r Údatar r r r i s z1test_raises_attribute_error.<locals>.Bad.__init__)r r r r r r r r ÚBadg s rG rE )Úmatchr. ) r% r0 r1 r7 r8 r9 ÚpytestZraisesr r2 rD )rG r r r Útest_raises_attribute_errord s
|
rJ ) Ú
|
contextlibÚtypingr rI Zpandasr0 Zpandas._testingZ_testingr= Z pandas.corer r Úcontextmanagerr% r& ÚmarkZ parametrizer1 r7 r8 r9 Z DataFrameZregister_dataframe_accessorZIndexZregister_index_accessorr5 r; rC rJ r r r r Ú<module> s( ýþ
|
|
