U
|
ý°d2 ã @ s¾ d Z ddlZddlZddlZddlmZ ddlmZ dd Z
|
dd Z dd Z d
|
d Z d d Zdd Zdd Zdd ZG dd dZG dd dZG dd dZG dd dZG d d d ZdS ) z9
|
Tests that can be parametrized over _any_ Index object.
|
é N)ÚInvalidIndexErrorc C sD t jtdd
|
| rW 5 Q R X t jtdd t| W 5 Q R X d S )NzThe truth value of a©Úmatch)ÚpytestÚraisesÚ
|
ValueErrorÚbool©Úindex© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/test_any_index.pyÚtest_boolean_context_compat s
|
|
r c C s* d}t jt|d | ¡ W 5 Q R X d S )Nz=cannot sort an Index object in-place, use sort_values insteadr )r r Ú TypeErrorÚsort©r
|
Úmsgr r r Ú test_sort s r c C s4 t jtdt| j dd t| W 5 Q R X d S )Nzunhashable type: 'ú'r )r r r ÚtypeÚ__name__Úhashr r r r Útest_hash_error s r c C s: t | s d S d}tjt|d | d | d< W 5 Q R X d S )Nz)Index does not support mutable operationsr r )Úlenr r r r r r r Útest_mutability# s
|
r c C sJ | dd ¡}| jtkr6|jtkr6| |k ¡ s2td S tj|| dd d S )Nc S s | S ©Nr )Úxr r r Ú<lambda>. ó z+test_map_identity_mapping.<locals>.<lambda>Úequiv©Úexact)ÚmapÚdtypeÚobjectr ÚallÚAssertionErrorÚtmÚassert_index_equal)r
|
ÚrequestÚresultr r r Útest_map_identity_mapping+ s
|
r* c C s4 | j dddg }tjtdd || _W 5 Q R X d S )NZappleZbananaZcarrotz^Lengthr )Únlevelsr r r Únames)r
|
r, r r r Útest_wrong_number_names6 s r- c C s | ¡ j| jkstd S r )ÚviewÚnamer% r r r r Útest_view_preserves_name< s r0 c C s | ¡ }t || ¡ d S r )Zravelr& r' )r
|
Úresr r r Ú
|
test_ravel@ s r2 c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestConversionc C s: | ¡ }|j|jk st|j|k s&t|j|jks6td S r ©Z to_seriesÚvaluesr% r
|
r/ ©Úselfr
|
Zserr r r Útest_to_seriesG s z TestConversion.test_to_seriesc C sx |j |d}|j|jk s t|j|ks*t|j|jks:t|j dd}|j|jk sVt|j|k sdt|j|jksttd S )Nr Z__test)r/ r4 r6 r r r Ú test_to_series_with_argumentsO s z,TestConversion.test_to_series_with_argumentsc C s | ¡ t|kstd S r )ÚtolistÚlistr% ©r7 r
|
r r r Útest_tolist_matches_list` s z'TestConversion.test_tolist_matches_listN)r Ú
|
__module__Ú __qualname__r8 r9 r= r r r r r3 F s r3 c @ s e Zd Zdd Zdd ZdS )ÚTestRoundTripsc C s6 t |¡}t j||dd |jdkr2| |¡s2td S )NTr é )r& Úround_trip_pickler' r+ Z equal_levelsr% ©r7 r
|
r) r r r Útest_pickle_roundtripe s
|
|
z$TestRoundTrips.test_pickle_roundtripc C s0 |j d }|_ t |¡}| |¡s&t||_ d S )NÚfoo)r/ r& rB Úequalsr% )r7 r
|
Ú original_nameZ unpickledr r r Útest_pickle_preserves_namel s
|
z)TestRoundTrips.test_pickle_preserves_nameN)r r> r? rD rH r r r r r@ d s r@ c @ s@ e Zd Zdd Zdd Zdd Zej ddd d
|
g¡d d Z d S )Ú TestIndexingc C sf t jddgt jd}tjtdd | |¡ W 5 Q R X tjtdd | | t¡¡ W 5 Q R X d S )Nr rA )r" z\[0 1\]r z\[False True\]) ÚnpÚarrayZintpr r r Zget_locZastyper )r7 r
|
Úkeyr r r Ú0test_get_loc_listlike_raises_invalid_index_errort s
|
z=TestIndexing.test_get_loc_listlike_raises_invalid_index_errorc C s& |d }| |¡st||k s"td S )N.)rF r% rC r r r Útest_getitem_ellipsis~ s z"TestIndexing.test_getitem_ellipsisc C s |j |dd
j kstd S )NrA )r/ r% r< r r r Útest_slice_keeps_name s z"TestIndexing.test_slice_keeps_nameÚitemée Zno_intg @c C s< d dt d¡dg¡}tjt|d || W 5 Q R X d S )Nú|z5index 101 is out of bounds for axis 0 with size [\d]+zuonly integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indiceszindex out of boundsr )ÚjoinÚreÚescaper r Ú
|
IndexError)r7 r
|
rP r r r r Útest_getitem_error s ÿùÿ z TestIndexing.test_getitem_errorN)
|
r r> r? rM rN rO r ÚmarkZ parametrizerW r r r r rI s s
|
|
rI c @ s e Zd Zdd ZdS )Ú TestRenderingc C s0 d|_ dt|kstt|jt|ks,td S )NrE z'foo')r/ Ústrr% r r r< r r r Útest_str s zTestRendering.test_strN)r r> r? r[ r r r r rY s rY c @ s e Zd Zdd ZdS )ÚTestReductionsc C s d}t jt|d |jdd W 5 Q R X t jt|d |jdd W 5 Q R X t jt|d |jdd W 5 Q R X t jt|d |jdd W 5 Q R X d S )Nz8`axis` must be fewer than the number of dimensions \(1\)r rA )Zaxisé éþÿÿÿéýÿÿÿ)r r r ZargmaxZargminÚminÚmax)r7 r
|
r r r r Útest_argmax_axis_invalid¡ s z'TestReductions.test_argmax_axis_invalidN)r r> r? rb r r r r r\ s r\ )Ú__doc__rT ÚnumpyrJ r Z pandas.errorsr Zpandas._testingZ_testingr& r r r r r* r- r0 r2 r3 r@ rI rY r\ r r r r Ú<module> s$
|
%
|
