U
|
ý°dÐ ã @ sV d dl mZ d dlZd dlZd dlZd dlmZmZm Z d dl
|
m Z G dd dZ dS )é )Ú methodcallerN)Ú
|
MultiIndexÚSeriesÚ
|
date_rangec @ sÆ e Zd Zej dddg¡dd Zdd Zdd Zd
|
d Z ej d e
|
j e j d dg¡dd Zej d e
|
j e j g¡dd Zej d ddg¡dd Zej d ddddg¡dd Zd d Zd d Zd!S )"Ú
|
TestSeriesÚfuncZ rename_axisÚ_set_axis_namec C s t dddgtjdd dD ddgd d
|
}t|d d g|}|jjd ksLt|jjddgks`t|jjd kspt|jjstd d gd S ) Né é é c S s g | ] }d |fqS )ÚA© )Ú.0Úxr r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/generic/test_series.pyÚ
|
<listcomp> s z4TestSeries.test_set_axis_name_mi.<locals>.<listcomp>)ÚaÚBÚcÚl1Úl2)Únames)ÚindexZL1ZL2)r r Ú from_tuplesr r ÚnameÚAssertionErrorr )Úselfr ÚserÚresultr r r Útest_set_axis_name_mi s ÿþz TestSeries.test_set_axis_name_mic C s: t dg}d}tjt|d |jddd W 5 Q R X d S )Né z&No axis named 1 for object type Series©Úmatchr )r Zaxis)r ÚpytestÚraisesÚ
|
ValueErrorr )r r Úmsgr r r Útest_set_axis_name_raises s
|
z$TestSeries.test_set_axis_name_raisesc C s& t dddg}| ¡ }t ||¡ d S ©NTF)r Z_get_bool_dataÚtmZassert_series_equal)r r r r r r Ú!test_get_bool_data_preserve_dtype% s z,TestSeries.test_get_bool_data_preserve_dtypec C s0 t dg}| ¡ stt dg}| ¡ r,td S r( )r Úboolr )r r r r r Útest_nonzero_single_element* s
|
|
z&TestSeries.test_nonzero_single_elementÚdataTFc C s4 t |g}d}tjt|d t| W 5 Q R X d S ©Nú(The truth value of a Series is ambiguousr! ©r r# r$ r% r+ ©r r- Zseriesr& r r r Ú#test_nonzero_single_element_raise_12 s
|
z.TestSeries.test_nonzero_single_element_raise_1c C s4 t |g}d}tjt|d | ¡ W 5 Q R X d S )Nú6bool cannot act on a non-boolean single element Seriesr! r0 r1 r r r Ú#test_nonzero_single_element_raise_2; s
|
z.TestSeries.test_nonzero_single_element_raise_2)TT)FFc C sV t |g}d}tjt|d t| W 5 Q R X tjt|d | ¡ W 5 Q R X d S r. r0 r1 r r r Ú#test_nonzero_multiple_element_raiseC s
|
z.TestSeries.test_nonzero_multiple_element_raiser r r g c C sZ t |g}d}tjt|d t| W 5 Q R X d}tjt|d | ¡ W 5 Q R X d S )Nr/ r! r3 r0 r1 r r r Ú!test_nonbool_single_element_raiseN s
|
z,TestSeries.test_nonbool_single_element_raisec C sx t tj d¡tdddddd}| d¡ ¡ }t ||¡ | d¡ ¡ }t ||¡ | d¡
|
dd ¡}t ||¡ d S )
|
Niè Z20130101Ús)ZperiodsÚfreqÚfoo)r r Z1Tc S s | ¡ S ©N)Úsum)r r r r Ú<lambda>h ó zETestSeries.test_metadata_propagation_indiv_resample.<locals>.<lambda>) r ÚnpÚrandomZrandr ZresampleZmeanr) Úassert_metadata_equivalentÚminÚapply)r Útsr r r r Ú(test_metadata_propagation_indiv_resample[ s
|
ý z3TestSeries.test_metadata_propagation_indiv_resamplec C sº t tdtd}d|_t tdtd}d|_|j}t ||¡ d dd}| ¡ \}| t dddg¡ | t d |¡ d|_d|_t
|
||g¡}|jd
|
kst |jd ks¬t W 5 Q R X d S ) Né r9 Úbarc s` | j D ]T |dkrD dkrDd fdd|jD ¡}t | |¡ qt | t| d ¡ q| S )NÚconcatÚfilenameú+c s" g | ]}t | d rt | qS r: )Úgetattr)r Úobj©r r r r z s þzPTestSeries.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>)Ú _metadataÚjoinÚobjsÚobjectÚ __setattr__rJ )r ÚotherÚmethodÚkwargsÚvaluer rL r Úfinalizev s
|
|
þÿz<TestSeries.test_metadata_propagation_indiv.<locals>.finalizerM r rH Z __finalize__zfoo+bar)N) r Úranger ÚTr) r@ ÚcontextÚsetattrrH ÚpdrG r )r Z monkeypatchr Zser2r rV Úmr r r Ú test_metadata_propagation_indivk s
|
|
z*TestSeries.test_metadata_propagation_indivN)Ú__name__Ú
|
__module__Ú __qualname__r# ÚmarkZ parametrizer r' r* r, r> Únanr[ ZNaTr2 r4 r5 r6 rD r] r r r r r s
|
|
|
|
|
|
r )Úoperatorr Únumpyr> r# Zpandasr[ r r r Zpandas._testingZ_testingr) r r r r r Ú<module> s
|
